e2e/test_utils: don't use query state after error

PeerDB-io · Jan 7, 2024 · 89b894d · 89b894d
1 parent 19178f2
commit 89b894d
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/flow/e2e/test_utils.go b/flow/e2e/test_utils.go
@@ -141,9 +141,7 @@ func SetupCDCFlowStatusQuery(env *testsuite.TestWorkflowEnvironment,
 			err = response.Get(&state)
 			if err != nil {
 				slog.Error(err.Error())
-			}
-
-			if state.CurrentFlowState == protos.FlowStatus_STATUS_RUNNING {
+			} else if state.CurrentFlowState == protos.FlowStatus_STATUS_RUNNING {
 				break
 			}
 		} else {
@@ -171,9 +169,7 @@ func NormalizeFlowCountQuery(env *testsuite.TestWorkflowEnvironment,
 			err = response.Get(&state)
 			if err != nil {
 				slog.Error(err.Error())
-			}
-
-			if len(state.NormalizeFlowStatuses) >= minCount {
+			} else if len(state.NormalizeFlowStatuses) >= minCount {
 				break
 			}
 		} else {