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

tests/run_agent/test_session_source.py

@@ -0,0 +1,31 @@
+import pytest
+
+from gateway.session_context import _UNSET, _VAR_MAP, clear_session_vars, set_session_vars
+from run_agent import _session_source_for_agent
+
+
+@pytest.fixture(autouse=True)
+def _reset_contextvars():
+    for var in _VAR_MAP.values():
+        var.set(_UNSET)
+    yield
+    for var in _VAR_MAP.values():
+        var.set(_UNSET)
+
+
+def test_session_source_context_overrides_platform(monkeypatch):
+    monkeypatch.delenv("HERMES_SESSION_SOURCE", raising=False)
+
+    tokens = set_session_vars(source="tool")
+    try:
+        assert _session_source_for_agent("tui") == "tool"
+    finally:
+        clear_session_vars(tokens)
+
+
+def test_session_source_falls_back_to_platform(monkeypatch):
+    monkeypatch.delenv("HERMES_SESSION_SOURCE", raising=False)
+
+    assert _session_source_for_agent("tui") == "tui"
+
+