@research-copilot/plugin 1.1.15 → 1.1.16

package/dist/hooks/scripts/__tests__/test_research_copilot_guard_main_session.py

@@ -0,0 +1,150 @@
+import json
+import sys
+from pathlib import Path
+
+sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
+
+
+def _write_transcript(path: Path, entries: list[dict]) -> None:
+    path.write_text("\n".join(json.dumps(e) for e in entries) + "\n", encoding="utf-8")
+
+
+# ---- M1 delegation gate (main session) ----
+
+def test_m1_blocks_main_session_experiment_bash(tmp_path):
+    from research_copilot_guard import check_m1_delegation
+    msg = check_m1_delegation("Bash", {"command": "python train.py --epochs 3"})
+    assert msg is not None and "delegate" in msg.lower()
+
+
+def test_m1_allows_read_only_bash(tmp_path):
+    from research_copilot_guard import check_m1_delegation
+    assert check_m1_delegation("Bash", {"command": "cat results.txt"}) is None
+
+
+def test_m1_blocks_main_session_research_mcp(tmp_path):
+    from research_copilot_guard import check_m1_delegation
+    msg = check_m1_delegation("mcp__arxiv-search__search_arxiv", {"query": "ssms"})
+    assert msg is not None and "delegate" in msg.lower()
+
+
+def test_m1_blocks_write_to_tex(tmp_path):
+    from research_copilot_guard import check_m1_delegation
+    msg = check_m1_delegation("Write", {"file_path": "sections/intro.tex"})
+    assert msg is not None
+
+
+def test_m1_blocks_write_to_ideas(tmp_path):
+    from research_copilot_guard import check_m1_delegation
+    msg = check_m1_delegation("Edit", {"file_path": ".copilot/ideas.md"})
+    assert msg is not None
+
+
+def test_m1_allows_write_to_state_md(tmp_path):
+    """Conductor owns state.md / decisions.md — never denied."""
+    from research_copilot_guard import check_m1_delegation
+    assert check_m1_delegation("Write", {"file_path": ".copilot/state.md"}) is None
+    assert check_m1_delegation("Edit", {"file_path": ".copilot/decisions.md"}) is None
+
+
+def test_m1_allows_unrelated_write(tmp_path):
+    from research_copilot_guard import check_m1_delegation
+    assert check_m1_delegation("Write", {"file_path": "notes/scratch.md"}) is None
+
+
+def test_m1_allows_lookalike_sections_dir(tmp_path):
+    """A dir merely containing 'sections' as a substring is not delegated."""
+    from research_copilot_guard import check_m1_delegation
+    assert check_m1_delegation("Write", {"file_path": "my-sections/notes.md"}) is None
+    assert check_m1_delegation("Write", {"file_path": "docs/subsections/a.tex"}) is None
+
+
+def test_m1_allows_lookalike_references_bib(tmp_path):
+    """'references.bib' must match by segment, not substring."""
+    from research_copilot_guard import check_m1_delegation
+    assert check_m1_delegation("Write", {"file_path": "old_references.bib"}) is None
+
+
+def test_m1_blocks_powershell_experiment(tmp_path):
+    from research_copilot_guard import check_m1_delegation
+    msg = check_m1_delegation("PowerShell", {"command": "python train.py"})
+    assert msg is not None and "delegate" in msg.lower()
+
+
+def test_m1_allows_powershell_read_only(tmp_path):
+    from research_copilot_guard import check_m1_delegation
+    assert check_m1_delegation("PowerShell", {"command": "Get-Content results.txt"}) is None
+
+
+# ---- M2 task-list gate (main session) ----
+
+def test_m2_blocks_dispatch_without_taskcreate(tmp_path):
+    from research_copilot_guard import check_m2_task_list
+    t = tmp_path / "s.jsonl"
+    _write_transcript(t, [{"type": "tool_use", "name": "Read",
+                           "input": {"file_path": ".copilot/state.md"}}])
+    msg = check_m2_task_list("Agent", {"subagent_type": "copilot-literature"}, str(t))
+    assert msg is not None and "taskcreate" in msg.lower()
+
+
+def test_m2_allows_dispatch_with_taskcreate(tmp_path):
+    from research_copilot_guard import check_m2_task_list
+    t = tmp_path / "s.jsonl"
+    _write_transcript(t, [{"type": "tool_use", "name": "TaskCreate",
+                           "input": {"subject": "S1"}}])
+    assert check_m2_task_list("Agent", {"subagent_type": "copilot-literature"}, str(t)) is None
+
+
+def test_m2_skips_non_copilot_dispatch(tmp_path):
+    from research_copilot_guard import check_m2_task_list
+    t = tmp_path / "s.jsonl"
+    _write_transcript(t, [])
+    assert check_m2_task_list("Agent", {"subagent_type": "general-purpose"}, str(t)) is None
+
+
+def test_m2_fail_open_no_transcript(tmp_path):
+    """No transcript_path => cannot inspect => fail-open (allow)."""
+    from research_copilot_guard import check_m2_task_list
+    assert check_m2_task_list("Agent", {"subagent_type": "copilot-writer"}, "") is None
+
+
+# ---- main() integration: attribution via agent_id ----
+
+def _run_main(monkeypatch, capsys, payload: dict) -> dict:
+    monkeypatch.setattr("sys.stdin", type("S", (), {"read": lambda self: json.dumps(payload)})())
+    import importlib, research_copilot_guard
+    importlib.reload(research_copilot_guard)
+    research_copilot_guard.main()
+    return json.loads(capsys.readouterr().out)
+
+
+def test_main_polices_main_session_train(tmp_path, monkeypatch, capsys):
+    """No agent_id => main session => Bash train.py denied."""
+    out = _run_main(monkeypatch, capsys, {
+        "tool_name": "Bash", "tool_input": {"command": "python train.py"},
+        "transcript_path": str(tmp_path / "x.jsonl"),
+    })
+    assert out["hookSpecificOutput"]["permissionDecision"] == "deny"
+
+
+def test_main_exempts_copilot_subagent_train(tmp_path, monkeypatch, capsys):
+    """agent_id present + copilot-experiment => exempt => allowed."""
+    out = _run_main(monkeypatch, capsys, {
+        "tool_name": "Bash", "tool_input": {"command": "python train.py"},
+        "transcript_path": str(tmp_path / "x.jsonl"),
+        "agent_id": "sa_01", "agent_type": "copilot-experiment",
+    })
+    assert out["hookSpecificOutput"]["permissionDecision"] == "allow"
+
+
+def test_main_fails_open_on_internal_exception(monkeypatch, capsys):
+    """An exception inside the decision path must yield allow, never crash."""
+    import importlib, research_copilot_guard
+    importlib.reload(research_copilot_guard)
+    monkeypatch.setattr(research_copilot_guard, "_decide",
+                        lambda payload: (_ for _ in ()).throw(RuntimeError("boom")))
+    monkeypatch.setattr("sys.stdin",
+                        type("S", (), {"read": lambda self: json.dumps({"tool_name": "Bash"})})())
+    research_copilot_guard.main()
+    out = json.loads(capsys.readouterr().out)
+    assert out["hookSpecificOutput"]["permissionDecision"] == "allow"

