mirror of
https://github.com/latinogino/dolibarr-mcp.git
synced 2026-04-12 05:45:35 +02:00
80 lines
3.2 KiB
Python
80 lines
3.2 KiB
Python
import pytest
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
from dolibarr_mcp.dolibarr_mcp_server import handle_call_tool
|
|
from dolibarr_mcp.dolibarr_client import DolibarrClient
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_search_products_by_ref():
|
|
# Mock DolibarrClient
|
|
with patch("dolibarr_mcp.dolibarr_mcp_server.DolibarrClient") as MockClient:
|
|
mock_instance = MockClient.return_value
|
|
mock_instance.__aenter__.return_value = mock_instance
|
|
|
|
# Mock search_products response
|
|
mock_instance.search_products = AsyncMock(return_value=[
|
|
{"id": 1, "ref": "PRJ-123", "label": "Project 123"}
|
|
])
|
|
|
|
# Call the tool
|
|
result = await handle_call_tool("search_products_by_ref", {"ref_prefix": "PRJ"})
|
|
|
|
# Verify the call
|
|
mock_instance.search_products.assert_called_once()
|
|
call_args = mock_instance.search_products.call_args
|
|
assert "sqlfilters" in call_args.kwargs
|
|
assert call_args.kwargs["sqlfilters"] == "(t.ref:like:'PRJ%')"
|
|
|
|
# Verify result
|
|
assert "PRJ-123" in result[0].text
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_resolve_product_ref_exact():
|
|
with patch("dolibarr_mcp.dolibarr_mcp_server.DolibarrClient") as MockClient:
|
|
mock_instance = MockClient.return_value
|
|
mock_instance.__aenter__.return_value = mock_instance
|
|
|
|
# Mock search_products response (exact match)
|
|
mock_instance.search_products = AsyncMock(return_value=[
|
|
{"id": 1, "ref": "PRJ-123", "label": "Project 123"}
|
|
])
|
|
|
|
result = await handle_call_tool("resolve_product_ref", {"ref": "PRJ-123"})
|
|
|
|
mock_instance.search_products.assert_called_once()
|
|
assert "ok" in result[0].text
|
|
assert "PRJ-123" in result[0].text
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_resolve_product_ref_ambiguous():
|
|
with patch("dolibarr_mcp.dolibarr_mcp_server.DolibarrClient") as MockClient:
|
|
mock_instance = MockClient.return_value
|
|
mock_instance.__aenter__.return_value = mock_instance
|
|
|
|
# Mock search_products response (multiple matches, none exact)
|
|
mock_instance.search_products = AsyncMock(return_value=[
|
|
{"id": 1, "ref": "PRJ-123-A", "label": "Project 123 A"},
|
|
{"id": 2, "ref": "PRJ-123-B", "label": "Project 123 B"}
|
|
])
|
|
|
|
# Search for "PRJ-123" which matches both partially (hypothetically) but neither exactly
|
|
result = await handle_call_tool("resolve_product_ref", {"ref": "PRJ-123"})
|
|
|
|
assert "ambiguous" in result[0].text
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_search_customers():
|
|
with patch("dolibarr_mcp.dolibarr_mcp_server.DolibarrClient") as MockClient:
|
|
mock_instance = MockClient.return_value
|
|
mock_instance.__aenter__.return_value = mock_instance
|
|
|
|
mock_instance.search_customers = AsyncMock(return_value=[
|
|
{"id": 1, "nom": "Acme Corp"}
|
|
])
|
|
|
|
result = await handle_call_tool("search_customers", {"query": "Acme"})
|
|
|
|
mock_instance.search_customers.assert_called_once()
|
|
call_args = mock_instance.search_customers.call_args
|
|
assert "sqlfilters" in call_args.kwargs
|
|
assert "Acme" in call_args.kwargs["sqlfilters"]
|