from investibook.api.chat import _latest_chat_context_window


def test_latest_chat_context_window_skips_auxiliary_usage_rows():
    rows = [
        {
            "channel": "evidence-postpass",
            "model_tier": "evidence-postpass",
            "context_used": 0,
            "context_size": 0,
        },
        {
            "channel": "chat",
            "model_tier": "standard",
            "context_used": 59222,
            "context_size": 200000,
        },
    ]

    assert _latest_chat_context_window(rows) == (59222, 200000)


def test_latest_chat_context_window_returns_empty_when_no_context_window():
    rows = [
        {
            "channel": "chat",
            "model_tier": "standard",
            "context_used": 0,
            "context_size": 0,
        },
    ]

    assert _latest_chat_context_window(rows) == (0, 0)