FEAT: Add ADMIN_EMAIL environment variable, update admin creation logic to use it and apply to test_admin and conftest

Author: IkeSalmonson

app/routers/admin/routes.py

@@ -18,11 +18,12 @@
 async def create_admin(session: AsyncSession):
     ADMIN_USER = os.getenv("ADMIN_USER")
     ADMIN_PASSWORD = os.getenv("ADMIN_PASSWORD")
+    ADMIN_EMAIL = os.getenv("ADMIN_EMAIL")
     password = ADMIN_PASSWORD
     hashed_password = auth.hash_password(password)
     community = DBCommunity(
         username=ADMIN_USER,
-        email="[email protected]",
+        email=ADMIN_EMAIL,
         password=hashed_password,
         role="admin",
     )

docker-compose.yaml

@@ -17,6 +17,7 @@ services:
       - ENCRYPTION_KEY=r0-QKv5qACJNFRqy2cNZCsfZ_zVvehlC-v8zDJb--EI=
       - ADMIN_USER=admin
       - ADMIN_PASSWORD=admin
+      - ADMIN_EMAIL = [email protected]
       - ALGORITHM=HS256
       - ACCESS_TOKEN_EXPIRE_MINUTES=20
     restart: unless-stopped

tests/conftest.py

@@ -1,3 +1,4 @@
+import os
 from collections.abc import AsyncGenerator
 
 import pytest
@@ -22,6 +23,10 @@
 # Usamos engine e AsyncSessionLocal apenas para os testes.
 # Isso garante que os testes são isolados e usam o banco de dados em memória.
 TEST_DATABASE_URL = "sqlite+aiosqlite:///:memory:"
+os.environ["ADMIN_USER"] = "ADMIN_USER"
+os.environ["ADMIN_PASSWORD"] = "ADMIN_PASSWORD"
+os.environ["ADMIN_EMAIL"] = "ADMIN_EMAIL"
+
 
 test_engine = create_async_engine(TEST_DATABASE_URL, echo=False, future=True)
 

tests/test_admin.py

@@ -0,0 +1,35 @@
+import pytest
+from sqlmodel import select
+from sqlmodel.ext.asyncio.session import AsyncSession
+
+from app.routers.admin.routes import create_admin
+from app.services.auth import verify_password
+from app.services.database.models import Community
+from app.services.encryption import decrypt_email
+
+# Dados de teste
+TEST_USERNAME = "test_user_crypto"
+TEST_EMAIL = "[email protected]"
+TEST_PASSWORD = "@SafePassword123"
+import os
+
+
+@pytest.mark.asyncio
+async def test_insert_admin(session: AsyncSession):
+    ADMIN_USER = os.getenv("ADMIN_USER")
+    ADMIN_EMAIL = os.getenv("ADMIN_EMAIL")
+    ADMIN_PASSWORD = os.getenv("ADMIN_PASSWORD")
+
+    community = Community(
+        username=ADMIN_USER,
+    )
+
+    await create_admin(session)
+    statement = select(Community).where(community.username == ADMIN_USER)
+    result = await session.exec(statement)
+    found = result.first()
+
+    assert found is not None
+    assert found.username == ADMIN_USER
+    assert decrypt_email(found.email) == ADMIN_EMAIL
+    assert verify_password(ADMIN_PASSWORD, found.password)