codebuff-agent 2.1.74__py3-none-any.whl

jupyter_ext/tests/test_handlers.py

@@ -0,0 +1,58 @@
+import os
+
+from jupyter_ext.handlers import (
+    DEFAULT_EXECUTE_COMMAND_TIMEOUT_MS,
+    _append_stream_output,
+    _resolve_path_in_workspace,
+    _resolve_stream_timeout_ms,
+    _resolve_timeout_ms,
+)
+
+
+def test_resolve_path_strips_duplicate_cwd_prefix() -> None:
+    workspace_root = "/workspace"
+    requested_cwd = os.path.join(workspace_root, "extensions", "jupyter")
+    path = os.path.join("extensions", "jupyter", "cal.py")
+    resolved = _resolve_path_in_workspace(path, workspace_root, requested_cwd)
+    assert resolved == os.path.join(requested_cwd, "cal.py")
+
+
+def test_resolve_path_relative_to_cwd() -> None:
+    workspace_root = "/workspace"
+    requested_cwd = os.path.join(workspace_root, "extensions", "jupyter")
+    resolved = _resolve_path_in_workspace("cal.py", workspace_root, requested_cwd)
+    assert resolved == os.path.join(requested_cwd, "cal.py")
+
+
+def test_resolve_timeout_ms_uses_default_on_invalid() -> None:
+    assert _resolve_timeout_ms(None) == DEFAULT_EXECUTE_COMMAND_TIMEOUT_MS
+    assert _resolve_timeout_ms("not-a-number") == DEFAULT_EXECUTE_COMMAND_TIMEOUT_MS
+    assert _resolve_timeout_ms(-1) == DEFAULT_EXECUTE_COMMAND_TIMEOUT_MS
+
+
+def test_resolve_timeout_ms_accepts_value() -> None:
+    assert _resolve_timeout_ms(120000) == 120000
+
+
+def test_resolve_stream_timeout_ms_uses_default_on_invalid() -> None:
+    assert _resolve_stream_timeout_ms(None) == DEFAULT_EXECUTE_COMMAND_TIMEOUT_MS
+    assert (
+        _resolve_stream_timeout_ms("not-a-number") == DEFAULT_EXECUTE_COMMAND_TIMEOUT_MS
+    )
+
+
+def test_resolve_stream_timeout_ms_disables_on_non_positive() -> None:
+    assert _resolve_stream_timeout_ms(0) is None
+    assert _resolve_stream_timeout_ms(-1) is None
+
+
+def test_append_stream_output_appends() -> None:
+    output, truncated = _append_stream_output("abc", "def", max_bytes=10)
+    assert output == "abcdef"
+    assert truncated is False
+
+
+def test_append_stream_output_truncates() -> None:
+    output, truncated = _append_stream_output("abc", "def", max_bytes=5)
+    assert output == "abcde"
+    assert truncated is True

jupyter_ext/tests/test_mcp_selftest.py

