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(setup_db):
    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