Fixed int test (#3148)

danswer-ai · Nov 16, 2024 · 618bc02 · 618bc02
1 parent b7de74f
commit 618bc02
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/backend/tests/integration/common_utils/managers/cc_pair.py b/backend/tests/integration/common_utils/managers/cc_pair.py
@@ -395,13 +395,19 @@ def get_doc_sync_statuses(
         response.raise_for_status()
         doc_sync_statuses: list[DocumentSyncStatus] = []
         for doc_sync_status in response.json():
+            last_synced = doc_sync_status.get("last_synced")
+            if last_synced:
+                last_synced = datetime.fromisoformat(last_synced)
+
+            last_modified = doc_sync_status.get("last_modified")
+            if last_modified:
+                last_modified = datetime.fromisoformat(last_modified)
+
             doc_sync_statuses.append(
                 DocumentSyncStatus(
                     doc_id=doc_sync_status["doc_id"],
-                    last_synced=datetime.fromisoformat(doc_sync_status["last_synced"]),
-                    last_modified=datetime.fromisoformat(
-                        doc_sync_status["last_modified"]
-                    ),
+                    last_synced=last_synced,
+                    last_modified=last_modified,
                 )
             )