30 lines
680 B
Python
30 lines
680 B
Python
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
|