OM-348: fixing test date assigment

openimis · Oct 22, 2024 · 4318415 · 4318415
1 parent f60276a
commit 4318415
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/worker_voucher/tests/test_gql_voucher_check.py b/worker_voucher/tests/test_gql_voucher_check.py
@@ -1,4 +1,4 @@
-from datetime import datetime
+from datetime import datetime, timedelta
 from django.test import TestCase
 from graphene import Schema
 from graphene.test import Client
@@ -38,9 +38,9 @@ def setUpClass(cls):
             mutation=Mutation
         )
 
-        cls.today = datetime.datetime.now()
-        cls.tomorrow = datetime.datetime.now() + datetime.datetimedelta(days=1)
-        cls.yesterday = datetime.datetime.now() - datetime.datetimedelta(days=1)
+        cls.today = datetime.now()
+        cls.tomorrow = datetime.now() + timedelta(days=1)
+        cls.yesterday = datetime.now() - timedelta(days=1)
 
         cls.gql_client = Client(gql_schema)
         cls.gql_context = cls.GQLContext(None)