@@ -0,0 +1,204 @@
+"""`/mcp_test` 백엔드 헬퍼 테스트.
+
+bdp_mcp 는 이 패키지의 의존성이 아니므로(별도 venv 설치, 3.9/3.10 요구 충돌),
+여기서는 **설치 여부와 무관하게 도는** 순수 함수만 검증한다.
+실제 프로토콜 왕복은 bdp_mcp 쪽 selftest 가 본다.
+"""
+
+import asyncio
+import json
+import os
+import sys
+
+
+from jupyter_ext.http_handlers import (
+    _kill_quietly,
+    _terminate_then_kill,
+    _BDP_MCP_PYTHON_ENV,
+    _bdp_mcp_python_candidates,
+    _build_selftest_cmd,
+    _parse_selftest_output,
+)
+
+
+def test_cmd_omits_transport_when_both() -> None:
+    """both 는 상류 기본값이라 넘기지 않는다 — 넘기면 인자만 늘고 의미는 같다."""
+    cmd = _build_selftest_cmd("/x/python", "both")
+    assert cmd == ["/x/python", "-m", "awslabs.bdp_mcp_server.selftest", "--json"]
+
+
+def test_cmd_includes_transport_and_port() -> None:
+    cmd = _build_selftest_cmd("/x/python", "streamable-http", 18099)
+    assert cmd[-4:] == ["--transport", "streamable-http", "--port", "18099"]
+
+
+def test_cmd_uses_module_not_console_script() -> None:
+    """콘솔 스크립트는 launchctl/nohup 기동 시 PATH 에 없을 수 있다."""
+    cmd = _build_selftest_cmd(sys.executable)
+    assert cmd[0] == sys.executable
+    assert cmd[1] == "-m"
+    assert "bdp-mcp-selftest" not in cmd
+
+
+def test_parse_reads_stdout_only() -> None:
+    """stderr 에 경고가 섞여도 판정은 stdout 으로만 한다.
+
+    stderr 를 섞어 파싱하면 멀쩡한 결과를 "깨진 출력" 으로 오판한다.
+    """
+    report = {
+        "ok": True,
+        "transports": [{"transport": "stdio", "ok": True, "checks": []}],
+    }
+    parsed = _parse_selftest_output(
+        json.dumps(report).encode(),
+        b"IncompleteFieldDefinitionWarning: Field 'lifespan' ...",
+        0,
+    )
+    assert parsed["report"]["ok"] is True
+    assert "error" not in parsed
+
+
+def test_parse_reports_empty_output() -> None:
+    parsed = _parse_selftest_output(b"", b"boom", 1)
+    assert "error" in parsed
+    assert parsed["stderr"] == "boom"
+
+
+def test_parse_reports_non_json_stdout() -> None:
+    """JSON 이 아니면 원문 꼬리를 남긴다 — 원인을 못 보면 진단이 아니다."""
+    parsed = _parse_selftest_output(b"Traceback (most recent call last):", b"", 1)
+    assert "error" in parsed
+    assert "Traceback" in parsed["stdout"]
+
+
+def test_parse_rejects_unexpected_schema() -> None:
+    """상류 스키마가 바뀌면 조용히 통과시키지 않는다."""
+    parsed = _parse_selftest_output(b'{"ok": true}', b"", 0)
+    assert "error" in parsed
+    assert "transports" in parsed["error"]
+
+
+def test_candidates_prefer_env_var(monkeypatch) -> None:
+    monkeypatch.setenv(_BDP_MCP_PYTHON_ENV, "/custom/venv/bin/python")
+    assert _bdp_mcp_python_candidates()[0] == "/custom/venv/bin/python"
+
+
+def test_candidates_fall_back_without_env(monkeypatch) -> None:
+    """환경변수가 없으면 ~/repo/bdp_mcp/.venv 관습 → 서버 venv 순."""
+    monkeypatch.delenv(_BDP_MCP_PYTHON_ENV, raising=False)
+    candidates = _bdp_mcp_python_candidates()
+    assert candidates[0].endswith("repo/bdp_mcp/.venv/bin/python")
+    assert candidates[-1] == sys.executable
+
+
+def test_candidates_deduplicate(monkeypatch) -> None:
+    """환경변수가 서버 venv 와 같으면 같은 인터프리터를 두 번 재지 않는다."""
+    monkeypatch.setenv(_BDP_MCP_PYTHON_ENV, sys.executable)
+    candidates = _bdp_mcp_python_candidates()
+    assert candidates.count(sys.executable) == 1
+
+
+def test_candidates_expand_tilde_in_env(monkeypatch) -> None:
+    """`~/...` 를 확장하지 않으면 조용히 다음 후보로 폴백해 환경변수가 무시된다.
+
+    진단 도구에서 "왜 내 설정이 안 먹나" 가 묻히는 것은 나쁜 실패다.
+    """
+    monkeypatch.setenv(_BDP_MCP_PYTHON_ENV, "~/myvenv/bin/python")
+    first = _bdp_mcp_python_candidates()[0]
+    assert "~" not in first
+    assert first.endswith("/myvenv/bin/python")
+    assert os.path.isabs(first)
+
+
+class _AlreadyGoneProc:
+    """returncode 확인과 시그널 사이에 스스로 끝난 프로세스."""
+
+    returncode = None
+
+    def kill(self):
+        raise ProcessLookupError
+
+    def terminate(self):
+        raise ProcessLookupError
+
+    async def wait(self):  # pragma: no cover - 도달하지 않아야 한다
+        raise AssertionError("이미 죽은 프로세스를 기다리면 안 된다")
+
+
+def test_kill_quietly_survives_process_lookup_error() -> None:
+    """ProcessLookupError 가 except 블록 안에서 터지면 응답 없는 500 이 된다.
+
+    진단 도구가 진단 불가로 죽는 형태라 반드시 삼켜야 한다.
+    """
+    asyncio.run(_kill_quietly(_AlreadyGoneProc()))
+
+
+def test_terminate_then_kill_survives_process_lookup_error() -> None:
+    asyncio.run(_terminate_then_kill(_AlreadyGoneProc()))
+
+
+def test_kill_helpers_ignore_finished_process() -> None:
+    class _Done:
+        returncode = 0
+
+    asyncio.run(_kill_quietly(_Done()))
+    asyncio.run(_terminate_then_kill(_Done()))
+
+
+def test_terminate_then_kill_prefers_sigterm() -> None:
+    """SIGKILL 로 바로 죽이면 selftest 의 finally 가 못 돌아 손자 MCP 서버가
+    고아로 남아 포트를 계속 점유한다. SIGTERM 을 먼저 준다."""
+    calls = []
+
+    class _Proc:
+        returncode = None
+
+        def terminate(self):
+            calls.append("terminate")
+            self.returncode = -15
+
+        def kill(self):  # pragma: no cover - 여기 오면 안 된다
+            calls.append("kill")
+
+        async def wait(self):
+            return self.returncode
+
+    asyncio.run(_terminate_then_kill(_Proc()))
+    assert calls == ["terminate"]
+
+
+def test_terminate_then_kill_escalates_when_sigterm_ignored() -> None:
+    calls = []
+
+    class _Stubborn:
+        returncode = None
+
+        def terminate(self):
+            calls.append("terminate")
+
+        def kill(self):
+            calls.append("kill")
+            self.returncode = -9
+
+        async def wait(self):
+            if self.returncode is None:
+                await asyncio.sleep(10)  # SIGTERM 을 무시하는 프로세스
+            return self.returncode
+
+    asyncio.run(_terminate_then_kill(_Stubborn(), grace=0.05))
+    assert calls == ["terminate", "kill"]
+
+
+def test_cmd_adds_call_flag() -> None:
+    """`--call` 은 필수 인자 없는 도구를 실제로 호출한다 (실 AWS)."""
+    assert _build_selftest_cmd("/x/python", call_tools=True)[-1] == "--call"
+
+
+def test_cmd_omits_call_flag_by_default() -> None:
+    """기본은 호출하지 않는다 — 자격 없는 환경에서 늘 빨간불이 되면 안 된다."""
+    assert "--call" not in _build_selftest_cmd("/x/python")
+
+
+def test_cmd_combines_transport_port_and_call() -> None:
+    cmd = _build_selftest_cmd("/x/python", "streamable-http", 18099, True)
+    assert cmd[-5:] == ["--transport", "streamable-http", "--port", "18099", "--call"]

