import pytest
from unittest.mock import AsyncMock, patch
from dolibarr_mcp.config import Config
from dolibarr_mcp.dolibarr_client import DolibarrClient
@pytest.mark.asyncio
class TestInvoiceAtomic:
@pytest.fixture
def client(self):
config = Config(
dolibarr_url="https://test.dolibarr.com/api/index.php",
api_key="test_key"
)
return DolibarrClient(config)
@patch('aiohttp.ClientSession.request')
async def test_add_invoice_line(self, mock_request, client):
mock_response = AsyncMock()
mock_response.status = 200
mock_response.text.return_value = '123' # Returns line ID usually
mock_request.return_value.__aenter__.return_value = mock_response
async with client:
await client.add_invoice_line(
invoice_id=1,
desc="Test Line",
qty=1,
subprice=100,
product_id=99
)
# Verify call
args, kwargs = mock_request.call_args
assert args[0] == "POST"
assert args[1] == "https://test.dolibarr.com/api/index.php/invoices/1/lines"
assert kwargs['json'] == {
"desc": "Test Line",
"qty": 1,
"subprice": 100,
"fk_product": 99
}
@patch('aiohttp.ClientSession.request')
async def test_update_invoice_line(self, mock_request, client):
mock_response = AsyncMock()
mock_response.status = 200
mock_response.text.return_value = '{"success": 1}'
mock_request.return_value.__aenter__.return_value = mock_response
async with client:
await client.update_invoice_line(
invoice_id=1,
line_id=10,
qty=5
)
args, kwargs = mock_request.call_args
assert args[0] == "PUT"
assert args[1] == "https://test.dolibarr.com/api/index.php/invoices/1/lines/10"
assert kwargs['json'] == {"qty": 5}
@patch('aiohttp.ClientSession.request')
async def test_delete_invoice_line(self, mock_request, client):
mock_response = AsyncMock()
mock_response.status = 200
mock_response.text.return_value = '{"success": 1}'
mock_request.return_value.__aenter__.return_value = mock_response
async with client:
await client.delete_invoice_line(invoice_id=1, line_id=10)
args, kwargs = mock_request.call_args
assert args[0] == "DELETE"
assert args[1] == "https://test.dolibarr.com/api/index.php/invoices/1/lines/10"
@patch('aiohttp.ClientSession.request')
async def test_validate_invoice(self, mock_request, client):
mock_response = AsyncMock()
mock_response.status = 200
mock_response.text.return_value = '{"success": 1}'
mock_request.return_value.__aenter__.return_value = mock_response
async with client:
await client.validate_invoice(invoice_id=1, warehouse_id=5)
args, kwargs = mock_request.call_args
assert args[0] == "POST"
assert args[1] == "https://test.dolibarr.com/api/index.php/invoices/1/validate"
assert kwargs['json'] == {"idwarehouse": 5, "not_trigger": 0}
