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

tests/agent/test_session_activity.py

@@ -0,0 +1,96 @@
+"""Unit tests for the shared session activity observation contract."""
+
+from types import SimpleNamespace
+
+from agent.session_activity import (
+    ACTIVITY_DESCRIPTION_MAX,
+    ActivityProvenance,
+    bound_activity_description,
+    build_activity_snapshot,
+    normalize_activity_provenance,
+    reset_session_activity_persist_window,
+)
+
+
+def test_bound_activity_description_truncates():
+    long = "x" * (ACTIVITY_DESCRIPTION_MAX + 80)
+    out = bound_activity_description(long)
+    assert len(out) == ACTIVITY_DESCRIPTION_MAX
+    assert out.endswith("…")
+
+
+def test_reset_session_activity_persist_window_clears_rate_limit():
+    agent = SimpleNamespace(_session_activity_last_persist_mono=1234.5)
+    reset_session_activity_persist_window(agent)
+    assert agent._session_activity_last_persist_mono == 0.0
+
+
+def test_reset_session_activity_persist_window_swallows_missing_attr():
+    reset_session_activity_persist_window(object())
+
+
+def test_normalize_activity_provenance_defaults_to_unknown():
+    assert normalize_activity_provenance(None) is ActivityProvenance.UNKNOWN
+    assert normalize_activity_provenance("") is ActivityProvenance.UNKNOWN
+    assert normalize_activity_provenance("not-a-real-source") is ActivityProvenance.UNKNOWN
+    assert normalize_activity_provenance("agent.activity") is ActivityProvenance.UNKNOWN
+    assert (
+        normalize_activity_provenance(ActivityProvenance.AGENT_COMPRESSION)
+        is ActivityProvenance.AGENT_COMPRESSION
+    )
+    assert (
+        normalize_activity_provenance("agent.compression_timeout")
+        is ActivityProvenance.AGENT_COMPRESSION_TIMEOUT
+    )
+
+
+def test_build_activity_snapshot_includes_compat_aliases():
+    snap = build_activity_snapshot(
+        last_activity_at=100.0,
+        last_activity_description="starting API call #1",
+        last_activity_provenance=ActivityProvenance.UNKNOWN,
+        now=110.0,
+        extra={"api_call_count": 1},
+    )
+    assert snap["last_activity_at"] == 100.0
+    assert snap["last_activity_description"] == "starting API call #1"
+    assert snap["last_activity_provenance"] == "unknown"
+    assert snap["seconds_since_activity"] == 10.0
+    assert snap["last_activity_ts"] == 100.0
+    assert snap["last_activity_desc"] == "starting API call #1"
+    assert snap["description"] == "starting API call #1"
+    assert snap["api_call_count"] == 1
+    assert "phase" not in snap
+    assert "last_progress_at" not in snap
+
+
+def test_build_activity_snapshot_maps_missing_provenance_to_unknown():
+    snap = build_activity_snapshot(
+        last_activity_at=1.0,
+        last_activity_description="starting new turn (cached)",
+        last_activity_provenance=None,
+        now=2.0,
+    )
+    assert snap["last_activity_provenance"] == "unknown"
+
+
+def test_build_activity_snapshot_preserves_compression_transition_provenances():
+    """Compaction / timeout / cooldown share the observation source (#72424)."""
+    for provenance, desc in (
+        (ActivityProvenance.AGENT_COMPRESSION, "context compression in progress"),
+        (ActivityProvenance.AGENT_COMPRESSION_TIMEOUT, "context compression timed out"),
+        (
+            ActivityProvenance.AGENT_COMPRESSION_COOLDOWN,
+            "compression blocked (cooldown: 30s remaining)",
+        ),
+    ):
+        snap = build_activity_snapshot(
+            last_activity_at=50.0,
+            last_activity_description=desc,
+            last_activity_provenance=provenance,
+            now=55.0,
+        )
+        assert snap["last_activity_provenance"] == provenance.value
+        assert snap["provenance"] == provenance.value
+        assert snap["last_activity_description"] == desc
+        assert snap["seconds_since_activity"] == 5.0