docker-compose added with local testing

.env

@@ -5,6 +5,6 @@ REDIS_PORT=6379
 REDIS_HOST=redis
 FLOWER_BASIC_AUTH=flower_user:flower_pass
 REDIS_URL=redis://redis:6379/0
-DATABASE_URL=postgresql+asyncpg://test_user:test_pass@postgres:5432/test_db
+DATABASE_URL=postgresql//test_user:test_pass@postgres:5432/test_db
 BROKER_URL=redis://redis:6379/0
 

app/core/config.py

@@ -0,0 +1,41 @@
+import os
+from pydantic.v1 import BaseSettings, Field
+from loguru import logger
+
+
+class Settings(BaseSettings):
+    POSTGRES_DB: str = Field("test_db", env="POSTGRES_DB")
+    POSTGRES_USER: str = Field("test_user", env="POSTGRES_USER")
+    POSTGRES_PASSWORD: str = Field("test_pass", env="POSTGRES_PASSWORD")
+
+    is_docker: bool = Field(default_factory=lambda: os.environ.get("DOCKER_ENV") == "1")
+
+    database_url: str = Field(..., env="DATABASE_URL")
+    redis_url: str = Field(..., env="REDIS_URL")
+    flower_basic_auth: str = Field(..., env="FLOWER_BASIC_AUTH")
+    broker_url: str = Field(..., env="BROKER_URL")
+
+    class Config:
+        env_file = ".env"
+
+    @classmethod
+    def create(cls):
+        """Create instance and dynamically set database and redis URLs."""
+        instance = cls()
+        # Set the correct database URL without the +asyncpg
+        instance.database_url = (
+            f"postgres://{instance.POSTGRES_USER}:{instance.POSTGRES_PASSWORD}@"
+            f"{'localhost' if not instance.is_docker else 'postgres'}:5432/{instance.POSTGRES_DB}"
+        )
+        instance.redis_url = (
+            f"redis://{'localhost' if not instance.is_docker else 'redis'}:6379/0"
+        )
+        instance.broker_url = instance.redis_url
+        return instance
+
+
+# Instantiate Settings
+settings = Settings.create()
+
+if __name__ == "__main__":
+    logger.info(f"Settings: {settings.dict()}")

docker-compose.yaml

@@ -7,6 +7,7 @@ services:
       POSTGRES_DB: ${POSTGRES_DB}
       POSTGRES_USER: ${POSTGRES_USER}
       POSTGRES_PASSWORD: ${POSTGRES_PASSWORD}
+      DOCKER_ENV: 1
     ports:
       - "5432:5432"
     volumes:
@@ -22,6 +23,8 @@ services:
     ports:
       - "6379:6379"
       - "8001:8001"
+    environment:
+      DOCKER_ENV: 1
     healthcheck:
       test: ["CMD", "redis-cli", "ping"]
       interval: 30s
@@ -30,6 +33,8 @@ services:
 
   flower:
     image: mher/flower:0.9.7
+    environment:
+      DOCKER_ENV: 1
     command: ["flower", "--broker=${BROKER_URL}", "--port=5555"]
     ports:
       - 5557:5555

src/test_infra.py

@@ -0,0 +1,36 @@
+import asyncpg
+import aioredis
+from loguru import logger
+from app.core.config import settings
+
+
+async def test_postgres_connection(database_url: str):
+    try:
+        conn = await asyncpg.connect(database_url)
+        logger.info("Successfully connected to PostgreSQL!")
+        await conn.close()
+    except Exception as e:
+        logger.error(f"Failed to connect to PostgreSQL: {e}")
+
+
+async def test_redis_connection(redis_url: str):
+    try:
+        redis = await aioredis.from_url(redis_url)
+        await redis.ping()  # Send a ping to check connection
+        logger.info("Successfully connected to Redis!")
+        await redis.close()
+    except Exception as e:
+        logger.error(f"Failed to connect to Redis: {e}")
+
+
+async def main():
+    logger.info(f"Settings: {settings.dict()}")
+
+    await test_postgres_connection(settings.database_url)
+    await test_redis_connection(settings.redis_url)
+
+
+if __name__ == "__main__":
+    import asyncio
+
+    asyncio.run(main())