Add Memory Gateway agent plugin

plugins/memory-gateway-agent/tests/test_client.py

@@ -0,0 +1,105 @@
+from __future__ import annotations
+
+import io
+import json
+import urllib.error
+
+from memory_gateway_plugin.client import MemoryGatewayClient
+from memory_gateway_plugin.config import PluginConfig
+
+
+class FakeResponse:
+    status = 200
+
+    def __init__(self, payload):
+        self.payload = payload
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, *args):
+        return False
+
+    def read(self):
+        return json.dumps(self.payload).encode("utf-8")
+
+
+def test_client_search_success(monkeypatch):
+    seen = {}
+
+    def fake_urlopen(request, timeout):
+        seen["url"] = request.full_url
+        seen["timeout"] = timeout
+        return FakeResponse({"results": [], "total": 0})
+
+    monkeypatch.setattr("urllib.request.urlopen", fake_urlopen)
+    client = MemoryGatewayClient(PluginConfig(gateway_url="http://gateway", timeout=7))
+    result = client.search_memory({"user_id": "u", "query": "demo"})
+
+    assert result["ok"] is True
+    assert result["endpoint"] == "/v1/memory/search"
+    assert seen["url"] == "http://gateway/v1/memory/search"
+    assert seen["timeout"] == 7
+
+
+def test_client_network_error(monkeypatch):
+    def fake_urlopen(request, timeout):
+        raise urllib.error.URLError("connection refused")
+
+    monkeypatch.setattr("urllib.request.urlopen", fake_urlopen)
+    client = MemoryGatewayClient(PluginConfig(gateway_url="http://gateway"))
+    result = client.search_memory({"user_id": "u", "query": "demo"})
+
+    assert result["ok"] is False
+    assert result["status_code"] is None
+    assert "connection refused" in result["error"]
+
+
+def test_commit_session_calls_correct_endpoint(monkeypatch):
+    seen = {}
+
+    def fake_urlopen(request, timeout):
+        seen["url"] = request.full_url
+        return FakeResponse({"session_id": "sess_1"})
+
+    monkeypatch.setattr("urllib.request.urlopen", fake_urlopen)
+    client = MemoryGatewayClient(PluginConfig(gateway_url="http://gateway"))
+    result = client.commit_session("sess_1", {"user_id": "u", "session_id": "sess_1"})
+
+    assert result["ok"] is True
+    assert seen["url"] == "http://gateway/v1/sessions/sess_1/commit"
+
+
+def test_feedback_calls_correct_endpoint(monkeypatch):
+    seen = {}
+
+    def fake_urlopen(request, timeout):
+        seen["url"] = request.full_url
+        return FakeResponse({"status": "ok"})
+
+    monkeypatch.setattr("urllib.request.urlopen", fake_urlopen)
+    client = MemoryGatewayClient(PluginConfig(gateway_url="http://gateway"))
+    result = client.send_feedback("mem_1", {"user_id": "u", "feedback": "incorrect"})
+
+    assert result["ok"] is True
+    assert seen["url"] == "http://gateway/v1/memory/mem_1/feedback"
+
+
+def test_client_http_error(monkeypatch):
+    def fake_urlopen(request, timeout):
+        raise urllib.error.HTTPError(
+            url=request.full_url,
+            code=401,
+            msg="unauthorized",
+            hdrs=None,
+            fp=io.BytesIO(b'{"detail":"Invalid or missing API key"}'),
+        )
+
+    monkeypatch.setattr("urllib.request.urlopen", fake_urlopen)
+    client = MemoryGatewayClient(PluginConfig(gateway_url="http://gateway"))
+    result = client.search_memory({"user_id": "u", "query": "demo"})
+
+    assert result["ok"] is False
+    assert result["status_code"] == 401
+    assert result["endpoint"] == "/v1/memory/search"
+