fithia2/tests/unit/test_oracle_client.py

"""Unit tests for Oracle client using pytest-httpx."""

import json
from pathlib import Path

import pytest
from pytest_httpx import HTTPXMock

FIXTURES_DIR = Path(__file__).parent.parent / "fixtures"


def load_fixture(name: str) -> dict:
    return json.loads((FIXTURES_DIR / name).read_text())


@pytest.mark.asyncio
async def test_search_filings(httpx_mock: HTTPXMock):
    from libs.oracle_client.client import OracleClient
    from libs.oracle_client.filings import FilingsService

    data = load_fixture("filing_search.json")
    httpx_mock.add_response(json=data, url="http://oracle:18001/api/v1/filings/search/AAPL")

    async with OracleClient("http://oracle:18001") as client:
        svc = FilingsService(client)
        result = await svc.search_filings("AAPL")

    assert result.ticker == "AAPL"
    assert len(result.filings) == 2
    assert result.filings[0].form_type == "8-K"


@pytest.mark.asyncio
async def test_get_exhibit(httpx_mock: HTTPXMock):
    from libs.oracle_client.client import OracleClient
    from libs.oracle_client.filings import FilingsService

    data = load_fixture("exhibit_content.json")
    httpx_mock.add_response(json=data)

    async with OracleClient("http://oracle:18001") as client:
        svc = FilingsService(client)
        result = await svc.get_exhibit("0000320193-26-000001")

    assert result.accession_no == "0000320193-26-000001"
    assert len(result.content) > 0


@pytest.mark.asyncio
async def test_get_daily_bars(httpx_mock: HTTPXMock):
    from libs.oracle_client.client import OracleClient
    from libs.oracle_client.price import PriceService

    data = load_fixture("price_data.json")
    httpx_mock.add_response(json=data)

    async with OracleClient("http://oracle:18001") as client:
        svc = PriceService(client)
        result = await svc.get_daily_bars("AAPL")

    assert result.ticker == "AAPL"
    assert len(result.bars) > 0
    assert result.bars[0].close > 0


@pytest.mark.asyncio
async def test_not_found_raises_oracle_not_found(httpx_mock: HTTPXMock):
    from libs.oracle_client.client import OracleClient
    from libs.oracle_client.exceptions import OracleNotFoundError
    from libs.oracle_client.filings import FilingsService

    httpx_mock.add_response(status_code=404)

    async with OracleClient("http://oracle:18001") as client:
        svc = FilingsService(client)
        with pytest.raises(OracleNotFoundError):
            await svc.get_exhibit("NONEXISTENT")


@pytest.mark.asyncio
async def test_server_error_raises_oracle_server_error(httpx_mock: HTTPXMock):
    from libs.oracle_client.client import OracleClient
    from libs.oracle_client.exceptions import OracleServerError
    from libs.oracle_client.filings import FilingsService

    # Must provide one response per retry attempt (3 total)
    httpx_mock.add_response(status_code=500)
    httpx_mock.add_response(status_code=500)
    httpx_mock.add_response(status_code=500)

    async with OracleClient("http://oracle:18001") as client:
        svc = FilingsService(client)
        with pytest.raises(OracleServerError):
            await svc.get_exhibit("ACC123")


@pytest.mark.asyncio
async def test_get_retries_on_transient_error_then_succeeds(httpx_mock: HTTPXMock):
    from libs.oracle_client.client import OracleClient
    from libs.oracle_client.filings import FilingsService

    data = load_fixture("exhibit_content.json")
    httpx_mock.add_response(status_code=500)  # attempt 1 fails
    httpx_mock.add_response(status_code=500)  # attempt 2 fails
    httpx_mock.add_response(json=data)  # attempt 3 succeeds

    async with OracleClient("http://oracle:18001") as client:
        svc = FilingsService(client)
        result = await svc.get_exhibit("0000320193-26-000001")

    assert result.accession_no == "0000320193-26-000001"


@pytest.mark.asyncio
async def test_get_short_volume(httpx_mock: HTTPXMock):
    from libs.oracle_client.client import OracleClient
    from libs.oracle_client.finra import FinraService

    data = load_fixture("short_volume.json")
    httpx_mock.add_response(json=data)

    async with OracleClient("http://oracle:18001") as client:
        svc = FinraService(client)
        result = await svc.get_short_volume("AAPL")

    assert result.symbol == "AAPL"
    assert len(result.data) == 3


@pytest.mark.asyncio
async def test_get_financial_data(httpx_mock: HTTPXMock):
    from libs.oracle_client.client import OracleClient
    from libs.oracle_client.financial import FinancialService

    data = load_fixture("financial_data.json")
    httpx_mock.add_response(json=data)

    async with OracleClient("http://oracle:18001") as client:
        svc = FinancialService(client)
        result = await svc.get_financial_data("AAPL")

    assert result.ticker == "AAPL"
    assert len(result.periods) == 2


@pytest.mark.asyncio
async def test_connection_error_raises_oracle_connection_error(httpx_mock: HTTPXMock):
    """ConnectError 시 OracleConnectionError가 발생한다 (3회 재시도 후)."""
    import httpx

    from libs.oracle_client.client import OracleClient
    from libs.oracle_client.exceptions import OracleConnectionError

    # 3번 모두 ConnectError (max_attempts=3)
    httpx_mock.add_exception(httpx.ConnectError("connection refused"))
    httpx_mock.add_exception(httpx.ConnectError("connection refused"))
    httpx_mock.add_exception(httpx.ConnectError("connection refused"))

    async with OracleClient("http://oracle:18001") as client:
        with pytest.raises(OracleConnectionError):
            await client.get("/health")


@pytest.mark.asyncio
async def test_timeout_raises_oracle_timeout_error(httpx_mock: HTTPXMock):
    """ReadTimeout 시 OracleTimeoutError가 발생한다 (3회 재시도 후)."""
    import httpx

    from libs.oracle_client.client import OracleClient
    from libs.oracle_client.exceptions import OracleTimeoutError

    httpx_mock.add_exception(httpx.ReadTimeout("read timeout"))
    httpx_mock.add_exception(httpx.ReadTimeout("read timeout"))
    httpx_mock.add_exception(httpx.ReadTimeout("read timeout"))

    async with OracleClient("http://oracle:18001") as client:
        with pytest.raises(OracleTimeoutError):
            await client.get("/health")


@pytest.mark.asyncio
async def test_client_without_context_manager_raises():
    """context manager 없이 get() 호출 시 RuntimeError가 발생한다."""
    from libs.oracle_client.client import OracleClient

    client = OracleClient("http://oracle:18001")
    with pytest.raises(RuntimeError, match="async context manager"):
        await client.get("/health")