mirror of
https://github.com/MinaSaad1/pbi-cli
synced 2026-04-21 13:37:19 +00:00
51a23668a7
Sprint 6 - REPL Mode + Polish: - Error hierarchy (PbiCliError, McpToolError, etc.) for clean REPL error handling - Interactive REPL with prompt-toolkit (persistent MCP connection, command completion, history) - REPL-aware run_tool() and connection commands that reuse shared client - README.md and CHANGELOG.md Sprint 7 - Tests + CI/CD: - 120 tests across unit, command, and e2e test files (79% coverage) - MockPbiMcpClient with canned responses for test isolation - GitHub Actions CI (lint + typecheck + test matrix: 3 OS x 3 Python) - GitHub Actions release workflow for PyPI trusted publishing Sprint 8 - Claude Skills + Installer: - 5 bundled SKILL.md files (modeling, dax, deployment, security, docs) - `pbi skills install/list/uninstall` command for Claude Code discovery - Skills packaged with wheel via setuptools package-data
92 lines
3.1 KiB
Python
92 lines
3.1 KiB
Python
"""Tests for pbi_cli.core.binary_manager."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
from pathlib import Path
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from pbi_cli.core.binary_manager import (
|
|
_binary_source,
|
|
_find_managed_binary,
|
|
get_binary_info,
|
|
resolve_binary,
|
|
)
|
|
|
|
|
|
def test_resolve_binary_env_var(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None:
|
|
fake_bin = tmp_path / "powerbi-modeling-mcp.exe"
|
|
fake_bin.write_text("fake", encoding="utf-8")
|
|
monkeypatch.setenv("PBI_MCP_BINARY", str(fake_bin))
|
|
|
|
result = resolve_binary()
|
|
assert result == fake_bin
|
|
|
|
|
|
def test_resolve_binary_env_var_missing_file(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setenv("PBI_MCP_BINARY", "/nonexistent/path")
|
|
with pytest.raises(FileNotFoundError, match="non-existent"):
|
|
resolve_binary()
|
|
|
|
|
|
def test_resolve_binary_not_found(
|
|
tmp_config: Path, monkeypatch: pytest.MonkeyPatch
|
|
) -> None:
|
|
monkeypatch.delenv("PBI_MCP_BINARY", raising=False)
|
|
with patch("pbi_cli.core.binary_manager.find_vscode_extension_binary", return_value=None):
|
|
with pytest.raises(FileNotFoundError, match="not found"):
|
|
resolve_binary()
|
|
|
|
|
|
def test_find_managed_binary(tmp_config: Path) -> None:
|
|
bin_dir = tmp_config / "bin" / "0.4.0"
|
|
bin_dir.mkdir(parents=True)
|
|
fake_bin = bin_dir / "powerbi-modeling-mcp.exe"
|
|
fake_bin.write_text("fake", encoding="utf-8")
|
|
|
|
with patch("pbi_cli.core.binary_manager.PBI_CLI_HOME", tmp_config), \
|
|
patch("pbi_cli.core.binary_manager.binary_name", return_value="powerbi-modeling-mcp.exe"):
|
|
result = _find_managed_binary()
|
|
assert result is not None
|
|
assert result.name == "powerbi-modeling-mcp.exe"
|
|
|
|
|
|
def test_find_managed_binary_empty_dir(tmp_config: Path) -> None:
|
|
bin_dir = tmp_config / "bin"
|
|
bin_dir.mkdir(parents=True)
|
|
|
|
with patch("pbi_cli.core.binary_manager.PBI_CLI_HOME", tmp_config):
|
|
result = _find_managed_binary()
|
|
assert result is None
|
|
|
|
|
|
def test_binary_source_env_var(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setenv("PBI_MCP_BINARY", "/some/path")
|
|
result = _binary_source(Path("/some/path"))
|
|
assert "environment variable" in result
|
|
|
|
|
|
def test_binary_source_managed() -> None:
|
|
with patch.dict(os.environ, {}, clear=False):
|
|
if "PBI_MCP_BINARY" in os.environ:
|
|
del os.environ["PBI_MCP_BINARY"]
|
|
result = _binary_source(Path("/home/user/.pbi-cli/bin/0.4.0/binary"))
|
|
assert "managed" in result
|
|
|
|
|
|
def test_binary_source_vscode() -> None:
|
|
with patch.dict(os.environ, {}, clear=False):
|
|
if "PBI_MCP_BINARY" in os.environ:
|
|
del os.environ["PBI_MCP_BINARY"]
|
|
result = _binary_source(Path("/home/user/.vscode/extensions/ext/server/binary"))
|
|
assert "VS Code" in result
|
|
|
|
|
|
def test_get_binary_info_not_found(tmp_config: Path, monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.delenv("PBI_MCP_BINARY", raising=False)
|
|
with patch("pbi_cli.core.binary_manager.find_vscode_extension_binary", return_value=None):
|
|
info = get_binary_info()
|
|
assert info["binary_path"] == "not found"
|
|
assert info["version"] == "none"
|