Update database script

Signed-off-by: Vikram Vaswani <112123850+vikram-dagger@users.noreply.github.com>

Author: vikram-dagger

conftest.py

@@ -1,17 +1,19 @@
 
 import pytest
+import os
 from sqlalchemy import create_engine, inspect
 from sqlalchemy.orm import sessionmaker
 from fastapi.testclient import TestClient
 from .main import create_app
 from .dependencies import get_db, database_url
 from .models import Base
 
+TEST_DATABASE_URL = os.getenv("TEST_DATABASE_URL", "postgresql://app_user:secret@db/app_db_test")
 
 @pytest.fixture(scope="session")
 def test_engine():
     """Create test database engine"""
-    test_engine = create_engine(database_url)
+    test_engine = create_engine(TEST_DATABASE_URL)
     Base.metadata.create_all(bind=test_engine)
     yield test_engine
     Base.metadata.drop_all(bind=test_engine)

docker-compose.yml

@@ -7,6 +7,7 @@ services:
       - "8000:8000"
     environment:
       - DATABASE_URL=postgresql://app_user:secret@db/app_db
+      - TEST_DATABASE_URL=postgresql://app_user:secret@db/app_db_test
     depends_on:
       db:
         condition: service_healthy
@@ -18,13 +19,13 @@ services:
     container_name: fastapi_db
     restart: unless-stopped
     environment:
-      POSTGRES_DB: app_db
       POSTGRES_USER: app_user
       POSTGRES_PASSWORD: secret
     ports:
       - "5432:5432"
     volumes:
       - postgres_data:/var/lib/postgresql/data
+      - ./init-dbs.sh:/docker-entrypoint-initdb.d/init-dbs.sh:ro
     healthcheck:
       test: ["CMD-SHELL", "pg_isready -U app_user -d app_db"]
       interval: 10s

init-dbs.sh

@@ -0,0 +1,8 @@
+# init-db.sh
+#!/bin/bash
+set -e
+
+psql -v ON_ERROR_STOP=1 --username "$POSTGRES_USER" <<-EOSQL
+  CREATE DATABASE app_db;
+  CREATE DATABASE app_db_test;
+EOSQL