setup tests and first endpoint tests

Author: dfm88

fastapi_2fa/api/endpoints/api_v1/auth.py

@@ -39,7 +39,7 @@
     },
 )
 async def signup(
-    db: Session = Depends(get_db), user_data: UserCreate = Depends()
+    user_data: UserCreate, db: Session = Depends(get_db)
 ) -> Any:
     try:
         user = await user_crud(transaction=True).create(

fastapi_2fa/core/config.py

@@ -87,7 +87,19 @@ class ProductionConfig(BaseConfig):
 
 
 class TestingConfig(BaseConfig):
-    LOGGING_LEVEL = "DEBUG"
+    # define new attributes with respect to BaseConfig
+    DATABASE_CONNECT_DICT: dict = {"check_same_thread": False}
+    LOGGING_LEVEL = 'DEBUG'
+    SQLALCHEMY_DATABASE_TEST_SYNC_URI: str = "sqlite:///tests/test_db.db"
+
+    # celery test config
+    task_always_eager = True
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        # override attributes of BaseConfig
+        # https://fastapi.tiangolo.com/advanced/testing-database/
+        self.SQLALCHEMY_DATABASE_URI: str = "sqlite+aiosqlite:///tests/test_db.db"
 
 
 @lru_cache()

fastapi_2fa/core/two_factor_auth.py

@@ -40,18 +40,21 @@ def get_decoded_two_factor_auth_key(value: str) -> str:
     return _fernet_decode(value=value)
 
 
+def _get_fake_otp_token(nr_digits: int = settings.TFA_TOKEN_LENGTH):
+    return ''.join(
+        str(random.randint(0, 9)) for _ in range(nr_digits)
+    )
+
+
 def get_fake_otp_tokens(
     nr_digits: int = settings.TFA_TOKEN_LENGTH,
     nr_tokens: int = settings.TFA_BACKUP_TOKENS_NR
 ) -> Iterator[str]:
     for _ in range(nr_tokens):
-        random_otp = ''.join(
-            str(random.randint(0, 9)) for _ in range(nr_digits)
-        )
+        random_otp = _get_fake_otp_token(nr_digits=nr_digits)
         yield random_otp
 
 
-
 def get_current_totp(user: User) -> pyotp.TOTP:
     assert user.tfa_enabled is True, 'User does not have TFA enabled'
     assert user.device is not None, 'User has no associated device'

poetry.lock

@@ -12,7 +12,7 @@ fastapi = "^0.87.0"
 passlib = {extras = ["bycrypt"], version = "^1.7.4"}
 pytest = "^7.2.0"
 python-jose = {extras = ["cryptography"], version = "^3.3.0"}
-celery = "^5.2.7"
+celery = {extras = ["pytest"], version = "^5.2.7"}
 pydantic = {extras = ["email"], version = "^1.10.2"}
 ipdb = "^0.13.9"
 python-dotenv = "^0.21.0"
@@ -31,6 +31,10 @@ qrcode = {extras = ["pil"], version = "^7.3.1"}
 pillow = "^9.3.0"
 redis = "^4.3.5"
 flower = "1.1.0"
+pytest-asyncio = "^0.20.2"
+aiosqlite = "^0.17.0"
+httpx = "^0.23.1"
+pytest-celery = "^0.0.0"
 
 
 [tool.poetry.group.dev.dependencies]