formatting

Author: devsetgo

dsg_lib/fastapi_functions/default_endpoints.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
+from typing import Dict, List
+
 from fastapi import APIRouter, Response
-from typing import List, Dict
 from loguru import logger
 
 

examples/fastapi_example.py

@@ -24,9 +24,7 @@
     database_operations,
 )
 from dsg_lib.common_functions import logging_config
-from dsg_lib.fastapi_functions import (
-    system_health_endpoints, default_endpoints
-)
+from dsg_lib.fastapi_functions import default_endpoints, system_health_endpoints
 
 config = [
     {"bot": "Bytespider", "allow": False},
@@ -185,7 +183,11 @@ async def root():
 }
 
 # Create and include the health router if enabled
-if config["enable_status_endpoint"] or config["enable_uptime_endpoint"] or config["enable_heapdump_endpoint"]:
+if (
+    config["enable_status_endpoint"]
+    or config["enable_uptime_endpoint"]
+    or config["enable_heapdump_endpoint"]
+):
     health_router = system_health_endpoints.create_health_router(config)
     app.include_router(health_router, prefix="/api/health", tags=["system-health"])
 
@@ -195,7 +197,6 @@ async def root():
     app.include_router(default_router, prefix="", tags=["default"])
 
 
-
 async def create_a_bunch_of_users(single_entry=0, many_entries=0):
     logger.info(f"single_entry: {single_entry}")
     await async_db.create_tables()

tests/test_common_functions/test_file_mover/__init__.py

@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 import asyncio
 import secrets
-
+import pytest_asyncio
 import pytest
 from sqlalchemy import Column, Integer, String, delete, insert, select
 from sqlalchemy.exc import IntegrityError, SQLAlchemyError
@@ -28,6 +28,22 @@ class User(async_db.Base):
     name = Column(String, unique=True)
 
 
+@pytest_asyncio.fixture(scope="class", autouse=True)
+async def setup_database():
+    await async_db.create_tables()
+    yield
+    # await async_db.drop_tables()
+
+
+@pytest_asyncio.fixture(scope="function", autouse=True)
+async def setup_teardown():
+    # Clean the database before each test
+    await db_ops.execute_one(delete(User))
+    yield
+    # Clean the database after each test
+    await db_ops.execute_one(delete(User))
+
+
 class TestDatabaseOperations:
     @pytest.fixture(scope="session")
     def db_ops(self):
@@ -440,19 +456,6 @@ async def test_delete_many_exception(self, db_ops):
         # assert result contains "error"
         assert "error" in result
 
-    @pytest.fixture(scope="class", autouse=True)
-    async def setup_database(self):
-        await async_db.create_tables()
-        yield
-        await async_db.drop_tables()
-
-    @pytest.fixture(scope="function", autouse=True)
-    async def setup_teardown(self):
-        # Clean the database before each test
-        await db_ops.execute_one(delete(User))
-        yield
-        # Clean the database after each test
-        await db_ops.execute_one(delete(User))
 
     @pytest.mark.asyncio
     async def test_execute_one_insert(self):

tests/test_database_functions/test_async_database.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 from fastapi import FastAPI
 from fastapi.testclient import TestClient
+
 from dsg_lib.fastapi_functions.default_endpoints import create_default_router
 
 # Create a FastAPI app for testing
@@ -34,6 +35,7 @@
 default_router = create_default_router(config)
 app.include_router(default_router, prefix="", tags=["default"])
 
+
 def test_robots_txt():
     response = client.get("/robots.txt")
     assert response.status_code == 200