Add test for parse_datetime

Sage-Bionetworks-Workflows · May 1, 2023 · a906559 · a906559
1 parent 70c9bc7
commit a906559
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/src/orca/services/nextflowtower/utils.py b/src/orca/services/nextflowtower/utils.py
@@ -1,4 +1,4 @@
-from datetime import datetime
+from datetime import datetime, timezone
 
 
 def parse_datetime(text: str) -> datetime:
@@ -10,4 +10,6 @@ def parse_datetime(text: str) -> datetime:
     Returns:
         Datetime object.
     """
-    return datetime.strptime(text, "%Y-%m-%dT%H:%M:%SZ")
+    parsed = datetime.strptime(text, "%Y-%m-%dT%H:%M:%SZ")
+    parsed = parsed.replace(tzinfo=timezone.utc)
+    return parsed
diff --git a/tests/services/nextflowtower/test_utils.py b/tests/services/nextflowtower/test_utils.py
@@ -0,0 +1,8 @@
+from datetime import datetime, timezone
+
+from orca.services.nextflowtower import utils
+
+
+def test_that_parse_datetime_works():
+    result = utils.parse_datetime("2023-04-26T00:49:49Z")
+    assert result == datetime(2023, 4, 26, 0, 49, 49, tzinfo=timezone.utc)