From 934acc7afdb8f9163a785e71c278beeb61ebfcfe Mon Sep 17 00:00:00 2001 From: twthorn Date: Mon, 12 Aug 2024 13:20:29 -0400 Subject: [PATCH] Fix format --- go/vt/wrangler/traffic_switcher_env_test.go | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/go/vt/wrangler/traffic_switcher_env_test.go b/go/vt/wrangler/traffic_switcher_env_test.go index cf19cb5353c..c39e6f0fd49 100644 --- a/go/vt/wrangler/traffic_switcher_env_test.go +++ b/go/vt/wrangler/traffic_switcher_env_test.go @@ -19,6 +19,8 @@ package wrangler import ( "fmt" "math/rand" + "strconv" + "strings" "sync" "testing" "time" @@ -378,9 +380,10 @@ func newTestTablePartialMigrater(ctx context.Context, t *testing.T, shards, shar now := time.Now().Unix() for i, shard := range shards { - for _, shardToMove := range shardsToMove { + for j, shardToMove := range shardsToMove { var streamInfoRows []string var streamExtInfoRows []string + var vreplIDs []string if shardToMove == shard { bls := &binlogdatapb.BinlogSource{ Keyspace: "ks1", @@ -397,8 +400,10 @@ func newTestTablePartialMigrater(ctx context.Context, t *testing.T, shards, shar } streamInfoRows = append(streamInfoRows, fmt.Sprintf("%d|%v|||", i+1, bls)) streamExtInfoRows = append(streamExtInfoRows, fmt.Sprintf("%d|||||Running|vt_ks1|%d|%d|0|0||||0", i+1, now, now)) + vreplIDs = append(vreplIDs, strconv.FormatInt(int64(j+1), 10)) } - tme.dbTargetClients[i].addInvariant(fmt.Sprintf(copyStateQuery, i+1, i+1), noResult) + vreplIDsJoined := strings.Join(vreplIDs, ", ") + tme.dbTargetClients[i].addInvariant(fmt.Sprintf(copyStateQuery, vreplIDsJoined, vreplIDsJoined), noResult) tme.dbTargetClients[i].addInvariant(streamInfoKs2, sqltypes.MakeTestResult(sqltypes.MakeTestFields( "id|source|message|cell|tablet_types", "int64|varchar|varchar|varchar|varchar"),