memory graph and services

tests/test_memory_graph_contracts.py

@@ -0,0 +1,164 @@
+import json
+from pathlib import Path
+
+import yaml
+from click.testing import CliRunner
+
+from markitect_tool.cli import main
+from markitect_tool.memory import (
+    compile_memory_graph_selection_to_context_package,
+    load_memory_graph_file,
+    load_memory_graph_selection_file,
+    load_memory_profile_file,
+    plan_memory_profile,
+    validate_memory_graph,
+    validate_memory_profile,
+)
+
+
+def test_memory_profile_validation_and_plan(tmp_path: Path):
+    profile_path = _write_profile(tmp_path)
+    profile = load_memory_profile_file(profile_path)
+
+    result = validate_memory_profile(profile, path=profile_path)
+    plan = plan_memory_profile(profile)
+
+    assert result.valid
+    assert result.metadata["memory_kinds"] == ["reasoning", "knowledge", "package"]
+    assert plan["runtime_boundary"]["services_launched_by_markitect_tool"] is False
+    assert plan["activation"]["max_items"] == 4
+
+
+def test_memory_graph_validation_and_context_package_compile(tmp_path: Path):
+    profile_path = _write_profile(tmp_path)
+    graph_path = _write_graph(tmp_path)
+    selection_path = _write_selection(tmp_path)
+
+    graph = load_memory_graph_file(graph_path)
+    profile = load_memory_profile_file(profile_path)
+    selection = load_memory_graph_selection_file(selection_path)
+    result = validate_memory_graph(graph, path=graph_path)
+    package = compile_memory_graph_selection_to_context_package(graph, selection, profile=profile)
+
+    assert result.valid
+    assert package.title == "Decision package"
+    assert len(package.items) == 1
+    assert package.items[0].source.path == "decision.md"
+    assert package.metadata["memory_graph"]["selected_edges"] == ["edge.supports"]
+    assert package.retrieval_recipes[0].kind == "memory-graph-selection"
+
+
+def test_mkt_memory_blueprint_and_graph_cli(tmp_path: Path):
+    _write_profile(tmp_path)
+    _write_graph(tmp_path)
+    selection_path = _write_selection(tmp_path)
+    runner = CliRunner()
+
+    blueprint = runner.invoke(
+        main,
+        ["memory", "blueprint", "validate", str(tmp_path / "profile.yaml"), "--format", "json"],
+    )
+    plan = runner.invoke(
+        main,
+        ["memory", "blueprint", "plan", str(tmp_path / "profile.yaml"), "--format", "json"],
+    )
+    graph = runner.invoke(
+        main,
+        ["memory", "graph", "validate", str(tmp_path / "graph.yaml"), "--format", "json"],
+    )
+    packed = runner.invoke(main, ["memory", "graph", "pack", str(selection_path), "--format", "json"])
+
+    assert blueprint.exit_code == 0, blueprint.output
+    assert json.loads(blueprint.output)["valid"] is True
+    assert plan.exit_code == 0, plan.output
+    assert json.loads(plan.output)["profile"]["id"] == "test-profile"
+    assert graph.exit_code == 0, graph.output
+    assert json.loads(graph.output)["metadata"]["nodes"] == 1
+    assert packed.exit_code == 0, packed.output
+    package = json.loads(packed.output)
+    assert package["title"] == "Decision package"
+    assert package["metadata"]["memory_profile"]["id"] == "test-profile"
+
+
+def _write_profile(root: Path) -> Path:
+    path = root / "profile.yaml"
+    path.write_text(
+        yaml.safe_dump(
+            {
+                "schema_version": "markitect.memory.profile.v1",
+                "id": "test-profile",
+                "title": "Test profile",
+                "intent": "Exercise local memory profile contracts.",
+                "memory_kinds": ["reasoning", "knowledge", "package"],
+                "stores": {
+                    "reasoning": "event-log",
+                    "knowledge": "graph-store",
+                    "package": "context-registry",
+                },
+                "activation": {"max_items": 4, "max_tokens": 1000},
+            },
+            sort_keys=False,
+        ),
+        encoding="utf-8",
+    )
+    return path
+
+
+def _write_graph(root: Path) -> Path:
+    path = root / "graph.yaml"
+    path.write_text(
+        yaml.safe_dump(
+            {
+                "schema_version": "markitect.memory.graph.v1",
+                "id": "test-graph",
+                "title": "Test graph",
+                "intent": "Compile one decision into context.",
+                "nodes": [
+                    {
+                        "id": "decision.keep-contract-local",
+                        "kind": "decision",
+                        "text": "Keep memory graph contracts in markitect-tool.",
+                        "source_spans": [
+                            {
+                                "path": "decision.md",
+                                "unit_kind": "section",
+                                "selector": "sections[heading=Decision]",
+                                "engine": "selector",
+                            }
+                        ],
+                    }
+                ],
+                "edges": [
+                    {
+                        "id": "edge.supports",
+                        "kind": "supports",
+                        "source": "decision.keep-contract-local",
+                        "target": "decision.keep-contract-local",
+                    }
+                ],
+            },
+            sort_keys=False,
+        ),
+        encoding="utf-8",
+    )
+    return path
+
+
+def _write_selection(root: Path) -> Path:
+    path = root / "selection.yaml"
+    path.write_text(
+        yaml.safe_dump(
+            {
+                "schema_version": "markitect.memory.selection.v1",
+                "graph": "graph.yaml",
+                "profile": "profile.yaml",
+                "title": "Decision package",
+                "intent": "Activate graph decisions.",
+                "node_ids": ["decision.keep-contract-local"],
+                "budget": {"max_items": 2},
+            },
+            sort_keys=False,
+        ),
+        encoding="utf-8",
+    )
+    return path