From 75728aaf0fb44f076046b9e2f587a28344a773f3 Mon Sep 17 00:00:00 2001 From: Rohit Nayak Date: Sun, 13 Oct 2024 18:09:14 +0200 Subject: [PATCH] Add shard read/write shortcut funcs Signed-off-by: Rohit Nayak --- .../vreplication_vtctldclient_cli_test.go | 22 +++++++++++++++++-- 1 file changed, 20 insertions(+), 2 deletions(-) diff --git a/go/test/endtoend/vreplication/vreplication_vtctldclient_cli_test.go b/go/test/endtoend/vreplication/vreplication_vtctldclient_cli_test.go index 3f9054c8eb3..0027bac29cb 100644 --- a/go/test/endtoend/vreplication/vreplication_vtctldclient_cli_test.go +++ b/go/test/endtoend/vreplication/vreplication_vtctldclient_cli_test.go @@ -457,15 +457,33 @@ 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")) + } + + shardReadsRouteToTarget := func() { + require.True(t, getShardRoute(t, keyspace, "-40", "primary")) + } + + shardWritesRouteToSource := func() { + require.True(t, getShardRoute(t, keyspace, "-80", "primary")) + } + + shardWritesRouteToTarget := func() { + require.True(t, getShardRoute(t, keyspace, "-40", "primary")) + } + rs.SwitchReadsAndWrites() waitForLowLag(t, keyspace, workflowName+"_reverse") vdiff(t, keyspace, workflowName+"_reverse", "zone1", true, false, nil) - require.True(t, getShardRoute(t, keyspace, "-40", "primary")) + shardReadsRouteToTarget() + shardWritesRouteToTarget() rs.ReverseReadsAndWrites() waitForLowLag(t, keyspace, workflowName) vdiff(t, keyspace, workflowName, "zone1", false, true, nil) - require.False(t, getShardRoute(t, keyspace, "-40", "primary")) + shardReadsRouteToSource() + shardWritesRouteToSource() rs.SwitchReads()