package/dist/hooks/scripts/__tests__/test_session_start_memory_injector.py

@@ -0,0 +1,66 @@
+import sys
+from pathlib import Path
+
+sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
+
+
+def test_extract_handoff_block_returns_block_text_when_present():
+    from session_start_memory_injector import extract_handoff_block
+    text = (
+        "# heading\n"
+        "some content\n"
+        "\n"
+        "## __HANDOFF__\n"
+        "- last_updated: 2026-05-23T00:00:00Z\n"
+        "- written_by: research-copilot\n"
+        "- key_facts:\n"
+        "  - locked baseline = Foo\n"
+        "- next_owner: copilot-ideation\n"
+    )
+    block = extract_handoff_block(text)
+    assert block is not None
+    assert "last_updated" in block
+    assert "key_facts" in block
+    assert "locked baseline = Foo" in block
+
+
+def test_extract_handoff_block_returns_none_when_absent():
+    from session_start_memory_injector import extract_handoff_block
+    assert extract_handoff_block("just a body\nno trailer\n") is None
+
+
+def test_extract_last_n_lines_returns_tail():
+    from session_start_memory_injector import extract_last_n_lines
+    text = "\n".join(f"line {i}" for i in range(1, 31)) + "\n"
+    result = extract_last_n_lines(text, n=5)
+    assert result == "line 26\nline 27\nline 28\nline 29\nline 30"
+
+
+def test_main_prints_summary_when_copilot_dir_has_handoff_blocks(tmp_path, monkeypatch, capsys):
+    copilot = tmp_path / ".copilot"
+    copilot.mkdir()
+    (copilot / "state.md").write_text(
+        "# state\n\n## __HANDOFF__\n"
+        "- last_updated: 2026-05-23T00:00:00Z\n"
+        "- written_by: research-copilot\n"
+        "- key_facts:\n"
+        "  - stage cursor at S2\n"
+        "- next_owner: copilot-ideation\n",
+        encoding="utf-8",
+    )
+    monkeypatch.chdir(tmp_path)
+    from session_start_memory_injector import main
+    rc = main()
+    assert rc == 0
+    captured = capsys.readouterr().out
+    assert "[memory-injector]" in captured
+    assert "stage cursor at S2" in captured
+
+
+def test_main_skips_when_no_copilot_dir(tmp_path, monkeypatch, capsys):
+    monkeypatch.chdir(tmp_path)
+    from session_start_memory_injector import main
+    rc = main()
+    assert rc == 0
+    captured = capsys.readouterr().out
+    assert "not initialized" in captured.lower() or "skipping" in captured.lower()

package/dist/hooks/scripts/__tests__/test_user_prompt_dispatch_reminder.py

@@ -0,0 +1,37 @@
+import sys
+from pathlib import Path
+
+sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
+
+
+class _StringIO:
+    def __init__(self, s): self._s = s
+    def read(self): return self._s
+
+
+def test_no_suppression_for_status_query(tmp_path, monkeypatch, capsys):
+    """Standing orders fire even on 'what's next' / 下一步 (no suppression)."""
+    monkeypatch.chdir(tmp_path)
+    monkeypatch.setattr("sys.stdin", _StringIO("下一步"))
+    from user_prompt_dispatch_reminder import main
+    assert main() == 0
+    assert "conductor" in capsys.readouterr().out.lower()
+
+
+def test_no_suppression_for_slash_or_at(tmp_path, monkeypatch, capsys):
+    monkeypatch.chdir(tmp_path)
+    monkeypatch.setattr("sys.stdin", _StringIO("/loop 1m check"))
+    from user_prompt_dispatch_reminder import main
+    assert main() == 0
+    assert capsys.readouterr().out.strip() != ""
+
+
+def test_main_respects_disabled_flag(tmp_path, monkeypatch, capsys):
+    (tmp_path / ".copilot").mkdir()
+    (tmp_path / ".copilot" / "dispatch-reminder.disabled").write_text("", encoding="utf-8")
+    monkeypatch.chdir(tmp_path)
+    monkeypatch.setattr("sys.stdin", _StringIO("anything at all"))
+    from user_prompt_dispatch_reminder import main
+    rc = main()
+    assert rc == 0
+    assert capsys.readouterr().out == ""