diff --git a/workspace/tests/test_mcp_memory.py b/workspace/tests/test_mcp_memory.py index 117e5417..d2a7ac35 100644 --- a/workspace/tests/test_mcp_memory.py +++ b/workspace/tests/test_mcp_memory.py @@ -63,7 +63,7 @@ async def test_commit_memory_success(monkeypatch): mcp = _load_mcp() client = FakeClient() - monkeypatch.setattr("a2a_tools.httpx.AsyncClient", lambda **kw: client) + monkeypatch.setattr("a2a_tools_memory.httpx.AsyncClient", lambda **kw: client) result = await mcp.handle_tool_call("commit_memory", { "content": "Architecture decision: use Go for backend", @@ -92,7 +92,7 @@ async def test_commit_memory_default_scope(monkeypatch): mcp = _load_mcp() client = FakeClient() - monkeypatch.setattr("a2a_tools.httpx.AsyncClient", lambda **kw: client) + monkeypatch.setattr("a2a_tools_memory.httpx.AsyncClient", lambda **kw: client) result = await mcp.handle_tool_call("commit_memory", { "content": "Some note", @@ -108,7 +108,7 @@ async def test_recall_memory_success(monkeypatch): mcp = _load_mcp() client = FakeClient() - monkeypatch.setattr("a2a_tools.httpx.AsyncClient", lambda **kw: client) + monkeypatch.setattr("a2a_tools_memory.httpx.AsyncClient", lambda **kw: client) result = await mcp.handle_tool_call("recall_memory", {"query": "architecture"}) @@ -127,7 +127,7 @@ async def test_recall_memory_empty(monkeypatch): async def get(self, url, params=None, headers=None, **kwargs): return FakeResponse(200, []) - monkeypatch.setattr("a2a_tools.httpx.AsyncClient", lambda **kw: EmptyClient()) + monkeypatch.setattr("a2a_tools_memory.httpx.AsyncClient", lambda **kw: EmptyClient()) result = await mcp.handle_tool_call("recall_memory", {}) assert "No memories found" in result @@ -139,7 +139,7 @@ async def test_recall_memory_with_scope_filter(monkeypatch): mcp = _load_mcp() client = FakeClient() - monkeypatch.setattr("a2a_tools.httpx.AsyncClient", lambda **kw: client) + monkeypatch.setattr("a2a_tools_memory.httpx.AsyncClient", lambda **kw: client) await mcp.handle_tool_call("recall_memory", {"scope": "TEAM"}) diff --git a/workspace/tests/test_secret_redact.py b/workspace/tests/test_secret_redact.py index d0975969..ecc268e8 100644 --- a/workspace/tests/test_secret_redact.py +++ b/workspace/tests/test_secret_redact.py @@ -357,7 +357,7 @@ class TestA2AToolCommitMemoryRedactsSecrets: fake_client.post = _capture - with patch("a2a_tools.httpx.AsyncClient", return_value=fake_client): + with patch("a2a_tools_memory.httpx.AsyncClient", return_value=fake_client): await a2a_tools.tool_commit_memory(content_with_secret) stored = captured.get("content", "") @@ -385,7 +385,7 @@ class TestA2AToolCommitMemoryRedactsSecrets: fake_client.post = _capture - with patch("a2a_tools.httpx.AsyncClient", return_value=fake_client): + with patch("a2a_tools_memory.httpx.AsyncClient", return_value=fake_client): await a2a_tools.tool_commit_memory(f"key={key}") stored = captured.get("content", "")