Switch different Reshard traffic back and forth

Signed-off-by: Rohit Nayak <rohit@planetscale.com>
planetscale · Oct 18, 2024 · 2d35bc4 · 2d35bc4
1 parent 75728aa
commit 2d35bc4
Showing 1 changed file with 24 additions and 3 deletions.
diff --git a/go/test/endtoend/vreplication/vreplication_vtctldclient_cli_test.go b/go/test/endtoend/vreplication/vreplication_vtctldclient_cli_test.go
@@ -458,11 +458,11 @@ func splitShard(t *testing.T, keyspace, workflowName, sourceShards, targetShards
 	vdiff(t, keyspace, workflowName, "zone1", false, true, nil)
 
 	shardReadsRouteToSource := func() {
-		require.True(t, getShardRoute(t, keyspace, "-80", "primary"))
+		require.True(t, getShardRoute(t, keyspace, "-80", "replica"))
 	}
 
 	shardReadsRouteToTarget := func() {
-		require.True(t, getShardRoute(t, keyspace, "-40", "primary"))
+		require.True(t, getShardRoute(t, keyspace, "-40", "replica"))
 	}
 
 	shardWritesRouteToSource := func() {
@@ -486,27 +486,48 @@ func splitShard(t *testing.T, keyspace, workflowName, sourceShards, targetShards
 	shardWritesRouteToSource()
 
 	rs.SwitchReads()
+	shardReadsRouteToTarget()
+	shardWritesRouteToSource()
 
 	rs.ReverseReads()
+	shardReadsRouteToSource()
+	shardWritesRouteToSource()
 
 	rs.SwitchReadsAndWrites()
+	shardReadsRouteToTarget()
+	shardWritesRouteToTarget()
 
 	rs.ReverseReadsAndWrites()
+	shardReadsRouteToSource()
+	shardWritesRouteToSource()
 
 	rs.SwitchReadsAndWrites()
+	shardReadsRouteToTarget()
+	shardWritesRouteToTarget()
 
 	rs.ReverseReads()
+	shardReadsRouteToSource()
+	shardWritesRouteToTarget()
 
 	rs.ReverseWrites()
+	shardReadsRouteToSource()
+	shardWritesRouteToSource()
 
 	rs.SwitchReadsAndWrites()
+	shardReadsRouteToTarget()
+	shardWritesRouteToTarget()
 
 	rs.ReverseWrites()
+	shardReadsRouteToTarget()
+	shardWritesRouteToSource()
 
 	rs.ReverseReads()
+	shardReadsRouteToSource()
+	shardWritesRouteToSource()
 
 	rs.SwitchReadsAndWrites()
-	require.True(t, getShardRoute(t, keyspace, "-40", "primary"))
+	shardReadsRouteToTarget()
+	shardWritesRouteToTarget()
 
 	rs.Complete()
 }