diff --git a/backend/tests/test_models.py b/backend/tests/test_models.py new file mode 100644 index 0000000..1ce3c02 --- /dev/null +++ b/backend/tests/test_models.py @@ -0,0 +1,29 @@ +import pytest +from sqlalchemy import inspect + +from app.database import Base +from app.models import ( + Conversation, + DigitalEmployee, + Document, + KnowledgeBase, + Tenant, + TenantConfig, +) +from tests.conftest import test_engine + + +@pytest.mark.asyncio +async def test_all_tables_created(): + expected_tables = { + "tenants", + "tenant_configs", + "digital_employees", + "conversations", + "messages", + "knowledge_bases", + "documents", + } + async with test_engine.begin() as conn: + tables = await conn.run_sync(lambda sync_conn: inspect(sync_conn).get_table_names()) + assert set(tables) == expected_tables