Antoni Nuñez Romeu
139 lines
5.6 KiB
Python
139 lines
5.6 KiB
Python
import asyncio
|
|
from uuid import UUID, uuid4
|
|
|
|
import pytest
|
|
|
|
from src.connectors.mock.mock_connector import MockConnector
|
|
from src.connectors.base import ConnectorResult
|
|
from src.core.exceptions import ConnectorException
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_connector() -> MockConnector:
|
|
return MockConnector()
|
|
|
|
|
|
@pytest.fixture
|
|
def pharmacy_id() -> UUID:
|
|
return uuid4()
|
|
|
|
|
|
@pytest.fixture
|
|
def medication_id() -> UUID:
|
|
return uuid4()
|
|
|
|
|
|
class TestMockConnectorConfiguration:
|
|
def test_default_configuration(self, mock_connector: MockConnector):
|
|
assert mock_connector._simulate_latency is True
|
|
assert mock_connector._error_rate == 0.0
|
|
|
|
def test_configure_latency(self, mock_connector: MockConnector):
|
|
mock_connector.configure({"simulate_latency": False, "latency_range_ms": [5, 10]})
|
|
assert mock_connector._simulate_latency is False
|
|
assert mock_connector._latency_range_ms == (5, 10)
|
|
|
|
def test_configure_error_rate(self, mock_connector: MockConnector):
|
|
mock_connector.configure({"error_rate": 0.5})
|
|
assert mock_connector._error_rate == 0.5
|
|
|
|
|
|
class TestMockConnectorGetStock:
|
|
@pytest.mark.asyncio
|
|
async def test_get_stock_all(self, mock_connector: MockConnector, pharmacy_id: UUID):
|
|
mock_connector.configure({"simulate_latency": False})
|
|
result = await mock_connector.get_stock(pharmacy_id)
|
|
assert result.success is True
|
|
assert "items" in result.data
|
|
assert result.data["total"] == 10
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_stock_by_nregistro(self, mock_connector: MockConnector, pharmacy_id: UUID):
|
|
mock_connector.configure({"simulate_latency": False})
|
|
result = await mock_connector.get_stock(pharmacy_id, nregistro="00001")
|
|
assert result.success is True
|
|
assert result.data["nregistro"] == "00001"
|
|
assert result.data["name"] == "Ibuprofeno 600mg"
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_stock_not_found(self, mock_connector: MockConnector, pharmacy_id: UUID):
|
|
mock_connector.configure({"simulate_latency": False})
|
|
result = await mock_connector.get_stock(pharmacy_id, nregistro="99999")
|
|
assert result.success is False
|
|
assert "not found" in result.error
|
|
|
|
|
|
class TestMockConnectorReserve:
|
|
@pytest.mark.asyncio
|
|
async def test_reserve(self, mock_connector: MockConnector, pharmacy_id: UUID, medication_id: UUID):
|
|
mock_connector.configure({"simulate_latency": False})
|
|
result = await mock_connector.reserve(pharmacy_id, medication_id, 5)
|
|
assert result.success is True
|
|
assert result.data["status"] == "confirmed"
|
|
assert result.data["quantity"] == 5
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_reserve_idempotency(self, mock_connector: MockConnector, pharmacy_id: UUID, medication_id: UUID):
|
|
mock_connector.configure({"simulate_latency": False})
|
|
key = "idem-key-123"
|
|
result1 = await mock_connector.reserve(pharmacy_id, medication_id, 3, idempotency_key=key)
|
|
result2 = await mock_connector.reserve(pharmacy_id, medication_id, 3, idempotency_key=key)
|
|
assert result1.success is True
|
|
assert result2.success is True
|
|
|
|
|
|
class TestMockConnectorCancelReservation:
|
|
@pytest.mark.asyncio
|
|
async def test_cancel_reservation(self, mock_connector: MockConnector, pharmacy_id: UUID):
|
|
mock_connector.configure({"simulate_latency": False})
|
|
reserve_result = await mock_connector.reserve(pharmacy_id, uuid4(), 2)
|
|
reservation_id = reserve_result.data["reservation_id"]
|
|
result = await mock_connector.cancel_reservation(pharmacy_id, reservation_id)
|
|
assert result.success is True
|
|
assert result.data["status"] == "cancelled"
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_cancel_nonexistent_reservation(self, mock_connector: MockConnector, pharmacy_id: UUID):
|
|
mock_connector.configure({"simulate_latency": False})
|
|
result = await mock_connector.cancel_reservation(pharmacy_id, "nonexistent-id")
|
|
assert result.success is False
|
|
|
|
|
|
class TestMockConnectorPharmacyInfo:
|
|
@pytest.mark.asyncio
|
|
async def test_pharmacy_information(self, mock_connector: MockConnector, pharmacy_id: UUID):
|
|
mock_connector.configure({"simulate_latency": False})
|
|
result = await mock_connector.pharmacy_information(pharmacy_id)
|
|
assert result.success is True
|
|
assert "FARM-001" in result.data["code"]
|
|
|
|
|
|
class TestMockConnectorSynchronize:
|
|
@pytest.mark.asyncio
|
|
async def test_synchronize(self, mock_connector: MockConnector, pharmacy_id: UUID):
|
|
mock_connector.configure({"simulate_latency": False, "error_rate": 0.0})
|
|
result = await mock_connector.synchronize(pharmacy_id, strategy="full")
|
|
assert result.success is True
|
|
assert result.data["items_total"] == 10
|
|
assert result.data["items_failed"] == 0
|
|
|
|
|
|
class TestMockConnectorHeartbeat:
|
|
@pytest.mark.asyncio
|
|
async def test_heartbeat(self, mock_connector: MockConnector, pharmacy_id: UUID):
|
|
result = await mock_connector.heartbeat(pharmacy_id)
|
|
assert result.success is True
|
|
assert result.data["status"] == "healthy"
|
|
assert result.data["connector_type"] == "mock"
|
|
|
|
|
|
class TestMockConnectorErrorSimulation:
|
|
def test_high_error_rate_triggers_exception(self, mock_connector: MockConnector):
|
|
mock_connector.configure({"error_rate": 1.0})
|
|
with pytest.raises(ConnectorException):
|
|
mock_connector._maybe_error()
|
|
|
|
def test_zero_error_rate_no_exception(self, mock_connector: MockConnector):
|
|
mock_connector.configure({"error_rate": 0.0})
|
|
mock_connector._maybe_error()
|