fix test

PeerDB-io · Nov 20, 2024 · 0752f71 · 0752f71
1 parent 5a086dd
commit 0752f71
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flow/e2e/test_utils.go b/flow/e2e/test_utils.go
@@ -221,7 +221,7 @@ func SetupCDCFlowStatusQuery(t *testing.T, env WorkflowRun, config *protos.FlowC
 			var status protos.FlowStatus
 			if err := response.Get(&status); err != nil {
 				t.Fatal(err)
-			} else if status == protos.FlowStatus_STATUS_RUNNING {
+			} else if status == protos.FlowStatus_STATUS_RUNNING || status == protos.FlowStatus_STATUS_COMPLETED {
 				return
 			} else if counter > 30 {
 				env.Cancel()