From b537be3008fbf137445489345d21cf397578427d Mon Sep 17 00:00:00 2001 From: nesquena-hermes Date: Thu, 16 Jul 2026 19:36:45 +0000 Subject: [PATCH] test(core): accept new limit kwarg in get_state_db_session_messages stub Co-authored-by: rh-id --- tests/test_core_data_loss_cases.py | 2 +- tests/test_session_lineage_full_transcript.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_core_data_loss_cases.py b/tests/test_core_data_loss_cases.py index 4bc487d5..5af2b308 100644 --- a/tests/test_core_data_loss_cases.py +++ b/tests/test_core_data_loss_cases.py @@ -499,7 +499,7 @@ def test_core_a_route_full_session_load_does_not_resurrect_deleted_turns(tmp_pat truncation_boundary=101.0, ) s.save() - routes.get_state_db_session_messages = lambda sid, profile=None: list(state) + routes.get_state_db_session_messages = lambda sid, profile=None, since_timestamp=None, include_inactive=False, limit=None: list(state) routes._session_visible_to_active_profile = lambda profile, handler: True routes._clear_stale_stream_state = lambda s: None routes._resolve_effective_session_model_for_display = lambda s: getattr(s, "model", None) diff --git a/tests/test_session_lineage_full_transcript.py b/tests/test_session_lineage_full_transcript.py index c83a0af2..f9e54e40 100644 --- a/tests/test_session_lineage_full_transcript.py +++ b/tests/test_session_lineage_full_transcript.py @@ -226,7 +226,7 @@ def test_webui_continuation_session_opens_with_snapshot_parent_messages(monkeypa monkeypatch.setattr(routes, "_clear_stale_stream_state", lambda s: None) monkeypatch.setattr(routes, "_lookup_cli_session_metadata", lambda sid: {}) monkeypatch.setattr(routes, "_is_messaging_session_record", lambda s: False) - monkeypatch.setattr(routes, "get_state_db_session_messages", lambda sid, profile=None: []) + monkeypatch.setattr(routes, "get_state_db_session_messages", lambda sid, profile=None, since_timestamp=None, include_inactive=False, limit=None: []) monkeypatch.setattr(routes.Session, "load", lambda sid: parent if sid == "parent-webui" else None) monkeypatch.setattr(routes, "_resolve_effective_session_model_for_display", lambda s: getattr(s, "model", None)) monkeypatch.setattr(routes, "_resolve_effective_session_model_provider_for_display", lambda s: None)