File size: 1,018 Bytes
fee2f22
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
"""Database tests for backend initialization."""

from __future__ import annotations

from sqlalchemy import inspect

from backend.database import Base, engine, init_db


def test_database_connection() -> None:
    """Engine can produce a SQLAlchemy connection object."""
    with engine.connect() as connection:
        assert connection is not None


def test_model_creation() -> None:
    """ORM metadata contains required tables."""
    init_db()
    inspector = inspect(engine)
    tables = set(inspector.get_table_names())

    assert "datasets" in tables
    assert "cleaning_jobs" in tables
    assert "audit_logs" in tables
    assert "model_versions" in tables


def test_crud_operations() -> None:
    """At minimum ensure table metadata is bound and creatable."""
    # A lightweight sanity check that table definitions are valid.
    assert Base.metadata.tables["datasets"] is not None
    assert Base.metadata.tables["cleaning_jobs"] is not None
    assert Base.metadata.tables["audit_logs"] is not None