mirror of
https://github.com/MinaSaad1/pbi-cli
synced 2026-04-21 13:37:19 +00:00
11691c8d92
- Fix Traversable import with TYPE_CHECKING guard for Python 3.10-3.14 compat - Add explicit dict[str, object] type annotations to all request dicts - Fix unused variable and import warnings in connection.py and test files - Break long lines to satisfy ruff E501 (100-char limit) - Fix click.MultiCommand -> click.Group for mypy arg-type in repl.py - Wrap binary_manager return in str() for mypy no-any-return
92 lines
3 KiB
Python
92 lines
3 KiB
Python
"""Tests for pbi_cli.commands._helpers."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from pbi_cli.commands._helpers import build_definition, run_tool
|
|
from pbi_cli.core.errors import McpToolError
|
|
from pbi_cli.main import PbiContext
|
|
from tests.conftest import MockPbiMcpClient
|
|
|
|
|
|
def test_build_definition_required_only() -> None:
|
|
result = build_definition(
|
|
required={"name": "Sales"},
|
|
optional={},
|
|
)
|
|
assert result == {"name": "Sales"}
|
|
|
|
|
|
def test_build_definition_filters_none() -> None:
|
|
result = build_definition(
|
|
required={"name": "Sales"},
|
|
optional={"description": None, "folder": "Finance"},
|
|
)
|
|
assert result == {"name": "Sales", "folder": "Finance"}
|
|
assert "description" not in result
|
|
|
|
|
|
def test_build_definition_preserves_falsy_non_none() -> None:
|
|
result = build_definition(
|
|
required={"name": "Sales"},
|
|
optional={"hidden": False, "count": 0, "label": ""},
|
|
)
|
|
assert result["hidden"] is False
|
|
assert result["count"] == 0
|
|
assert result["label"] == ""
|
|
|
|
|
|
def test_run_tool_adds_connection(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
mock = MockPbiMcpClient()
|
|
monkeypatch.setattr("pbi_cli.commands._helpers.get_client", lambda repl_mode=False: mock)
|
|
|
|
ctx = PbiContext(json_output=True, connection="my-conn")
|
|
run_tool(ctx, "measure_operations", {"operation": "List"})
|
|
|
|
assert mock.calls[0][1]["connectionName"] == "my-conn"
|
|
|
|
|
|
def test_run_tool_no_connection(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
mock = MockPbiMcpClient()
|
|
monkeypatch.setattr("pbi_cli.commands._helpers.get_client", lambda repl_mode=False: mock)
|
|
|
|
ctx = PbiContext(json_output=True)
|
|
run_tool(ctx, "measure_operations", {"operation": "List"})
|
|
|
|
assert "connectionName" not in mock.calls[0][1]
|
|
|
|
|
|
def test_run_tool_stops_client_in_oneshot(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
mock = MockPbiMcpClient()
|
|
monkeypatch.setattr("pbi_cli.commands._helpers.get_client", lambda repl_mode=False: mock)
|
|
|
|
ctx = PbiContext(json_output=True, repl_mode=False)
|
|
run_tool(ctx, "measure_operations", {"operation": "List"})
|
|
|
|
assert mock.stopped is True
|
|
|
|
|
|
def test_run_tool_keeps_client_in_repl(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
mock = MockPbiMcpClient()
|
|
monkeypatch.setattr("pbi_cli.commands._helpers.get_client", lambda repl_mode=False: mock)
|
|
|
|
ctx = PbiContext(json_output=True, repl_mode=True)
|
|
run_tool(ctx, "measure_operations", {"operation": "List"})
|
|
|
|
assert mock.stopped is False
|
|
|
|
|
|
def test_run_tool_raises_mcp_tool_error_on_failure(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
class FailingClient(MockPbiMcpClient):
|
|
def call_tool(self, tool_name: str, request: dict) -> None:
|
|
raise RuntimeError("server crashed")
|
|
|
|
mock = FailingClient()
|
|
monkeypatch.setattr("pbi_cli.commands._helpers.get_client", lambda repl_mode=False: mock)
|
|
|
|
ctx = PbiContext(json_output=True)
|
|
with pytest.raises(McpToolError):
|
|
run_tool(ctx, "measure_operations", {"operation": "List"})
|