fix waiting coordinator nil

pg-sharding · Aug 29, 2023 · 34c9a6b · 34c9a6b
1 parent f80ad60
commit 34c9a6b
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/test/feature/spqr_test.go b/test/feature/spqr_test.go
@@ -253,6 +253,7 @@ func (tctx *testContext) establishCoordinatorConnection() error {
 			return nil
 		}
 	}
+	delete(tctx.dbs, "coordinator")
 
 	if !ok || err != nil {
 		// check coordinator
@@ -270,12 +271,14 @@ func (tctx *testContext) establishCoordinatorConnection() error {
 				}
 
 				tctx.dbs["coordinator"] = db
-				return nil
 			}
 		}
 	}
 
-	return fmt.Errorf("failed to connect to coordinator")
+	if _, ok := tctx.dbs["coordinator"]; !ok {
+		return fmt.Errorf("failed to connect to coordinator")
+	}
+	return nil
 }
 
 func (tctx *testContext) getPostgresqlConnection(host string) (*sqlx.DB, error) {