Method test_persistence_roundtrip

tests/test_stats.py:229–266

(self)

        assert rs.summary().total_requests == 0

    def test_persistence_roundtrip(self) -> None:
        storage = InMemoryRouteStatsStorage()
        rs1 = RouteStats(storage=storage)
        rs1.record(RouteRecord(**{
            **_make_record(model="test/model", confidence=0.77).__dict__,
            "transport": "anthropic-messages",
            "cache_mode": "cache_control",
            "cache_family": "anthropic",
            "cache_breakpoints": 2,
            "route_reasoning": "selected for low latency",
            "routing_features_payload": {"step_type": "research"},
            "fallback_chain_payload": [{"model": "fallback/model", "reason": "cost"}],
            "candidate_scores_payload": [{"model": "test/model", "score": 0.91}],
            "status_code": 502,
            "error_code": "connect_error",
            "error_stage": "upstream_request",
            "error_message": "Upstream unreachable",
        }))
        rs1.record(_make_record(model="test/model2", tier="REASONING"))

        rs2 = RouteStats(storage=storage)
        assert rs2.count == 2
        h = rs2.history()
        assert h[0].model == "test/model2"
        assert h[0].tier == "COMPLEX"
        assert h[1].confidence == 0.77
        assert h[1].transport == "anthropic-messages"
        assert h[1].cache_mode == "cache_control"
        assert h[1].cache_family == "anthropic"
        assert h[1].cache_breakpoints == 2
        assert h[1].route_reasoning == "selected for low latency"
        assert h[1].routing_features_payload == {"step_type": "research"}
        assert h[1].fallback_chain_payload == [{"model": "fallback/model", "reason": "cost"}]
        assert h[1].candidate_scores_payload == [{"model": "test/model", "score": 0.91}]
        assert h[1].status_code == 502
        assert h[1].error_code == "connect_error"
        assert h[1].error_stage == "upstream_request"
        assert h[1].error_message == "Upstream unreachable"

    def test_retention_cleanup(self) -> None:
        t = 1_000_000.0