Fix breakage from concurrent merge

We merged these two PRs concurrently and thus ended up with a breakage: - #14735 - #14786 This PR addresses that issue by using a locally scoped vtgate connection according to the newly established model. Signed-off-by: Matt Lord <[email protected]>
vitessio · Dec 27, 2023 · ef7fbd2 · ef7fbd2
1 parent 4bb84ed
commit ef7fbd2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/go/test/endtoend/vreplication/vdiff2_test.go b/go/test/endtoend/vreplication/vdiff2_test.go
@@ -172,6 +172,8 @@ func TestVDiff2(t *testing.T) {
 }
 
 func testWorkflow(t *testing.T, vc *VitessCluster, tc *testCase, tks *Keyspace, cells []*Cell) {
+	vtgateConn := vc.GetVTGateConn(t)
+	defer vtgateConn.Close()
 	arrTargetShards := strings.Split(tc.targetShards, ",")
 	if tc.typ == "Reshard" {
 		require.NoError(t, vc.AddShards(t, cells, tks, tc.targetShards, 0, 0, tc.tabletBaseID, targetKsOpts))