ensure total links is maintained

varun-edachali-dbx · varun-edachali-dbx · commit 3c1ff9b4ab9e · 2025-07-21T09:58:16.000+05:30
Signed-off-by: varun-edachali-dbx &lt;varun.edachali@databricks.com&gt;
diff --git a/tests/unit/test_cloud_fetch_queue.py b/tests/unit/test_cloud_fetch_queue.py
@@ -69,8 +69,11 @@ def test_initializer_adds_links(self, mock_create_next_table):
             description=description,
         )
 
-        assert len(queue.download_manager._pending_links) == 10
-        assert len(queue.download_manager._download_tasks) == 0
+        assert (
+            len(queue.download_manager._pending_links)
+            + len(queue.download_manager._download_tasks)
+            == 10
+        )
         mock_create_next_table.assert_called()
 
     def test_initializer_no_links_to_add(self):
