diff --git a/tests/incident/test_incident.py b/tests/incident/test_incident.py index 90ac27333..175e417cf 100644 --- a/tests/incident/test_incident.py +++ b/tests/incident/test_incident.py @@ -1,4 +1,4 @@ -from datetime import timedelta +from datetime import timedelta, datetime from django.test import TestCase from django.utils import timezone @@ -116,8 +116,8 @@ def test_stateless_event_on_stateful_incident_should_fix_end_time(self): def test_closing_event_without_finite_end_time_and_reopen_event_should_fix_endt_time(self): incident = StatefulIncidentFactory(level=1) end_time = incident.end_time - self.assertEqual(incident.end_time, INFINITY_REPR) + self.assertEqual(incident.end_time, datetime.max) EventFactory(incident=incident, type=Event.Type.INCIDENT_END) self.asserTrue(incident.repair_end_time()) self.assertNotEqual(end_time, incident.end_time) - self.assertLessThan(incident.end_time, INFINITY_REPR) + self.assertLessThan(incident.end_time, datetime.max)