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
761 lines
26 KiB
Python
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"
|