Testfix: Make tb tests timezone aware

openhealthcare · Apr 8, 2020 · 3b4705e · 3b4705e
1 parent b0fa27b
commit 3b4705e
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/apps/tb/tests/test_api.py b/apps/tb/tests/test_api.py
@@ -1,13 +1,17 @@
 import datetime
-from plugins.labtests import models
+
+from django.utils import timezone
 from rest_framework.reverse import reverse
 from opal.core.test import OpalTestCase
+
+from plugins.labtests import models
+
 from apps.tb import utils
 
 
 class TestMultipleResults(OpalTestCase):
-    LAB_TEST_DT = datetime.datetime(2019, 1, 4)
-    OBS_DATE_TIME = datetime.datetime(2019, 1, 5)
+    LAB_TEST_DT   = timezone.make_aware(datetime.datetime(2019, 1, 4))
+    OBS_DATE_TIME = timezone.make_aware(datetime.datetime(2019, 1, 5))
 
     TESTS_TO_OBS_TO_VALUES = {
         "C REACTIVE PROTEIN": [
@@ -103,4 +107,3 @@ def test_get_lab_tests(self):
             ]
         }
         self.assertEqual(result.json(), expected)
-