lukas/uis-cloud-computing
1
0
Fork 0
mirror of https://github.com/dat515-2025/Group-8.git synced 2026-03-22 15:12:08 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(tests): fixed test runtime errors regarding database connection

Browse Source
This commit is contained in:
ribardej
2025-11-11 14:50:43 +01:00
parent f58083870f
commit 1da927dc07
3 changed files with 36 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
7project/backend/app/core/db.py
View File

@@ -1,27 +1,35 @@
import os
from urllib.parse import urlparse
from sqlalchemy.ext.asyncio import create_async_engine, async_sessionmaker
from app.core.base import Base
# Determine DATABASE_URL with sensible defaults for local testing
DATABASE_URL = os.getenv("DATABASE_URL")
if not DATABASE_URL:
mariadb_host = os.getenv("MARIADB_HOST", "localhost")
mariadb_host = os.getenv("MARIADB_HOST")
mariadb_port = os.getenv("MARIADB_PORT", "3306")
mariadb_db = os.getenv("MARIADB_DB", "group_project")
mariadb_user = os.getenv("MARIADB_USER", "root")
mariadb_password = os.getenv("MARIADB_PASSWORD", "strongpassword")
mariadb_db = os.getenv("MARIADB_DB")
mariadb_user = os.getenv("MARIADB_USER")
mariadb_password = os.getenv("MARIADB_PASSWORD")
if mariadb_host and mariadb_db and mariadb_user and mariadb_password:
DATABASE_URL = f"mysql+asyncmy://{mariadb_user}:{mariadb_password}@{mariadb_host}:{mariadb_port}/{mariadb_db}"
else:
raise Exception("Only MariaDB is supported. Please set the DATABASE_URL environment variable.")
# Default to local SQLite for tests/development when nothing is configured
DATABASE_URL = os.getenv("SQLITE_URL", "sqlite+aiosqlite:///./test.db")
# Load all models to register them
from app.models.user import User
from app.models.transaction import Transaction
from app.models.categories import Category
host_env = os.getenv("MARIADB_HOST", "localhost")
ssl_enabled = host_env not in {"localhost", "127.0.0.1"}
connect_args = {"ssl": {"ssl": True}} if ssl_enabled else {}
# Configure connect args based on backend
parsed = urlparse(DATABASE_URL)
scheme = parsed.scheme
connect_args = {}
if scheme.startswith("mysql"):
host_env = os.getenv("MARIADB_HOST", parsed.hostname or "localhost")
ssl_enabled = host_env not in {"localhost", "127.0.0.1"}
connect_args = {"ssl": {"ssl": True}} if ssl_enabled else {}
engine = create_async_engine(
DATABASE_URL,