blob: c7afaf9f9fcb93b5331a9a96b11a1e2cf85a5981 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
#!/usr/bin/env python3
"""Pytest configuration and fixtures for GitLab CLI MCP server tests."""
from unittest.mock import MagicMock, patch
import pytest
from mcp_server_gitlab_glab.server import GitLabServer, create_server
@pytest.fixture
def gitlab_server() -> GitLabServer:
"""Fixture for a GitLabServer instance."""
return GitLabServer()
@pytest.fixture
def mock_subprocess_run() -> MagicMock:
"""Fixture for mocking subprocess.run."""
with patch("subprocess.run") as mock_run:
yield mock_run
@pytest.fixture
def mock_successful_command(mock_subprocess_run: MagicMock) -> MagicMock:
"""Fixture for mocking a successful command execution."""
mock_process = MagicMock()
mock_process.returncode = 0
mock_process.stdout = "command output"
mock_process.stderr = ""
mock_subprocess_run.return_value = mock_process
return mock_subprocess_run
@pytest.fixture
def mock_failed_command(mock_subprocess_run: MagicMock) -> MagicMock:
"""Fixture for mocking a failed command execution."""
mock_process = MagicMock()
mock_process.returncode = 1
mock_process.stdout = ""
mock_process.stderr = "command failed"
mock_subprocess_run.return_value = mock_process
return mock_subprocess_run
@pytest.fixture
def mock_auth_error_command(mock_subprocess_run: MagicMock) -> MagicMock:
"""Fixture for mocking an authentication error command execution."""
mock_process = MagicMock()
mock_process.returncode = 1
mock_process.stdout = ""
mock_process.stderr = "authentication required"
mock_subprocess_run.return_value = mock_process
return mock_subprocess_run
@pytest.fixture
def mock_successful_api_command(mock_subprocess_run: MagicMock) -> MagicMock:
"""Fixture for mocking a successful API command execution."""
mock_process = MagicMock()
mock_process.returncode = 0
mock_process.stdout = '[{"id": 1, "name": "test-project"}]'
mock_process.stderr = ""
mock_subprocess_run.return_value = mock_process
return mock_subprocess_run
@pytest.fixture
def mock_empty_api_command(mock_subprocess_run: MagicMock) -> MagicMock:
"""Fixture for mocking an API command with empty results."""
mock_process = MagicMock()
mock_process.returncode = 0
mock_process.stdout = "[]"
mock_process.stderr = ""
mock_subprocess_run.return_value = mock_process
return mock_subprocess_run
@pytest.fixture
def mock_invalid_json_api_command(mock_subprocess_run: MagicMock) -> MagicMock:
"""Fixture for mocking an API command with invalid JSON response."""
mock_process = MagicMock()
mock_process.returncode = 0
mock_process.stdout = "invalid json"
mock_process.stderr = ""
mock_subprocess_run.return_value = mock_process
return mock_subprocess_run
@pytest.fixture
def mock_command_not_found(mock_subprocess_run: MagicMock) -> MagicMock:
"""Fixture for mocking a command not found error."""
mock_subprocess_run.side_effect = FileNotFoundError(
"No such file or directory: 'glab'"
)
return mock_subprocess_run
@pytest.fixture
def mcp_server() -> MagicMock:
"""Fixture for a mocked MCP server."""
with patch("mcp.server.fastmcp.FastMCP") as mock_fastmcp:
mock_server = MagicMock()
mock_fastmcp.return_value = mock_server
yield create_server()
|
