diff --git a/apps/api/tests/conftest.py b/apps/api/tests/conftest.py index 9c2d53a..9a2af3a 100644 --- a/apps/api/tests/conftest.py +++ b/apps/api/tests/conftest.py @@ -16,7 +16,9 @@ from app.db import get_db_session from app.main import app from app.models import Base -TEST_DATABASE_URL = settings.database_url.replace("/headquarter", "/headquarter_test") +TEST_DATABASE_URL = settings.database_url +if "/headquarter_test" not in TEST_DATABASE_URL: + TEST_DATABASE_URL = TEST_DATABASE_URL.replace("/headquarter", "/headquarter_test") if TEST_DATABASE_URL.startswith("postgresql://"): TEST_DATABASE_URL = TEST_DATABASE_URL.replace("postgresql://", "postgresql+asyncpg://", 1)