70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
import os
|
|
|
|
import pytest
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.utils.database import get_db, get_db_credentials
|
|
from tests.utils.db_mocks import mock_env, mock_ssm, session_mock
|
|
|
|
|
|
def test_get_db_credentials_env(mock_env):
|
|
"""Test getting DB credentials from environment variables"""
|
|
conn_str = get_db_credentials()
|
|
assert conn_str == "postgresql://testuser:testpass@localhost/testdb"
|
|
|
|
|
|
def test_get_db_credentials_ssm(mock_ssm):
|
|
"""Test getting DB credentials from SSM"""
|
|
os.environ.pop("MOCK_AUTH", None)
|
|
conn_str = get_db_credentials()
|
|
expected_conn = "postgresql://mocked_value:mocked_value@mocked_value/mocked_value"
|
|
assert expected_conn in conn_str
|
|
mock_ssm.get_parameter.assert_called()
|
|
|
|
|
|
def test_get_db_credentials_ssm_exception(mock_ssm):
|
|
"""Test SSM credential fetching failure raises RuntimeError"""
|
|
os.environ.pop("MOCK_AUTH", None)
|
|
mock_ssm.get_parameter.side_effect = Exception("SSM timeout")
|
|
|
|
with pytest.raises(RuntimeError) as excinfo:
|
|
get_db_credentials()
|
|
|
|
assert "Failed to fetch DB credentials from SSM: SSM timeout" in str(excinfo.value)
|
|
|
|
|
|
def test_session_creation():
|
|
"""Test database session creation"""
|
|
session = session_mock()
|
|
assert isinstance(session, Session)
|
|
session.close()
|
|
|
|
|
|
def test_get_db_generator():
|
|
"""Test get_db dependency generator"""
|
|
db_gen = get_db()
|
|
db = next(db_gen)
|
|
assert isinstance(db, Session)
|
|
try:
|
|
next(db_gen) # Should raise StopIteration
|
|
except StopIteration:
|
|
pass
|
|
|
|
|
|
def test_init_db(mocker, mock_env):
|
|
"""Test database initialization creates tables"""
|
|
mock_create_all = mocker.patch("app.models.Base.metadata.create_all")
|
|
|
|
# Mock get_db_credentials to return SQLite test connection
|
|
mocker.patch(
|
|
"app.utils.database.get_db_credentials",
|
|
return_value="sqlite:///:memory:",
|
|
)
|
|
|
|
from app.utils.database import engine, init_db
|
|
|
|
init_db()
|
|
|
|
# Verify create_all was called with the engine
|
|
mock_create_all.assert_called_once_with(bind=engine)
|