Files
wehub-resource-sync 94b8d5d118
Docker / docker-build (map[name:amd64 platform:linux/amd64 runs_on:ubuntu-24.04], map[bake_target:runtime-slim name:slim]) (push) Waiting to run
Docker / docker-build (map[name:amd64 platform:linux/amd64 runs_on:ubuntu-24.04], map[bake_target:runtime-slim-nonroot name:slim-nonroot]) (push) Waiting to run
Docker / docker-build (map[name:arm64 platform:linux/arm64 runs_on:ubuntu-24.04-arm], map[bake_target:runtime name:]) (push) Waiting to run
Docker / docker-build (map[name:arm64 platform:linux/arm64 runs_on:ubuntu-24.04-arm], map[bake_target:runtime-code name:code]) (push) Waiting to run
Docker / docker-build (map[name:amd64 platform:linux/amd64 runs_on:ubuntu-24.04], map[bake_target:runtime name:]) (push) Waiting to run
Docker / docker-build (map[name:amd64 platform:linux/amd64 runs_on:ubuntu-24.04], map[bake_target:runtime-code name:code]) (push) Waiting to run
Docker / docker-build (map[name:amd64 platform:linux/amd64 runs_on:ubuntu-24.04], map[bake_target:runtime-code-nonroot name:code-nonroot]) (push) Waiting to run
Docker / docker-build (map[name:amd64 platform:linux/amd64 runs_on:ubuntu-24.04], map[bake_target:runtime-code-slim name:code-slim]) (push) Waiting to run
Docker / docker-build (map[name:amd64 platform:linux/amd64 runs_on:ubuntu-24.04], map[bake_target:runtime-code-slim-nonroot name:code-slim-nonroot]) (push) Waiting to run
Docker / docker-build (map[name:amd64 platform:linux/amd64 runs_on:ubuntu-24.04], map[bake_target:runtime-nonroot name:nonroot]) (push) Waiting to run
Docker / docker-build (map[name:arm64 platform:linux/arm64 runs_on:ubuntu-24.04-arm], map[bake_target:runtime-code-nonroot name:code-nonroot]) (push) Waiting to run
Docker / docker-build (map[name:arm64 platform:linux/arm64 runs_on:ubuntu-24.04-arm], map[bake_target:runtime-code-slim name:code-slim]) (push) Waiting to run
Docker / docker-build (map[name:arm64 platform:linux/arm64 runs_on:ubuntu-24.04-arm], map[bake_target:runtime-slim name:slim]) (push) Waiting to run
Docker / docker-build (map[name:arm64 platform:linux/arm64 runs_on:ubuntu-24.04-arm], map[bake_target:runtime-slim-nonroot name:slim-nonroot]) (push) Waiting to run
Docker / docker-manifest (map[bake_target:runtime name:]) (push) Blocked by required conditions
Docker / docker-manifest (map[bake_target:runtime-code name:code]) (push) Blocked by required conditions
Docker / docker-manifest (map[bake_target:runtime-code-nonroot name:code-nonroot]) (push) Blocked by required conditions
Docker / docker-manifest (map[bake_target:runtime-code-slim name:code-slim]) (push) Blocked by required conditions
Docker / docker-manifest (map[bake_target:runtime-code-slim-nonroot name:code-slim-nonroot]) (push) Blocked by required conditions
Docker / docker-build (map[name:arm64 platform:linux/arm64 runs_on:ubuntu-24.04-arm], map[bake_target:runtime-code-slim-nonroot name:code-slim-nonroot]) (push) Waiting to run
Docker / docker-build (map[name:arm64 platform:linux/arm64 runs_on:ubuntu-24.04-arm], map[bake_target:runtime-nonroot name:nonroot]) (push) Waiting to run
Docker / docker-manifest (map[bake_target:runtime-nonroot name:nonroot]) (push) Blocked by required conditions
Docker / docker-manifest (map[bake_target:runtime-slim name:slim]) (push) Blocked by required conditions
Docker / docker-manifest (map[bake_target:runtime-slim-nonroot name:slim-nonroot]) (push) Blocked by required conditions
Docker / promote-latest (push) Blocked by required conditions
Init Native E2E / init-native (macos-latest, claude) (push) Waiting to run
Init Native E2E / init-native (macos-latest, codex) (push) Waiting to run
Init Native E2E / init-native (macos-latest, copilot) (push) Waiting to run
Install Native E2E / install-native (macos-latest) (push) Waiting to run
Wrap Native E2E / wrap-native (macos-latest) (push) Waiting to run
Security / Dependency audit (pip-audit) (push) Has been cancelled
Security / CodeQL (javascript-typescript) (push) Has been cancelled
Security / CodeQL (python) (push) Has been cancelled
Security / Secret scan (gitleaks) (push) Has been cancelled
rust / test (ubuntu) (push) Has been cancelled
rust / simulator e2e (macos-latest) (push) Has been cancelled
rust / simulator e2e (ubuntu-latest) (push) Has been cancelled
rust / simulator e2e (windows-latest) (push) Has been cancelled
rust / wheels (aarch64-apple-darwin) (push) Has been cancelled
rust / wheels (x86_64-unknown-linux-gnu) (push) Has been cancelled
rust / wheels (x86_64-apple-darwin) (push) Has been cancelled
rust / audit (push) Has been cancelled
rust / parity (nightly, allowed to fail during Phase 0) (push) Has been cancelled
CI / changes (push) Failing after 2s
Deploy Documentation / deploy (push) Failing after 0s
CI / commitlint (push) Has been skipped
Dev Containers / validate (.devcontainer/devcontainer.json, default) (push) Failing after 1s
Dev Containers / validate-worktree (push) Failing after 1s
Dev Containers / validate (.devcontainer/memory-stack/devcontainer.json, memory-stack) (push) Failing after 1s
Wrap Native E2E / wrap-native (ubuntu-latest) (push) Failing after 4s
Deploy Documentation / validate (push) Has been skipped
Init E2E / docker-init-e2e (push) Failing after 1s
Init Native E2E / init-native (ubuntu-latest, claude) (push) Failing after 1s
Init Native E2E / init-native (ubuntu-latest, codex) (push) Failing after 0s
Install Native E2E / install-native (ubuntu-latest) (push) Failing after 1s
Release Please / release-please (push) Failing after 0s
Wrap E2E / docker-wrap-e2e (push) Failing after 1s
Merge Conflicts / merge-conflicts (push) Failing after 2s
OpenCode Plugin / typecheck + build + test (push) Failing after 2s
Init Native E2E / init-native (ubuntu-latest, copilot) (push) Failing after 3s
CI / test (1) (push) Has been cancelled
CI / test (2) (push) Has been cancelled
CI / test (3) (push) Has been cancelled
CI / test (4) (push) Has been cancelled
CI / test-extras (push) Has been cancelled
CI / test-agno (push) Has been cancelled
CI / test-dashboard-ui (push) Has been cancelled
CI / docker-native-e2e (push) Has been cancelled
CI / lint (push) Has been cancelled
CI / build-wheel (push) Has been cancelled
CI / build-wheel-windows (push) Has been cancelled
CI / prefetch-model (push) Has been cancelled
CI / windows-native-wrapper (push) Has been cancelled
CI / build (push) Has been cancelled
CI / workflow-validation (push) Has been cancelled
CI / macos-native-wrapper (push) Has been cancelled
chore: import upstream snapshot with attribution
2026-07-13 12:03:29 +08:00