jupyter_ext/tests/test_mermaid_fixer.py

@@ -0,0 +1,272 @@
+"""Tests for mermaid_fixer — deterministic Mermaid syntax repair."""
+
+from jupyter_ext.agent.mermaid_fixer import fix_mermaid_blocks, fix_mermaid_code
+
+
+# ---------------------------------------------------------------------------
+# Category 1: Comments
+# ---------------------------------------------------------------------------
+
+
+def test_fix_comments():
+    code = "graph TD\n  A --> B %% this is a comment\n  B --> C"
+    result = fix_mermaid_code(code)
+    assert "%%" not in result
+    assert "A --> B" in result
+
+
+def test_fix_comments_standalone_line():
+    code = "graph TD\n  %% full line comment\n  A --> B"
+    result = fix_mermaid_code(code)
+    assert "%%" not in result
+
+
+# ---------------------------------------------------------------------------
+# Category 2: Reserved keyword `end`
+# ---------------------------------------------------------------------------
+
+
+def test_fix_end_keyword_as_node():
+    code = "graph TD\n  start --> end"
+    result = fix_mermaid_code(code)
+    assert "end_node" in result
+
+
+def test_fix_end_preserves_subgraph_closer():
+    code = "graph TD\n  subgraph sub1\n    A --> B\n  end"
+    result = fix_mermaid_code(code)
+    # The standalone `end` (subgraph closer) must remain
+    assert result.strip().endswith("end")
+
+
+def test_fix_end_preserves_directive():
+    code = "graph TD\n  class end myClass"
+    result = fix_mermaid_code(code)
+    # `end` inside a class directive must remain
+    assert "class end" in result
+
+
+# ---------------------------------------------------------------------------
+# Category 3: Escaped quotes
+# ---------------------------------------------------------------------------
+
+
+def test_fix_escaped_quotes():
+    code = 'A["hello \\"world\\""]'
+    result = fix_mermaid_code(code)
+    assert '\\"' not in result
+    assert """ in result
+
+
+# ---------------------------------------------------------------------------
+# Category 4: Arrow syntax
+# ---------------------------------------------------------------------------
+
+
+def test_fix_single_arrow():
+    code = "graph TD\n  A -> B"
+    result = fix_mermaid_code(code)
+    assert " --> " in result
+
+
+def test_fix_dotted_arrow():
+    code = "graph TD\n  A -. B"
+    result = fix_mermaid_code(code)
+    assert "-.-> " in result
+
+
+def test_double_arrow_unchanged():
+    code = "graph TD\n  A --> B"
+    result = fix_mermaid_code(code)
+    assert " --> " in result
+
+
+# ---------------------------------------------------------------------------
+# Category 5: Nested brackets
+# ---------------------------------------------------------------------------
+
+
+def test_fix_nested_brackets():
+    code = 'graph TD\n  H["dask[dataframe] 설치"]'
+    result = fix_mermaid_code(code)
+    assert "[" in result
+    assert "]" in result
+
+
+def test_fix_misquoted_labels():
+    """Misquoted: opening " but no matching closing "."""
+    code = "graph TD\n  E[\"rcParams['family'\"] 설정]"
+    result = fix_mermaid_code(code)
+    # Should handle gracefully without crashing
+    assert "E[" in result
+
+
+# ---------------------------------------------------------------------------
+# Category 6: Node labels
+# ---------------------------------------------------------------------------
+
+
+def test_fix_unquoted_node_label():
+    code = "graph TD\n  A[Hello World]"
+    result = fix_mermaid_code(code)
+    assert '["Hello World"]' in result
+
+
+def test_fix_already_quoted_label():
+    code = 'graph TD\n  A["Hello World"]'
+    result = fix_mermaid_code(code)
+    assert '["Hello World"]' in result
+
+
+def test_fix_subroutine_label():
+    code = "graph TD\n  A[[My Process]]"
+    result = fix_mermaid_code(code)
+    assert '[["My Process"]]' in result
+
+
+# ---------------------------------------------------------------------------
+# Category 7: Edge labels
+# ---------------------------------------------------------------------------
+
+
+def test_fix_edge_labels_with_brackets():
+    code = "graph TD\n  A -->|[text]| B"
+    result = fix_mermaid_code(code)
+    assert "[" in result
+    assert "]" in result
+
+
+def test_edge_labels_no_brackets_unchanged():
+    code = "graph TD\n  A -->|yes| B"
+    result = fix_mermaid_code(code)
+    assert "|yes|" in result
+
+
+# ---------------------------------------------------------------------------
+# Category 8: Subgraph titles
+# ---------------------------------------------------------------------------
+
+
+def test_fix_subgraph_unquoted_title():
+    code = "subgraph My Section\n  A --> B\nend"
+    result = fix_mermaid_code(code)
+    assert '"My Section"' in result
+
+
+def test_subgraph_quoted_title_unchanged():
+    code = 'subgraph "My Section"\n  A --> B\nend'
+    result = fix_mermaid_code(code)
+    assert '"My Section"' in result
+
+
+# ---------------------------------------------------------------------------
+# Category 9: Empty labels
+# ---------------------------------------------------------------------------
+
+
+def test_fix_empty_labels():
+    code = 'graph TD\n  A[""] --> B'
+    result = fix_mermaid_code(code)
+    assert '[""]' not in result
+    assert "A --> B" in result
+
+
+# ---------------------------------------------------------------------------
+# Category 10: Direction
+# ---------------------------------------------------------------------------
+
+
+def test_fix_invalid_direction():
+    code = "graph TD\n  direction XX\n  A --> B"
+    result = fix_mermaid_code(code)
+    assert "direction TD" in result
+
+
+def test_valid_direction_unchanged():
+    for d in ("TB", "BT", "LR", "RL", "TD"):
+        code = f"graph TD\n  direction {d}\n  A --> B"
+        result = fix_mermaid_code(code)
+        assert f"direction {d}" in result
+
+
+# ---------------------------------------------------------------------------
+# Category 11: Trailing semicolons
+# ---------------------------------------------------------------------------
+
+
+def test_fix_trailing_semicolons():
+    code = "graph TD\n  A --> B;\n  B --> C;"
+    result = fix_mermaid_code(code)
+    assert ";" not in result
+
+
+def test_no_semicolons_unchanged():
+    code = "graph TD\n  A --> B\n  B --> C"
+    result = fix_mermaid_code(code)
+    assert result.strip() == code.strip()
+
+
+# ---------------------------------------------------------------------------
+# Category 12: Duplicate edges
+# ---------------------------------------------------------------------------
+
+
+def test_fix_duplicate_edges():
+    code = "graph TD\n  A --> B\n  A --> B\n  B --> C"
+    result = fix_mermaid_code(code)
+    assert result.count("A --> B") == 1
+    assert "B --> C" in result
+
+
+def test_different_edges_preserved():
+    code = "graph TD\n  A --> B\n  B --> C"
+    result = fix_mermaid_code(code)
+    assert "A --> B" in result
+    assert "B --> C" in result
+
+
+# ---------------------------------------------------------------------------
+# Integration: fix_mermaid_blocks
+# ---------------------------------------------------------------------------
+
+
+def test_no_mermaid_passthrough():
+    text = "Hello world, no mermaid here."
+    assert fix_mermaid_blocks(text) == text
+
+
+def test_multiple_blocks():
+    text = (
+        "Some text\n"
+        "```mermaid\n"
+        "graph TD\n  A -> B\n"
+        "```\n"
+        "Middle text\n"
+        "```mermaid\n"
+        "graph LR\n  C -> D\n"
+        "```\n"
+        "End text"
+    )
+    result = fix_mermaid_blocks(text)
+    assert result.count("-->") == 2
+    assert "Some text" in result
+    assert "Middle text" in result
+    assert "End text" in result
+
+
+def test_composition_multiple_issues():
+    """Multiple issues coexist in a single block."""
+    code = (
+        "graph TD\n"
+        "  %% comment\n"
+        "  start --> end;\n"
+        "  A[Hello World] -> B\n"
+        "  direction XX\n"
+    )
+    result = fix_mermaid_code(code)
+    assert "%%" not in result
+    assert "end_node" in result
+    assert ";" not in result
+    assert " --> " in result  # single arrow fixed
+    assert "direction TD" in result
+    assert '["Hello World"]' in result