test_notes_unit.py•3.49 kB
"""Unit tests for PagerDuty incident note operations."""
import pytest
from pagerduty_mcp_server import incidents, utils
from pagerduty_mcp_server.parsers.notes_parser import parse_note
"""
Parser Tests
"""
@pytest.mark.unit
@pytest.mark.parsers
def test_parse_note(mock_notes, mock_notes_parsed):
"""Test that note parsing works correctly."""
parsed_note = parse_note(mock_notes[0])
assert parsed_note == mock_notes_parsed[0]
@pytest.mark.unit
@pytest.mark.parsers
def test_parse_note_none():
"""Test parsing None input."""
result = parse_note(None)
assert result is None
@pytest.mark.unit
@pytest.mark.parsers
def test_parse_note_missing_fields():
"""Test parsing note with missing fields."""
input_data = {"id": "PWL7QXS", "content": "Test content"}
result = parse_note(input_data)
assert result["id"] == "PWL7QXS"
assert result["content"] == "Test content"
assert result["created_at"] is None
assert result["user"]["id"] is None
assert result["user"]["name"] is None
assert result["channel"]["summary"] is None
"""
API Function Tests
"""
@pytest.mark.unit
@pytest.mark.incidents
def test__list_notes(mock_get_api_client, mock_notes):
"""Test that notes are fetched correctly."""
incident_id = "ABC123"
mock_get_api_client.jget.return_value = {"notes": mock_notes}
result = incidents._list_notes(incident_id=incident_id)
mock_get_api_client.jget.assert_called_once_with(
f"{incidents.INCIDENTS_URL}/{incident_id}/notes"
)
expected_response = utils.api_response_handler(
results=[parse_note(note) for note in mock_notes], resource_name="notes"
)
assert result == expected_response
@pytest.mark.unit
@pytest.mark.incidents
def test__list_notes_empty_id(mock_get_api_client):
"""Test that _list_notes raises ValueError for empty incident ID."""
with pytest.raises(ValueError, match="incident_id cannot be empty"):
incidents._list_notes(incident_id="")
@pytest.mark.unit
@pytest.mark.incidents
def test__list_notes_api_error(mock_get_api_client):
"""Test that _list_notes handles API errors correctly."""
incident_id = "ABC123"
mock_get_api_client.jget.side_effect = RuntimeError("API Error")
with pytest.raises(RuntimeError) as exc_info:
incidents._list_notes(incident_id=incident_id)
assert str(exc_info.value) == "API Error"
@pytest.mark.unit
@pytest.mark.incidents
def test__list_notes_missing_notes_field(mock_get_api_client):
"""Test that _list_notes handles missing notes field in response correctly."""
incident_id = "ABC123"
mock_get_api_client.jget.return_value = {} # Missing 'notes' key
with pytest.raises(RuntimeError) as exc_info:
incidents._list_notes(incident_id=incident_id)
assert (
str(exc_info.value)
== f"Failed to fetch notes for incident {incident_id}: Response missing 'notes' field"
)
@pytest.mark.unit
@pytest.mark.incidents
def test__list_notes_empty_list(mock_get_api_client):
"""Test that _list_notes handles empty notes list correctly."""
incident_id = "ABC123"
mock_get_api_client.jget.return_value = {"notes": []}
result = incidents._list_notes(incident_id=incident_id)
mock_get_api_client.jget.assert_called_once_with(
f"{incidents.INCIDENTS_URL}/{incident_id}/notes"
)
expected_response = utils.api_response_handler(results=[], resource_name="notes")
assert result == expected_response