add create time tests

bloomberg · Apr 5, 2023 · 644ec28 · 644ec28
1 parent 89c8e61
commit 644ec28
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/test/test_task_creator_helpers.py b/test/test_task_creator_helpers.py
@@ -27,6 +27,12 @@ def test_current_expiration_is_after(self):
         exp = _get_task_expiration(current_exp, None, 60)
         assert are_almost_equal(exp, now + timedelta(days=60))
 
+    def test_create_date_is_set(self):
+        now = datetime.utcnow()
+        current_exp = now + timedelta(days=90)
+        exp = _get_task_expiration(current_exp, now, 60)
+        assert are_almost_equal(exp, now + timedelta(days=60))
+
     def test_current_expiration_is_none(self):
         now = datetime.utcnow()
         exp = _get_task_expiration(None, None, 60)
@@ -47,3 +53,9 @@ def test_current_expiration_is_after(self):
         exp = _get_task_expiration(current_exp.isoformat(), None, 60)
         assert are_almost_equal(
             to_datetime(exp), now + timedelta(days=60))
+
+    def test_create_date_is_set(self):
+        now = datetime.utcnow()
+        current_exp = now + timedelta(days=30)
+        exp = _get_task_expiration(current_exp.isoformat(), now, 60)
+        assert to_datetime(exp) == current_exp