Spaces:
Running
Running
File size: 1,001 Bytes
fd1472e |
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 |
"""Tests for MCP Tools domain support."""
from unittest.mock import MagicMock, patch
from src.mcp_tools import search_pubmed
class TestMCPToolsDomain:
@patch("src.mcp_tools._pubmed.search")
async def test_search_pubmed_accepts_domain(self, mock_search):
mock_search.return_value = []
result = await search_pubmed("query", domain="sexual_health")
# The function returns "No PubMed results found..." if empty
assert "No PubMed results" in result
# Let's mock results
mock_evidence = MagicMock()
mock_evidence.citation.title = "Test Title"
mock_evidence.citation.authors = ["Author"]
mock_evidence.citation.date = "2024"
mock_evidence.citation.url = "http://url"
mock_evidence.content = "content"
mock_search.return_value = [mock_evidence]
result = await search_pubmed("query", domain="sexual_health")
assert "## PubMed Results for: query (Sexual Health Research)" in result
|