"""
Pytest fixtures and test configuration.
"""

from datetime import date
from decimal import Decimal

import pytest
from sqlalchemy import create_engine
from sqlalchemy.orm import Session, sessionmaker

from pote.db import Base
from pote.db.models import Official, Price, Security, Trade


@pytest.fixture(scope="function")
def test_db_session() -> Session:
    """
    Create an in-memory SQLite database for testing.
    Each test gets a fresh database.
    """
    engine = create_engine("sqlite:///:memory:", echo=False)
    Base.metadata.create_all(engine)

    TestSessionLocal = sessionmaker(bind=engine)
    session = TestSessionLocal()

    yield session

    session.close()
    engine.dispose()


@pytest.fixture
def sample_official(test_db_session: Session) -> Official:
    """Create a sample official for testing."""
    official = Official(
        name="Jane Doe",
        chamber="Senate",
        party="Independent",
        state="CA",
        bioguide_id="D000123",
    )
    test_db_session.add(official)
    test_db_session.commit()
    test_db_session.refresh(official)
    return official


@pytest.fixture
def sample_security(test_db_session: Session) -> Security:
    """Create a sample security for testing."""
    security = Security(
        ticker="AAPL",
        name="Apple Inc.",
        exchange="NASDAQ",
        sector="Technology",
        asset_type="stock",
    )
    test_db_session.add(security)
    test_db_session.commit()
    test_db_session.refresh(security)
    return security


@pytest.fixture
def sample_trade(
    test_db_session: Session, sample_official: Official, sample_security: Security
) -> Trade:
    """Create a sample trade for testing."""
    trade = Trade(
        official_id=sample_official.id,
        security_id=sample_security.id,
        source="test",
        external_id="test-001",
        transaction_date=date(2024, 1, 15),
        filing_date=date(2024, 2, 1),
        side="buy",
        value_min=Decimal("15000.00"),
        value_max=Decimal("50000.00"),
        currency="USD",
    )
    test_db_session.add(trade)
    test_db_session.commit()
    test_db_session.refresh(trade)
    return trade


@pytest.fixture
def sample_price(test_db_session: Session, sample_security: Security) -> Price:
    """Create a sample price record for testing."""
    price = Price(
        security_id=sample_security.id,
        date=date(2024, 1, 15),
        open=Decimal("180.50"),
        high=Decimal("182.75"),
        low=Decimal("179.00"),
        close=Decimal("181.25"),
        volume=50000000,
        source="yfinance",
    )
    test_db_session.add(price)
    test_db_session.commit()
    test_db_session.refresh(price)
    return price