Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

tests/gateway/test_10710_auto_reset_evicts_cached_agent.py

@@ -0,0 +1,121 @@
+"""Regression test for #10710 — stale context summary leak after auto-reset.
+
+The gateway agent cache is keyed on the stable chat ``session_key``, which does
+NOT change when a session is auto-reset (daily schedule / idle timeout /
+suspended). So unless the cached agent is explicitly evicted on auto-reset, the
+NEXT message reuses the old ``AIAgent`` instance — carrying its
+``context_compressor._previous_summary`` — and prior-conversation content leaks
+into the new session's compaction summaries.
+
+Manual ``/reset`` and the compression-exhausted path (#9893) already evict the
+cached agent. This pins the matching eviction onto the auto-reset cleanup block
+in ``_handle_message_with_agent``.
+
+These are AST invariants — load-bearing pins that fail if the eviction is
+removed from the cleanup block (mirrors
+test_48031_model_switch_after_auto_reset.py's approach).
+"""
+from __future__ import annotations
+
+import ast
+import inspect
+
+from gateway import run as gateway_run
+
+
+def _calls(node: ast.AST) -> set[str]:
+    """Method-call attribute names invoked anywhere under ``node``."""
+    return {
+        n.func.attr
+        for n in ast.walk(node)
+        if isinstance(n, ast.Call) and isinstance(n.func, ast.Attribute)
+    }
+
+
+def _assigns_false(node: ast.AST, attr: str) -> bool:
+    """True if ``node`` contains an assignment ``<something>.<attr> = False``."""
+    for sub in ast.walk(node):
+        if isinstance(sub, ast.Assign):
+            for tgt in sub.targets:
+                if (
+                    isinstance(tgt, ast.Attribute)
+                    and tgt.attr == attr
+                    and isinstance(sub.value, ast.Constant)
+                    and sub.value.value is False
+                ):
+                    return True
+    return False
+
+
+def test_auto_reset_cleanup_evicts_cached_agent():
+    """The auto-reset cleanup block in gateway/run.py must call
+    ``_evict_cached_agent`` so the fresh session does not reuse the previous
+    conversation's cached agent (and its leaked
+    ``context_compressor._previous_summary``) — the cache is keyed on the
+    stable ``session_key`` (#10710)."""
+    tree = ast.parse(inspect.getsource(gateway_run))
+
+    # Fingerprint the cleanup branch: the `if <was_auto_reset>:` block that
+    # clears the conversation scope via the funnel (post-#64934 refactor:
+    # one _clear_conversation_scope call replaced the inline pops) and
+    # consumes the flag by setting was_auto_reset = False. The eviction must
+    # live in that same block — the funnel deliberately does NOT evict the
+    # agent cache (it has its own resource-cleanup path).
+    found = False
+    for node in ast.walk(tree):
+        if not isinstance(node, ast.If):
+            continue
+        calls = _calls(node)
+        if (
+            "_clear_conversation_scope" in calls
+            and _assigns_false(node, "was_auto_reset")
+        ):
+            assert "_evict_cached_agent" in calls, (
+                "gateway/run.py auto-reset cleanup block must call "
+                "`_evict_cached_agent(session_key)` so the auto-reset session "
+                "does not reuse the previous cached agent and leak its "
+                "context_compressor._previous_summary into new compaction "
+                "summaries (#10710)."
+            )
+            found = True
+            break
+    assert found, (
+        "could not locate the auto-reset transient-state cleanup block in "
+        "gateway/run.py (fingerprint: _clear_conversation_scope + "
+        "was_auto_reset = False)."
+    )
+
+
+def test_evict_cached_agent_method_exists():
+    """The eviction helper the cleanup relies on must exist on the runner."""
+    assert hasattr(gateway_run.GatewayRunner, "_evict_cached_agent"), (
+        "GatewayRunner._evict_cached_agent is the helper the auto-reset "
+        "cleanup depends on (#10710)."
+    )
+
+
+def _references_name(node: ast.AST, literal: str) -> bool:
+    """True if a string constant equal to ``literal`` appears anywhere under ``node``."""
+    return any(
+        isinstance(n, ast.Constant) and n.value == literal for n in ast.walk(node)
+    )
+
+
+def test_auto_reset_cleanup_clears_last_resolved_model():
+    """Regression test for #58403.
+
+    Auto-reset is a full conversation boundary and now routes through the
+    `_clear_conversation_scope` funnel. The funnel must clear
+    `_last_resolved_model` — without it, the fresh auto-reset session could
+    serve a model cached before the reset on a transient config-cache miss.
+    Behavioral: exercises the real funnel instead of pinning source shape.
+    """
+    runner = object.__new__(gateway_run.GatewayRunner)
+    key = "agent:main:telegram:dm:58403"
+    runner._last_resolved_model = {key: "stale/model", "other": "keep/me"}
+    runner._clear_conversation_scope(key, reason="auto_reset")
+    assert key not in runner._last_resolved_model, (
+        "the conversation-boundary funnel must pop the session's entry from "
+        "`_last_resolved_model` (#58403) — auto-reset routes through it"
+    )
+    assert runner._last_resolved_model.get("other") == "keep/me"