OM-348: fixing strptime

openimis · Oct 22, 2024 · 155882c · 155882c
1 parent 4318415
commit 155882c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/worker_voucher/tests/test_gql_voucher_check.py b/worker_voucher/tests/test_gql_voucher_check.py
@@ -54,7 +54,7 @@ def test_get_existed_voucher_by_code(self):
         query_data = query_result['data']['voucherCheck']
         self.assertEqual(query_data['isExisted'], True)
         self.assertEqual(query_data['isValid'], True)
-        assigned_date = datetime.strptime(query_data['assignedDate'], '%Y-%m-%d').date()
+        assigned_date = datetime.strptime(query_data['assignedDate'], '%Y-%m-%dT%H:%M:%S.%f').date()
         self.assertEqual(assigned_date, voucher.assigned_date.date())
         self.assertEqual(query_data['employerCode'], voucher.policyholder.code)
         self.assertEqual(query_data['employerName'], voucher.policyholder.trade_name)
@@ -68,7 +68,7 @@ def test_get_existed_voucher_from_tomorrow_by_code(self):
         query_data = query_result['data']['voucherCheck']
         self.assertEqual(query_data['isExisted'], True)
         self.assertEqual(query_data['isValid'], True)
-        assigned_date = datetime.strptime(query_data['assignedDate'], '%Y-%m-%d').date()
+        assigned_date = datetime.strptime(query_data['assignedDate'], '%Y-%m-%dT%H:%M:%S.%f').date()
         self.assertEqual(assigned_date, voucher.assigned_date.date())
         self.assertEqual(query_data['employerCode'], voucher.policyholder.code)
         self.assertEqual(query_data['employerName'], voucher.policyholder.trade_name)
@@ -94,7 +94,7 @@ def test_get_existed_voucher_from_yesterday_by_code(self):
         query_data = query_result['data']['voucherCheck']
         self.assertEqual(query_data['isExisted'], True)
         self.assertEqual(query_data['isValid'], False)
-        assigned_date = datetime.strptime(query_data['assignedDate'], '%Y-%m-%d').date()
+        assigned_date = datetime.strptime(query_data['assignedDate'], '%Y-%m-%dT%H:%M:%S.%f').date()
         self.assertEqual(assigned_date, voucher.assigned_date.date())
         self.assertEqual(query_data['employerCode'], voucher.policyholder.code)
         self.assertEqual(query_data['employerName'], voucher.policyholder.trade_name)