Function TestGetShardSizeForUser

pkg/ruler/ruler_test.go:3138–3284

(t *testing.T)

}

func TestGetShardSizeForUser(t *testing.T) {
	tests := []struct {
		name               string
		userID             string
		replicationFactor  int
		rulerInstanceCount int
		tenantShardSize    float64
		expectedShardSize  int
	}{
		{
			name:               "User with fixed shard size with 10 ruler instances",
			userID:             "user1",
			rulerInstanceCount: 10,
			replicationFactor:  1,
			tenantShardSize:    2,
			expectedShardSize:  2,
		},
		{
			name:               "User with fixed shard size with 50 ruler instances",
			userID:             "user1",
			rulerInstanceCount: 50,
			replicationFactor:  1,
			tenantShardSize:    30,
			expectedShardSize:  30,
		},
		{
			name:               "User with percentage shard size with 10 ruler instances",
			userID:             "user1",
			rulerInstanceCount: 10,
			replicationFactor:  1,
			tenantShardSize:    0.6,
			expectedShardSize:  6,
		},
		{
			name:               "User with percentage shard size with 80 ruler instances",
			userID:             "user1",
			rulerInstanceCount: 80,
			replicationFactor:  1,
			tenantShardSize:    0.25,
			expectedShardSize:  20,
		},
		{
			name:               "Ensure shard size is at least replication factor",
			userID:             "user1",
			rulerInstanceCount: 10,
			replicationFactor:  3,
			tenantShardSize:    0.1,
			expectedShardSize:  3,
		},
	}

	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {

			rulerStateMap := make(map[string]ring.InstanceState)
			rulerAZEvenSpread := make(map[string]string)
			rulerIDs := make([]string, tc.rulerInstanceCount)