add a couple tests

slackhq · Jul 8, 2024 · 308d7c8 · 308d7c8
1 parent f5522fc
commit 308d7c8
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/go/vt/vtgate/balancer/balancer_test.go b/go/vt/vtgate/balancer/balancer_test.go
@@ -268,6 +268,22 @@ func TestBalancedPick(t *testing.T) {
 
 			[]string{"a", "b", "c", "d"},
 		},
+		{
+			"one target same cell",
+			[]*discovery.TabletHealth{
+				createTestTablet("a"),
+			},
+
+			[]string{"a"},
+		},
+		{
+			"one target other cell",
+			[]*discovery.TabletHealth{
+				createTestTablet("a"),
+			},
+
+			[]string{"b", "c", "d"},
+		},
 	}
 
 	target := &querypb.Target{Keyspace: "k", Shard: "s", TabletType: topodatapb.TabletType_REPLICA}