more helper, missed a CancelWorkflow

PeerDB-io · Jan 6, 2024 · d88f76d · d88f76d
1 parent 6809ab3
commit d88f76d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/flow/e2e/bigquery/peer_flow_bq_test.go b/flow/e2e/bigquery/peer_flow_bq_test.go
@@ -1586,6 +1586,8 @@ func (s PeerFlowE2ETestSuiteBQ) Test_Soft_Delete_UD_Same_Batch() {
 				return e2eshared.CheckEqualRecordBatches(s.t, pgRows, rows)
 			},
 		)
+
+		env.CancelWorkflow()
 	}()
 
 	env.ExecuteWorkflow(peerflow.CDCFlowWorkflowWithConfig, config, &limits, nil)

diff --git a/flow/e2e/test_utils.go b/flow/e2e/test_utils.go
@@ -160,6 +160,8 @@ func EnvWaitForEqualTablesWithNames(
 	t.Helper()
 
 	EnvWaitFor(t, env, time.Minute, reason, func(ctx context.Context) bool {
+		t.Helper()
+
 		suffix := suite.Suffix()
 		pool := suite.Pool()
 		pgRows, err := GetPgRows(pool, suffix, srcTable, cols)
-Original file line number
+Diff line change
@@ Expand Up @@
     				return e2eshared.CheckEqualRecordBatches(s.t, pgRows, rows)
     			},
     		)
+    		env.CancelWorkflow()
     	}()
     	env.ExecuteWorkflow(peerflow.CDCFlowWorkflowWithConfig, config, &limits, nil)
@@ Expand Down @@