def test_update_timeline_event(arf, api_user, update_key, update_value,
expected_value):
incident = IncidentFactory.create()
event_model = incident.timeline_events()[0]
event_data = serializers.TimelineEventSerializer(event_model).data
if update_key:
event_data[update_key] = update_value
req = arf.put(
reverse("incident-timeline-event-list",
kwargs={"incident_pk": incident.pk}),
event_data,
format="json",
)
force_authenticate(req, user=api_user)
response = IncidentTimelineEventViewSet.as_view({"put": "update"})(
req, incident_pk=incident.pk, pk=event_model.pk)
assert response.status_code == 200, "Got non-200 response from API"
if update_key:
new_event = TimelineEvent.objects.get(pk=event_model.pk)
expected = expected_value or update_value
assert (getattr(new_event, update_key) == expected
), "Updated value wasn't persisted to the DB"
def test_delete_timeline_event(arf, api_user):
incident = IncidentFactory.create()
event_model = incident.timeline_events()[0]
req = arf.delete(
reverse("incident-timeline-event-list",
kwargs={"incident_pk": incident.pk}))
force_authenticate(req, user=api_user)
response = IncidentTimelineEventViewSet.as_view({"delete": "destroy"})(
req, incident_pk=incident.pk, pk=event_model.pk)
assert response.status_code == 204, "Got non-204 response from API"
with pytest.raises(TimelineEvent.DoesNotExist):
TimelineEvent.objects.get(pk=event_model.pk)
def assert_create_timeline_event(arf, api_user, incident, event_data):
req = arf.post(
reverse("incident-timeline-event-list",
kwargs={"incident_pk": incident.pk}),
event_data,
format="json",
)
force_authenticate(req, user=api_user)
response = IncidentTimelineEventViewSet.as_view({"post": "create"})(
req, incident_pk=incident.pk)
assert response.status_code == 201, "Got non-201 response from API"
assert TimelineEvent.objects.filter(
incident=incident, timestamp=event_data["timestamp"]).exists()
def test_list_actions_by_incident(arf, api_user):
incident = IncidentFactory.create()
req = arf.get(
reverse("incident-timeline-event-list",
kwargs={"incident_pk": incident.pk}))
force_authenticate(req, user=api_user)
response = IncidentTimelineEventViewSet.as_view({"get": "list"})(
req, incident_pk=incident.pk)
content = json.loads(response.rendered_content)
print(content)
assert response.status_code == 200, "Got non-200 response from API"
assert len(content["results"]) == len(incident.timeline_events())
for event in content["results"]:
assert event["timestamp"]
assert event["text"]
assert event["event_type"]
def test_create_timeline_event(arf, api_user):
incident = IncidentFactory.create()
event_model = TimelineEventFactory.build(incident=incident)
event_data = serializers.TimelineEventSerializer(event_model).data
req = arf.post(
reverse("incident-timeline-event-list",
kwargs={"incident_pk": incident.pk}),
event_data,
format="json",
)
force_authenticate(req, user=api_user)
response = IncidentTimelineEventViewSet.as_view({"post": "create"})(
req, incident_pk=incident.pk)
assert response.status_code == 201, "Got non-201 response from API"
new_action = TimelineEvent.objects.get(incident=incident,
timestamp=event_model.timestamp)