761 lines
26 KiB
Python

"""Comprehensive tests for the universal memory sync engine.
Tests cover:
- Core sync: import, export, bidirectional
- Idempotency and deduplication
- Fast no-op detection
- Lineage and governance metadata
- Claude Code adapter: read/write frontmatter files
- Codex adapter: read/write AGENTS.md sections
- Cross-agent interop: save in one agent, find in another
"""
from __future__ import annotations
import hashlib
import json
import time
from dataclasses import dataclass, field
from datetime import datetime, timezone
from pathlib import Path
from typing import Any
import pytest
from headroom.memory.sync import (
_build_sync_backend,
sync,
sync_export,
sync_import,
)
from headroom.memory.sync_adapters.claude_code import (
ClaudeCodeAdapter,
_parse_frontmatter,
encode_claude_project_path,
get_claude_memory_dir,
)
from headroom.memory.sync_adapters.codex_agent import CodexAdapter
# ---------------------------------------------------------------------------
# Fake backend for testing (no real DB/embeddings needed)
# ---------------------------------------------------------------------------
@dataclass
class FakeMemory:
id: str = ""
content: str = ""
user_id: str = ""
category: str = ""
importance: float = 0.5
created_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))
metadata: dict[str, Any] = field(default_factory=dict)
class FakeBackend:
"""In-memory backend for testing sync without real DB."""
def __init__(self) -> None:
self._memories: list[FakeMemory] = []
self._next_id = 1
async def get_user_memories(self, user_id: str, limit: int = 500) -> list[FakeMemory]:
return [m for m in self._memories if m.user_id == user_id][:limit]
async def save_memory(
self,
content: str,
user_id: str,
importance: float = 0.5,
metadata: dict[str, Any] | None = None,
**kwargs: Any,
) -> FakeMemory:
mem = FakeMemory(
id=f"mem_{self._next_id:04d}",
content=content,
user_id=user_id,
importance=importance,
metadata=metadata or {},
)
self._next_id += 1
self._memories.append(mem)
return mem
def add_memory(self, content: str, user_id: str = "tcms", **kwargs: Any) -> FakeMemory:
"""Sync helper to pre-populate memories."""
mem = FakeMemory(
id=f"mem_{self._next_id:04d}",
content=content,
user_id=user_id,
metadata=kwargs.get("metadata", {}),
importance=kwargs.get("importance", 0.5),
)
self._next_id += 1
self._memories.append(mem)
return mem
# ---------------------------------------------------------------------------
# Core sync tests
# ---------------------------------------------------------------------------
class TestSyncImport:
"""Test importing from agent files into DB."""
@pytest.fixture
def backend(self):
return FakeBackend()
@pytest.fixture
def claude_dir(self, tmp_path):
d = tmp_path / "memory"
d.mkdir()
return d
def _write_claude_memory(
self, memory_dir: Path, name: str, content: str, **fm_fields: str
) -> None:
slug = name.lower().replace(" ", "_")
fields = {"name": name, "description": content[:80], "type": "project", **fm_fields}
fm_lines = ["---"]
for k, v in fields.items():
fm_lines.append(f"{k}: {v}")
fm_lines.append("---")
(memory_dir / f"{slug}.md").write_text("\n".join(fm_lines) + f"\n\n{content}\n")
@pytest.mark.asyncio
async def test_import_claude_files_to_db(self, backend, claude_dir):
self._write_claude_memory(claude_dir, "Project codename", "The secret name is TC")
self._write_claude_memory(claude_dir, "Dark mode", "User prefers dark mode")
adapter = ClaudeCodeAdapter(claude_dir)
imported = await sync_import(backend, adapter, "tcms")
assert imported == 2
mems = await backend.get_user_memories("tcms")
contents = {m.content for m in mems}
assert "The secret name is TC" in contents
assert "User prefers dark mode" in contents
@pytest.mark.asyncio
async def test_import_skips_existing(self, backend, claude_dir):
"""Memories already in DB are not re-imported."""
backend.add_memory(
"The secret name is TC",
metadata={"content_hash": hashlib.sha256(b"The secret name is TC").hexdigest()[:16]},
)
self._write_claude_memory(claude_dir, "Project codename", "The secret name is TC")
self._write_claude_memory(claude_dir, "New fact", "Something new")
adapter = ClaudeCodeAdapter(claude_dir)
imported = await sync_import(backend, adapter, "tcms")
assert imported == 1 # Only "Something new"
@pytest.mark.asyncio
async def test_import_preserves_lineage(self, backend, claude_dir):
self._write_claude_memory(claude_dir, "Fact", "Important fact")
adapter = ClaudeCodeAdapter(claude_dir)
await sync_import(backend, adapter, "tcms")
mems = await backend.get_user_memories("tcms")
assert len(mems) == 1
assert mems[0].metadata["source_agent"] == "claude"
assert mems[0].metadata["source_file"] == "fact.md"
assert "content_hash" in mems[0].metadata
assert mems[0].metadata["sync_direction"] == "import"
class TestSyncExport:
"""Test exporting from DB to agent files."""
@pytest.fixture
def backend(self):
return FakeBackend()
@pytest.fixture
def claude_dir(self, tmp_path):
d = tmp_path / "memory"
d.mkdir()
return d
@pytest.mark.asyncio
async def test_export_new_memory_to_claude_files(self, backend, claude_dir):
backend.add_memory(
"Project uses Python 3.12",
metadata={
"source_agent": "codex",
"sync_direction": "export", # Not from claude import
},
)
adapter = ClaudeCodeAdapter(claude_dir)
exported = await sync_export(backend, adapter, "tcms")
assert exported == 1
# Check file was created
md_files = list(claude_dir.glob("headroom_*.md"))
assert len(md_files) == 1
content = md_files[0].read_text()
assert "Python 3.12" in content
assert "headroom_id: mem_0001" in content
assert "source_agent: codex" in content
@pytest.mark.asyncio
async def test_export_skips_claude_originated(self, backend, claude_dir):
"""Don't re-export memories that were imported FROM claude (anti-echo)."""
backend.add_memory(
"From claude",
metadata={
"source_agent": "claude",
"sync_direction": "import",
},
)
backend.add_memory(
"From codex",
metadata={
"source_agent": "codex",
},
)
adapter = ClaudeCodeAdapter(claude_dir)
exported = await sync_export(backend, adapter, "tcms")
assert exported == 1 # Only "From codex"
@pytest.mark.asyncio
async def test_export_updates_memory_md_index(self, backend, claude_dir):
# Create an existing MEMORY.md
(claude_dir / "MEMORY.md").write_text("# Memory\n\n## User\n- Some existing entry\n")
backend.add_memory("New fact from codex", metadata={"source_agent": "codex"})
adapter = ClaudeCodeAdapter(claude_dir)
await sync_export(backend, adapter, "tcms")
memory_md = (claude_dir / "MEMORY.md").read_text()
assert "Headroom Shared Memory" in memory_md
assert "New fact from codex" in memory_md
assert "Some existing entry" in memory_md # Preserved
class TestBidirectionalSync:
"""Test full bidirectional sync."""
@pytest.fixture
def backend(self):
return FakeBackend()
@pytest.fixture
def claude_dir(self, tmp_path):
d = tmp_path / "memory"
d.mkdir()
return d
@pytest.fixture
def state_path(self, tmp_path):
return tmp_path / "sync_state.json"
def _write_claude_memory(self, memory_dir: Path, name: str, content: str) -> None:
slug = name.lower().replace(" ", "_")
fm = f"---\nname: {name}\ndescription: {content[:80]}\ntype: project\n---"
(memory_dir / f"{slug}.md").write_text(f"{fm}\n\n{content}\n")
@pytest.mark.asyncio
async def test_bidirectional_sync(self, backend, claude_dir, state_path):
# Claude has a memory file
self._write_claude_memory(claude_dir, "Convention", "Always use ruff for linting")
# DB has a memory from Codex
backend.add_memory("Secret name is TC", metadata={"source_agent": "codex"})
adapter = ClaudeCodeAdapter(claude_dir)
result = await sync(backend, adapter, "tcms", state_path=state_path, force=True)
assert result.imported == 1 # Claude file → DB
assert result.exported == 1 # Codex memory → Claude file
# Verify DB has both
mems = await backend.get_user_memories("tcms")
contents = {m.content for m in mems}
assert "Always use ruff for linting" in contents
assert "Secret name is TC" in contents
# Verify Claude dir has the exported file
all_files = list(claude_dir.glob("headroom_*.md"))
assert len(all_files) >= 1
exported_content = " ".join(f.read_text() for f in all_files)
assert "TC" in exported_content
@pytest.mark.asyncio
async def test_sync_idempotent(self, backend, claude_dir, state_path):
"""Running sync twice produces no duplicates."""
self._write_claude_memory(claude_dir, "Fact", "Python 3.12 is required")
backend.add_memory("Port 8787 is default", metadata={"source_agent": "codex"})
adapter = ClaudeCodeAdapter(claude_dir)
r1 = await sync(backend, adapter, "tcms", state_path=state_path, force=True)
assert r1.imported == 1
assert r1.exported == 1
r2 = await sync(backend, adapter, "tcms", state_path=state_path, force=True)
assert r2.imported == 0 # Already imported
assert r2.exported == 0 # Already exported
# No duplicates in DB
mems = await backend.get_user_memories("tcms")
assert len(mems) == 2
@pytest.mark.asyncio
async def test_fast_noop_when_unchanged(self, backend, claude_dir, state_path):
"""Second sync with no changes completes in < 10ms."""
self._write_claude_memory(claude_dir, "Fact", "Some fact")
adapter = ClaudeCodeAdapter(claude_dir)
# First sync (populates state)
await sync(backend, adapter, "tcms", state_path=state_path, force=True)
# Second sync (should be fast no-op)
start = time.monotonic()
r = await sync(backend, adapter, "tcms", state_path=state_path)
elapsed = (time.monotonic() - start) * 1000
assert r.imported == 0
assert r.exported == 0
assert elapsed < 50 # Generous threshold for CI
class TestLineageAndGovernance:
"""Test metadata tracking for audit and lineage."""
@pytest.fixture
def backend(self):
return FakeBackend()
@pytest.fixture
def claude_dir(self, tmp_path):
d = tmp_path / "memory"
d.mkdir()
return d
@pytest.mark.asyncio
async def test_lineage_tracks_source_agent(self, backend, claude_dir):
fm = "---\nname: test\ndescription: test\ntype: project\n---"
(claude_dir / "test.md").write_text(f"{fm}\n\nClaude discovered this\n")
adapter = ClaudeCodeAdapter(claude_dir)
await sync_import(backend, adapter, "tcms")
mems = await backend.get_user_memories("tcms")
assert mems[0].metadata["source_agent"] == "claude"
@pytest.mark.asyncio
async def test_exported_files_have_headroom_id(self, backend, claude_dir):
backend.add_memory("From codex", metadata={"source_agent": "codex"})
adapter = ClaudeCodeAdapter(claude_dir)
await sync_export(backend, adapter, "tcms")
md_files = list(claude_dir.glob("headroom_*.md"))
assert len(md_files) == 1
content = md_files[0].read_text()
assert "headroom_id:" in content
@pytest.mark.asyncio
async def test_sync_state_records_timestamps(self, backend, claude_dir, tmp_path):
state_path = tmp_path / "state.json"
fm = "---\nname: t\ndescription: t\ntype: project\n---"
(claude_dir / "t.md").write_text(f"{fm}\n\nFact\n")
adapter = ClaudeCodeAdapter(claude_dir)
await sync(backend, adapter, "tcms", state_path=state_path, force=True)
state = json.loads(state_path.read_text())
key = "claude:tcms"
assert key in state
assert "last_sync" in state[key]
assert "agent_fingerprint" in state[key]
assert "db_fingerprint" in state[key]
# ---------------------------------------------------------------------------
# Claude Code adapter tests
# ---------------------------------------------------------------------------
class TestClaudeCodeAdapter:
"""Test Claude Code adapter read/write."""
@pytest.fixture
def memory_dir(self, tmp_path):
d = tmp_path / "memory"
d.mkdir()
return d
def test_parse_frontmatter(self):
content = "---\nname: Test\ntype: project\n---\n\nBody content here."
fm, body = _parse_frontmatter(content)
assert fm["name"] == "Test"
assert fm["type"] == "project"
assert body == "Body content here."
def test_parse_frontmatter_no_frontmatter(self):
content = "Just plain content."
fm, body = _parse_frontmatter(content)
assert fm == {}
assert body == "Just plain content."
def test_encode_claude_project_path_windows_user_with_dot(self):
assert encode_claude_project_path(r"C:\Users\john.doe\work") == "-C-Users-john.doe-work"
def test_get_claude_memory_dir_uses_windows_safe_project_encoding(self):
memory_dir = get_claude_memory_dir(Path(r"C:\Users\john.doe\work"))
rendered = str(memory_dir)
assert "-C-Users-john.doe-work" in rendered
assert "john-doe" not in rendered
assert rendered.endswith("memory")
@pytest.mark.asyncio
async def test_read_memories_skips_memory_md(self, memory_dir):
(memory_dir / "MEMORY.md").write_text("# Index\n- entry")
(memory_dir / "fact.md").write_text(
"---\nname: Fact\ntype: project\n---\n\nImportant fact."
)
adapter = ClaudeCodeAdapter(memory_dir)
mems = await adapter.read_memories()
assert len(mems) == 1
assert mems[0].content == "Important fact."
assert mems[0].source_file == "fact.md"
@pytest.mark.asyncio
async def test_write_creates_valid_md(self, memory_dir):
adapter = ClaudeCodeAdapter(memory_dir)
written = await adapter.write_memories(
[
{
"content": "Project uses FastAPI",
"category": "architecture",
"headroom_id": "mem_001",
"source_agent": "codex",
"content_hash": "abc123",
}
]
)
assert written == 1
files = list(memory_dir.glob("headroom_*.md"))
assert len(files) == 1
content = files[0].read_text()
fm, body = _parse_frontmatter(content)
assert fm["type"] == "architecture"
assert fm["headroom_id"] == "mem_001"
assert fm["source_agent"] == "codex"
assert "FastAPI" in body
@pytest.mark.asyncio
async def test_write_distinct_memories_sharing_first_line_do_not_clobber(self, memory_dir):
"""Two different memories that share a first line must not overwrite one
another. The filename slug is derived from the first line only, so before
the fix the second write clobbered the first (data loss)."""
adapter = ClaudeCodeAdapter(memory_dir)
written = await adapter.write_memories(
[
{
"content": "# Project conventions\nUse tabs for indentation.",
"headroom_id": "mem_a",
"content_hash": "hash_a",
},
{
"content": "# Project conventions\nDeploy on Fridays only.",
"headroom_id": "mem_b",
"content_hash": "hash_b",
},
]
)
assert written == 2
files = sorted(memory_dir.glob("headroom_*.md"))
# Both memories must survive on disk (distinct files).
assert len(files) == 2
bodies = "\n".join(f.read_text() for f in files)
assert "tabs for indentation" in bodies
assert "Deploy on Fridays only" in bodies
@pytest.mark.asyncio
async def test_write_same_memory_updates_in_place(self, memory_dir):
"""An update to the *same* memory (matching headroom_id) rewrites the
original slug file rather than spawning a disambiguated duplicate."""
adapter = ClaudeCodeAdapter(memory_dir)
await adapter.write_memories(
[{"content": "# Note\nfirst version", "headroom_id": "mem_x", "content_hash": "h1"}]
)
await adapter.write_memories(
[{"content": "# Note\nsecond version", "headroom_id": "mem_x", "content_hash": "h2"}]
)
files = list(memory_dir.glob("headroom_*.md"))
assert len(files) == 1
assert "second version" in files[0].read_text()
def test_fingerprint_changes_on_modification(self, memory_dir):
(memory_dir / "test.md").write_text("content 1")
adapter = ClaudeCodeAdapter(memory_dir)
fp1 = adapter.fingerprint()
(memory_dir / "test.md").write_text("content 2")
fp2 = adapter.fingerprint()
assert fp1 != fp2
def test_fingerprint_stable_when_unchanged(self, memory_dir):
(memory_dir / "test.md").write_text("stable content")
adapter = ClaudeCodeAdapter(memory_dir)
assert adapter.fingerprint() == adapter.fingerprint()
def test_fingerprint_empty_dir(self, tmp_path):
empty = tmp_path / "empty"
empty.mkdir()
adapter = ClaudeCodeAdapter(empty)
assert adapter.fingerprint() == "empty"
# ---------------------------------------------------------------------------
# Codex adapter tests
# ---------------------------------------------------------------------------
class TestCodexAdapter:
"""Test Codex AGENTS.md adapter."""
@pytest.fixture
def agents_md(self, tmp_path):
return tmp_path / "AGENTS.md"
@pytest.mark.asyncio
async def test_read_from_agents_md(self, agents_md):
agents_md.write_text(
"# Instructions\n\n"
"<!-- headroom:memory:start -->\n"
"## Headroom Shared Memory\n\n"
"- Secret name is TC\n"
"- Uses Python 3.12\n"
"<!-- headroom:memory:end -->\n"
)
adapter = CodexAdapter(agents_md)
mems = await adapter.read_memories()
assert len(mems) == 2
assert mems[0].content == "Secret name is TC"
assert mems[1].content == "Uses Python 3.12"
@pytest.mark.asyncio
async def test_write_to_agents_md(self, agents_md):
agents_md.write_text("# Existing instructions\n")
adapter = CodexAdapter(agents_md)
written = await adapter.write_memories(
[
{"content": "Port 8787 is default"},
{"content": "Uses ruff for linting"},
]
)
assert written == 2
content = agents_md.read_text()
assert "headroom:memory:start" in content
assert "Port 8787 is default" in content
assert "Uses ruff for linting" in content
assert "Existing instructions" in content # Preserved
@pytest.mark.asyncio
async def test_write_merges_into_existing_section(self, agents_md):
"""Additive: an existing managed fact is preserved when a new one is
written. ``sync_export`` hands the adapter only the delta, so a
replace-the-whole-section write would erase prior memories."""
agents_md.write_text(
"# Instructions\n\n"
"<!-- headroom:memory:start -->\n"
"## Headroom Shared Memory\n\n- old fact\n"
"<!-- headroom:memory:end -->\n"
)
adapter = CodexAdapter(agents_md)
await adapter.write_memories([{"content": "new fact"}])
content = agents_md.read_text()
assert "new fact" in content
assert "old fact" in content # preserved, not clobbered
@pytest.mark.asyncio
async def test_write_preserves_existing_fact_with_literal_backslashes(self, agents_md):
agents_md.write_text(
"# Instructions\n\n"
"<!-- headroom:memory:start -->\n"
"## Headroom Shared Memory\n\n- old fact\n"
"<!-- headroom:memory:end -->\n"
)
adapter = CodexAdapter(agents_md)
await adapter.write_memories([{"content": r"Use C:\Users\john.doe\repo and literal \u"}])
content = agents_md.read_text()
# Backslashes / \u land literally (function replacement, not a template).
assert r"C:\Users\john.doe\repo" in content
assert r"literal \u" in content
assert "old fact" in content # preserved
@pytest.mark.asyncio
async def test_write_accumulates_across_syncs(self, agents_md):
"""Regression: exporting deltas across successive syncs must accumulate,
not thrash between disjoint subsets."""
adapter = CodexAdapter(agents_md)
await adapter.write_memories([{"content": "fact A"}, {"content": "fact B"}])
# Second sync only sees the new memory as a delta.
added = await adapter.write_memories([{"content": "fact C"}])
content = agents_md.read_text()
assert "fact A" in content
assert "fact B" in content
assert "fact C" in content
assert added == 1
# Re-writing an already-present fact adds nothing and keeps the rest.
again = await adapter.write_memories([{"content": "fact A"}])
assert again == 0
assert (await adapter.read_memories()).__len__() == 3
@pytest.mark.asyncio
async def test_read_empty_agents_md(self, agents_md):
agents_md.write_text("# No memory section\n")
adapter = CodexAdapter(agents_md)
mems = await adapter.read_memories()
assert mems == []
@pytest.mark.asyncio
async def test_read_nonexistent_file(self, tmp_path):
adapter = CodexAdapter(tmp_path / "nonexistent.md")
mems = await adapter.read_memories()
assert mems == []
# ---------------------------------------------------------------------------
# Cross-agent integration tests
# ---------------------------------------------------------------------------
class TestCrossAgentInterop:
"""Test that memories flow between agents via sync."""
@pytest.fixture
def backend(self):
return FakeBackend()
@pytest.fixture
def claude_dir(self, tmp_path):
d = tmp_path / "claude_memory"
d.mkdir()
return d
@pytest.fixture
def agents_md(self, tmp_path):
return tmp_path / "AGENTS.md"
@pytest.fixture
def state_path(self, tmp_path):
return tmp_path / "state.json"
@pytest.mark.asyncio
async def test_codex_saves_claude_finds(self, backend, claude_dir, state_path):
"""Memory saved via Codex MCP appears in Claude's files after sync."""
# Simulate Codex saving via MCP (directly to backend)
backend.add_memory(
"Secret name is TC",
metadata={"source_agent": "codex", "content_hash": "x"},
)
# Sync to Claude
adapter = ClaudeCodeAdapter(claude_dir)
result = await sync(backend, adapter, "tcms", state_path=state_path, force=True)
assert result.exported == 1
# Claude's memory dir should have the file
files = list(claude_dir.glob("headroom_*.md"))
assert len(files) == 1
assert "TC" in files[0].read_text()
@pytest.mark.asyncio
async def test_claude_saves_codex_finds(self, backend, claude_dir, agents_md, state_path):
"""Memory saved in Claude's files appears in Codex AGENTS.md after sync."""
# Claude has a memory
fm = "---\nname: Linting\ndescription: use ruff\ntype: project\n---"
(claude_dir / "linting.md").write_text(f"{fm}\n\nAlways use ruff for linting\n")
# Sync Claude → DB
claude_adapter = ClaudeCodeAdapter(claude_dir)
await sync(backend, claude_adapter, "tcms", state_path=state_path, force=True)
# Sync DB → Codex AGENTS.md
codex_adapter = CodexAdapter(agents_md)
result = await sync(backend, codex_adapter, "tcms", state_path=state_path, force=True)
assert result.exported >= 1
assert "ruff" in agents_md.read_text()
@pytest.mark.asyncio
async def test_full_round_trip(self, backend, claude_dir, agents_md, state_path):
"""Full round trip: Claude → DB → Codex, Codex → DB → Claude."""
# Claude has a memory
fm = "---\nname: Framework\ntype: project\n---"
(claude_dir / "framework.md").write_text(f"{fm}\n\nUses FastAPI\n")
# Codex has a memory (in DB via MCP)
backend.add_memory("Port is 8787", metadata={"source_agent": "codex"})
# Sync both adapters
claude_adapter = ClaudeCodeAdapter(claude_dir)
codex_adapter = CodexAdapter(agents_md)
await sync(backend, claude_adapter, "tcms", state_path=state_path, force=True)
await sync(backend, codex_adapter, "tcms", state_path=state_path, force=True)
# DB has both memories
mems = await backend.get_user_memories("tcms")
contents = {m.content for m in mems}
assert "Uses FastAPI" in contents
assert "Port is 8787" in contents
# Claude files have Codex's memory
all_claude = " ".join(f.read_text() for f in claude_dir.glob("headroom_*.md"))
assert "8787" in all_claude
# AGENTS.md has both (from DB)
agents_content = agents_md.read_text()
assert "FastAPI" in agents_content or "8787" in agents_content
def test_sync_backend_uses_onnx_embedder(tmp_path):
"""#1092: the sync subprocess must pick the torch-free ONNX embedder.
Defaulting to the LOCAL (sentence-transformers) embedder makes
`wrap --memory` crash with an ImportError on the proxy extras. The backend
must match the proxy MCP server, which uses ONNX.
"""
backend = _build_sync_backend(str(tmp_path / "memory.db"))
assert backend._config.embedder_backend == "onnx"