.

Author: pentest-gg

pyproject.toml

@@ -6,6 +6,10 @@ ignore_missing_imports = true
 asyncio_mode = "strict"
 asyncio_default_fixture_loop_scope = "function"
 addopts = "--ignore=test_build.py"
+filterwarnings = [
+    "ignore::DeprecationWarning:.*datetime.datetime.utcnow.*",
+    "ignore::DeprecationWarning:.*botocore.*",
+]
 
 [tool.ruff]
 

tests/conftest.py

@@ -5,6 +5,8 @@
 from fastapi_users import schemas
 
 DATABASE_REGION: str = "eu-central-1"
+DATABASE_TOKENTABLE_PRIMARY_KEY: str = "token"
+DATABASE_USERTABLE_PRIMARY_KEY: str = "id"
 
 
 class User(schemas.BaseUser):

tests/tables.py

@@ -2,18 +2,23 @@
 import botocore.exceptions
 
 
-async def ensure_table_exists(session: aioboto3.Session, table_name: str, region: str):
+async def ensure_table_exists(
+    session: aioboto3.Session,
+    table_name: str,
+    primary_key: str,
+    region: str,
+):
     async with session.client("dynamodb", region_name=region) as client:
         try:
             await client.describe_table(TableName=table_name)
         except botocore.exceptions.ClientError:
             await client.create_table(
                 TableName=table_name,
                 KeySchema=[
-                    {"AttributeName": "id", "KeyType": "HASH"},
+                    {"AttributeName": primary_key, "KeyType": "HASH"},
                 ],
                 AttributeDefinitions=[
-                    {"AttributeName": "id", "AttributeType": "S"},
+                    {"AttributeName": primary_key, "AttributeType": "S"},
                 ],
                 ProvisionedThroughput={"ReadCapacityUnits": 5, "WriteCapacityUnits": 5},
             )