fix test problem

Uninett · Nov 15, 2024 · d6468ba · d6468ba
1 parent bca5d2f
commit d6468ba
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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)