import pytest

from haiku.rag.client import HaikuRAG
from haiku.rag.config import AppConfig


async def test_database_not_created_without_create_flag(tmp_path):
    """Test that database is not created without create=True."""
    db_path = tmp_path / "test.lancedb"

    config = AppConfig()

    with pytest.raises(FileNotFoundError, match="Database does not exist"):
        async with HaikuRAG(db_path=db_path, config=config):
            pass


async def test_database_created_with_create_flag(tmp_path):
    """Test that database is created with create=True."""
    db_path = tmp_path / "test.lancedb"

    config = AppConfig()

    async with HaikuRAG(db_path=db_path, config=config, create=True):
        assert db_path.exists()


@pytest.mark.vcr()
async def test_operations_work_after_database_created(tmp_path):
    """Test that operations work after DB is created."""
    db_path = tmp_path / "test.lancedb"

    config = AppConfig()

    # First, create DB with create=True and add document
    async with HaikuRAG(db_path=db_path, config=config, create=True) as client:
        await client.create_document("Test content", uri="test://doc1")

    # Re-open without create flag and verify we can read the document
    async with HaikuRAG(db_path=db_path, config=config) as client:
        docs = await client.list_documents()
        assert len(docs) == 1
        doc = await client.get_document_by_id(docs[0].id)
        assert doc is not None
        assert doc.content == "Test content"