Spaces:
Sleeping
Sleeping
File size: 2,964 Bytes
8e07f8c |
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 |
import pytest
from unittest.mock import patch, MagicMock
from cred_db_mcp_server.tools import (
sync_provider_from_npi,
add_or_update_credential,
list_expiring_credentials,
get_provider_snapshot
)
import httpx
# Mock response data
MOCK_PROVIDER = {"id": 1, "npi": "1234567890", "name": "Dr. Test"}
MOCK_CREDENTIAL = {
"id": 10,
"provider_id": 1,
"type": "Medical License",
"issuer": "State Board",
"number": "MD12345",
"expiry_date": "2025-01-01"
}
MOCK_EXPIRING_LIST = [
{
"provider": MOCK_PROVIDER,
"credential": MOCK_CREDENTIAL,
"days_to_expiry": 30,
"risk_score": 0.5
}
]
MOCK_SNAPSHOT = {
"provider": MOCK_PROVIDER,
"credentials": [MOCK_CREDENTIAL],
"alerts": []
}
@patch("cred_db_mcp_server.tools.httpx.Client")
def test_sync_provider_from_npi(mock_client_cls):
mock_client = MagicMock()
mock_client_cls.return_value.__enter__.return_value = mock_client
mock_client.post.return_value.status_code = 200
mock_client.post.return_value.json.return_value = MOCK_PROVIDER
result = sync_provider_from_npi("1234567890")
assert result == MOCK_PROVIDER
mock_client.post.assert_called_once()
assert "sync_from_npi" in mock_client.post.call_args[0][0]
@patch("cred_db_mcp_server.tools.httpx.Client")
def test_add_or_update_credential(mock_client_cls):
mock_client = MagicMock()
mock_client_cls.return_value.__enter__.return_value = mock_client
mock_client.post.return_value.status_code = 200
mock_client.post.return_value.json.return_value = MOCK_CREDENTIAL
result = add_or_update_credential(1, "Medical License", "State Board", "MD12345", "2025-01-01")
assert result == MOCK_CREDENTIAL
mock_client.post.assert_called_once()
assert "add_or_update" in mock_client.post.call_args[0][0]
@patch("cred_db_mcp_server.tools.httpx.Client")
def test_list_expiring_credentials(mock_client_cls):
mock_client = MagicMock()
mock_client_cls.return_value.__enter__.return_value = mock_client
mock_client.post.return_value.status_code = 200
mock_client.post.return_value.json.return_value = MOCK_EXPIRING_LIST
result = list_expiring_credentials(30)
assert result == MOCK_EXPIRING_LIST
mock_client.post.assert_called_once()
assert "expiring" in mock_client.post.call_args[0][0]
@patch("cred_db_mcp_server.tools.httpx.Client")
def test_get_provider_snapshot(mock_client_cls):
mock_client = MagicMock()
mock_client_cls.return_value.__enter__.return_value = mock_client
mock_client.post.return_value.status_code = 200
mock_client.post.return_value.json.return_value = MOCK_SNAPSHOT
result = get_provider_snapshot(provider_id=1)
assert result == MOCK_SNAPSHOT
mock_client.post.assert_called_once()
assert "snapshot" in mock_client.post.call_args[0][0]
def test_get_provider_snapshot_no_args():
result = get_provider_snapshot()
assert "error" in result
|