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_create_timeline_event(arf, api_user):
incident = IncidentFactory.create()
event_model = TimelineEventFactory.build(incident=incident)
event_data = serializers.TimelineEventSerializer(event_model).data
assert_create_timeline_event(arf, api_user, incident, event_data)
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)
