import pytest
from sqlalchemy import select
from app.models import (
    Setting, User, ProviderKey, HubKey, ModelAlias, FallbackChain,
    RoutingRule, CustomProvider, ModelPreset, ABTest, RequestTransformer,
    ResponseFilter, Webhook, InjectionLog, ProviderUptime, RequestLog,
    ConversationHistory, RagDocument, ExactCache, SemanticCache,
    ModelCatalog, BatchJob, HubKeyModelLimit, RateLimitLog,
    RealtimeSession, AgentConversation, Base,
)
from tests._helpers import db_session


class TestModels:
    @pytest.mark.asyncio
    async def test_create_setting(self, db_session):
        s = Setting(key="test_key", value="test_value")
        db_session.add(s)
        await db_session.commit()
        result = await db_session.execute(select(Setting).where(Setting.key == "test_key"))
        row = result.scalar_one()
        assert row.value == "test_value"

    @pytest.mark.asyncio
    async def test_create_user(self, db_session):
        u = User(username="testuser", password_hash="hash123", role="admin")
        db_session.add(u)
        await db_session.commit()
        result = await db_session.execute(select(User).where(User.username == "testuser"))
        row = result.scalar_one()
        assert row.role == "admin"

    @pytest.mark.asyncio
    async def test_create_provider_key(self, db_session):
        pk = ProviderKey(provider="openai", name="test-key", api_key="sk-xxx")
        db_session.add(pk)
        await db_session.commit()
        result = await db_session.execute(select(ProviderKey).where(ProviderKey.name == "test-key"))
        row = result.scalar_one()
        assert row.provider == "openai"
        assert row.is_active == 1

    @pytest.mark.asyncio
    async def test_create_hub_key(self, db_session):
        hk = HubKey(name="test-hub", key_hash="abc123", key_prefix="nxs-test")
        db_session.add(hk)
        await db_session.commit()
        result = await db_session.execute(select(HubKey).where(HubKey.name == "test-hub"))
        row = result.scalar_one()
        assert row.key_prefix == "nxs-test"

    @pytest.mark.asyncio
    async def test_all_tables_registered(self):
        tables = Base.metadata.tables
        assert "settings" in tables
        assert "users" in tables
        assert "provider_keys" in tables
        assert "hub_keys" in tables
        assert "model_aliases" in tables
        assert "fallback_chains" in tables
        assert "routing_rules" in tables
        assert "custom_providers" in tables
        assert "model_presets" in tables
        assert "ab_tests" in tables
        assert "request_transformers" in tables
        assert "response_filters" in tables
        assert "webhooks" in tables
        assert "injection_logs" in tables
        assert "provider_uptime" in tables
        assert "request_logs" in tables
        assert "conversation_history" in tables
        assert "rag_documents" in tables
        assert "exact_cache" in tables
        assert "semantic_cache" in tables
        assert "model_catalog" in tables
        assert "batch_jobs" in tables
        assert "hub_key_model_limits" in tables
        assert "rate_limit_log" in tables
        assert "realtime_sessions" in tables
        assert "agent_conversations" in tables
        assert len(tables) == 27