aserrador
c670ea5bc9
- Add test optional dependencies to pyproject.toml - Add conftest.py with fixtures for mocking settings and MCP client - Add test_server.py with tests for tryton_call and MCP tools
142 lines
5.0 KiB
Python
142 lines
5.0 KiB
Python
import pytest
|
|
from unittest.mock import MagicMock, patch
|
|
import datetime
|
|
import json
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
sys.path.insert(0, str(Path(__file__).parent / "src"))
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
class TestTrytonCall:
|
|
async def test_tryton_call_success(self, mock_settings):
|
|
from tryton_mcp.server import tryton_call
|
|
|
|
mock_client = mock_settings.get_client.return_value
|
|
mock_client.call.return_value = [{"id": 1, "name": "Test"}]
|
|
|
|
result = tryton_call("model.test.read", [[1], ["name"]])
|
|
|
|
assert result["success"] is True
|
|
assert result["data"] == [{"id": 1, "name": "Test"}]
|
|
mock_client.call.assert_called_once_with("model.test.read", [[1], ["name"]])
|
|
|
|
async def test_tryton_call_failure(self, mock_settings):
|
|
from tryton_mcp.server import tryton_call
|
|
|
|
mock_client = mock_settings.get_client.return_value
|
|
mock_client.call.side_effect = Exception("Connection error")
|
|
|
|
result = tryton_call("model.test.read", [[1], ["name"]])
|
|
|
|
assert result["success"] is False
|
|
assert "Connection error" in result["error"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
class TestCreateSchedule:
|
|
async def test_create_schedule_success(self, mcp_client, mock_settings):
|
|
from tryton_mcp.server import create_schedule
|
|
|
|
mock_client = mock_settings.get_client.return_value
|
|
mock_client.call.return_value = [1]
|
|
|
|
result = await mcp_client.call_tool(
|
|
"create_schedule",
|
|
{
|
|
"professional": 6,
|
|
"description": "Test appointment",
|
|
"customer": 4,
|
|
"date": "2026-03-15T10:00:00",
|
|
"service_center": 11,
|
|
},
|
|
)
|
|
|
|
assert result.content is not None
|
|
assert len(result.content) == 1
|
|
assert result.content[0].text == "[1]"
|
|
|
|
mock_client.call.assert_called_once()
|
|
|
|
async def test_create_schedule_failure(self, mcp_client, mock_settings):
|
|
from tryton_mcp.server import create_schedule
|
|
|
|
mock_client = mock_settings.get_client.return_value
|
|
mock_client.call.side_effect = Exception("Create failed")
|
|
|
|
with pytest.raises(Exception, match="Failed to create schedule"):
|
|
await mcp_client.call_tool(
|
|
"create_schedule",
|
|
{
|
|
"professional": 6,
|
|
"description": "Test appointment",
|
|
"customer": 4,
|
|
"date": "2026-03-15T10:00:00",
|
|
"service_center": 11,
|
|
},
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
class TestFindServiceCenters:
|
|
async def test_find_service_centers_success(self, mcp_client, mock_settings):
|
|
from tryton_mcp.server import find_service_centers
|
|
|
|
mock_client = mock_settings.get_client.return_value
|
|
mock_client.call.return_value = [
|
|
{"id": 11, "name": "Center 1", "address.street": "Street 1"}
|
|
]
|
|
|
|
result = await mcp_client.call_tool("find_service_centers", {})
|
|
|
|
result_data = json.loads(result.content[0].text)
|
|
assert result_data == [
|
|
{"id": 11, "name": "Center 1", "address.street": "Street 1"}
|
|
]
|
|
call_args = mock_client.call.call_args[0]
|
|
assert call_args[0] == "model.naliia.service_center.search_read"
|
|
assert call_args[1][0] == [[]]
|
|
|
|
async def test_find_service_centers_failure(self, mcp_client, mock_settings):
|
|
from tryton_mcp.server import find_service_centers
|
|
|
|
mock_client = mock_settings.get_client.return_value
|
|
mock_client.call.side_effect = Exception("Search failed")
|
|
|
|
with pytest.raises(Exception, match="Failed to find service centers"):
|
|
await mcp_client.call_tool("find_service_centers", {})
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
class TestFindProductsAndServices:
|
|
async def test_find_products_and_services_success(self, mcp_client, mock_settings):
|
|
from tryton_mcp.server import find_products_and_services
|
|
|
|
mock_client = mock_settings.get_client.return_value
|
|
mock_client.call.return_value = [
|
|
{"id": 1, "name": "Product 1", "list_price": 100.0}
|
|
]
|
|
|
|
result = await mcp_client.call_tool("find_products_and_services", {})
|
|
|
|
result_data = json.loads(result.content[0].text)
|
|
assert result_data == [{"id": 1, "name": "Product 1", "list_price": 100.0}]
|
|
|
|
mock_client.call.assert_called_once()
|
|
|
|
call_args = mock_client.call.call_args[0]
|
|
|
|
assert call_args[0] == "model.product.product.search_read"
|
|
assert call_args[1][0] == [["active", "=", True], ["salable", "=", True]]
|
|
|
|
async def test_find_products_and_services_failure(self, mcp_client, mock_settings):
|
|
from tryton_mcp.server import find_products_and_services
|
|
|
|
mock_client = mock_settings.get_client.return_value
|
|
mock_client.call.side_effect = Exception("Search failed")
|
|
|
|
with pytest.raises(Exception, match="Failed to find products"):
|
|
await mcp_client.call_tool("find_products_and_services", {})
|