added test cases

Signed-off-by: Shashank <[email protected]>

Author: moki1202

pkg/storegateway/sharding_strategy_test.go

@@ -41,6 +41,8 @@ func TestDefaultShardingStrategy(t *testing.T) {
 		zoneAwarenessEnabled bool
 		setupRing            func(*ring.Desc)
 		expectedBlocks       map[string][]ulid.ULID
+		enableTenants        []string
+		disableTenants       []string
 	}{
 		"one ACTIVE instance in the ring with replication factor = 1": {
 			replicationFactor: 1,
@@ -63,6 +65,19 @@ func TestDefaultShardingStrategy(t *testing.T) {
 				"127.0.0.2": {block2, block4},
 			},
 		},
+		"two ACTIVE instances in the ring with replication factor = 1 and one tenant disabled": {
+			replicationFactor: 1,
+			setupRing: func(r *ring.Desc) {
+				r.AddIngester("instance-1", "127.0.0.1", "", []uint32{block1Hash + 1, block3Hash + 1}, ring.ACTIVE, registeredAt)
+				r.AddIngester("instance-2", "127.0.0.2", "", []uint32{block2Hash + 1, block4Hash + 1}, ring.ACTIVE, registeredAt)
+			},
+			enableTenants:  []string{"instance-1"},
+			disableTenants: []string{"instance-2"},
+			expectedBlocks: map[string][]ulid.ULID{
+				"127.0.0.1": {block1, block3}, // Tenant 1 blocks expected
+				"127.0.0.2": {},               // Tenant 2 blocks disabled
+			},
+		},
 		"one ACTIVE instance in the ring with replication factor = 2": {
 			replicationFactor: 2,
 			setupRing: func(r *ring.Desc) {