Make TestUploadTestProcessor less flaky

codecov · Nov 5, 2024 · 1adff7e · 1adff7e
1 parent 362c609
commit 1adff7e
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/tasks/tests/unit/test_test_results_processor_task.py b/tasks/tests/unit/test_test_results_processor_task.py
@@ -525,12 +525,9 @@ def test_upload_processor_task_call_existing_test_diff_flags_hash(
 
         test_flag_bridges = dbsession.query(TestFlagBridge).all()
 
-        assert [bridge.test_id for bridge in test_flag_bridges] == [
-            tests[1].id,
-            tests[2].id,
-            tests[3].id,
-            tests[4].id,
-        ]
+        assert set(bridge.test_id for bridge in test_flag_bridges) == set(
+            instance.test_id for instance in test_instances
+        )
         for bridge in test_flag_bridges:
             assert bridge.flag == repo_flag