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

tests/run_agent/test_notice_spine.py

@@ -0,0 +1,139 @@
+"""Regression tests for the notice-spine (AgentNotice + emitter callbacks).
+
+Covers:
+  A. _emit_notice / _emit_notice_clear emitter behaviour (bare AIAgent via
+     object.__new__ — same pattern as test_steer.py and test_file_mutation_verifier.py).
+  B. Constructor / init_agent signature threading.
+  C. TUI _agent_cbs notice binding — mirrors the status_callback tests already
+     in tests/test_tui_gateway_server.py.
+"""
+from __future__ import annotations
+
+import inspect
+from unittest.mock import patch
+
+import pytest
+
+from agent.credits_tracker import AgentNotice
+from run_agent import AIAgent
+
+
+# ── A. Emitter behaviour ─────────────────────────────────────────────────────
+
+
+def _bare_agent() -> AIAgent:
+    """Build an AIAgent without running __init__ (no heavy init required).
+
+    Only the two callback slots used by _emit_notice / _emit_notice_clear are
+    installed — mirrors the pattern in test_steer.py.
+    """
+    agent = object.__new__(AIAgent)
+    agent.notice_callback = None
+    agent.notice_clear_callback = None
+    return agent
+
+
+class TestEmitNotice:
+    def test_emit_notice_calls_callback_with_exact_notice(self):
+        agent = _bare_agent()
+        received = []
+        notice = AgentNotice(
+            text="credits 90% used",
+            level="warn",
+            kind="sticky",
+            ttl_ms=None,
+            key="credits.warn90",
+            id="n1",
+        )
+        agent.notice_callback = received.append
+        agent._emit_notice(notice)
+        assert received == [notice]
+
+
+
+
+
+
+
+# ── B. Constructor / init_agent signature threading ─────────────────────────
+
+
+class TestSignatureThreading:
+    def test_agent_init_exposes_notice_callback(self):
+        sig = inspect.signature(AIAgent.__init__)
+        assert "notice_callback" in sig.parameters
+
+
+
+
+
+# ── C. TUI _agent_cbs binding ────────────────────────────────────────────────
+
+
+class TestAgentCbsNoticeBinding:
+    """Mirror test_status_callback_emits_kind_and_text from test_tui_gateway_server.py."""
+
+    def test_notice_callback_emits_notification_show(self):
+        from tui_gateway import server
+
+        with patch("tui_gateway.server._emit") as mock_emit:
+            cbs = server._agent_cbs("sid123")
+            notice = AgentNotice(
+                text="credits 90% used",
+                level="warn",
+                kind="sticky",
+                ttl_ms=None,
+                key="credits.warn90",
+                id="n1",
+            )
+            cbs["notice_callback"](notice)
+
+        mock_emit.assert_called_once_with(
+            "notification.show",
+            "sid123",
+            {
+                "text": "credits 90% used",
+                "level": "warn",
+                "kind": "sticky",
+                "ttl_ms": None,
+                "key": "credits.warn90",
+                "id": "n1",
+            },
+        )
+
+    def test_notice_callback_payload_is_full_snake_case_dict(self):
+        """All six snake_case fields must be present in the payload — no extras,
+        no camelCase variants."""
+        from tui_gateway import server
+
+        captured = []
+        with patch("tui_gateway.server._emit", side_effect=lambda *a: captured.append(a)):
+            cbs = server._agent_cbs("sid123")
+            cbs["notice_callback"](
+                AgentNotice(
+                    text="credits 90% used",
+                    level="warn",
+                    kind="sticky",
+                    ttl_ms=None,
+                    key="credits.warn90",
+                    id="n1",
+                )
+            )
+
+        assert len(captured) == 1
+        _event_type, _sid, payload = captured[0]
+        assert set(payload.keys()) == {"text", "level", "kind", "ttl_ms", "key", "id"}
+
+
+
+    def test_notice_clear_callback_event_type_is_notification_clear(self):
+        from tui_gateway import server
+
+        captured = []
+        with patch("tui_gateway.server._emit", side_effect=lambda *a: captured.append(a)):
+            cbs = server._agent_cbs("sid123")
+            cbs["notice_clear_callback"]("some.key")
+
+        assert captured[0][0] == "notification.clear"
+        assert captured[0][1] == "sid123"
+        assert captured[0][2] == {"key": "some.key"}