Function TestCountTokensSingleZone

ring/ring_test.go:4445–4495

(t *testing.T)

}

func TestCountTokensSingleZone(t *testing.T) {
	tests := map[string]struct {
		ring     *Desc
		expected map[string]int64
	}{
		"empty ring": {
			ring:     &Desc{},
			expected: map[string]int64{},
		},
		"1 instance with 1 token in the ring": {
			ring: &Desc{
				Ingesters: map[string]InstanceDesc{
					"ingester-1": {Tokens: []uint32{1000000}},
				},
			},
			expected: map[string]int64{
				"ingester-1": math.MaxUint32 + 1,
			},
		},
		"1 instance with multiple tokens in the ring": {
			ring: &Desc{
				Ingesters: map[string]InstanceDesc{
					"ingester-1": {Tokens: []uint32{1000000, 2000000, 3000000}},
				},
			},
			expected: map[string]int64{
				"ingester-1": math.MaxUint32 + 1,
			},
		},
		"multiple instances with multiple tokens in the ring": {
			ring: &Desc{
				Ingesters: map[string]InstanceDesc{
					"ingester-1": {Tokens: []uint32{1000000, 3000000, 6000000}},
					"ingester-2": {Tokens: []uint32{2000000, 4000000, 8000000}},
					"ingester-3": {Tokens: []uint32{5000000, 9000000}},
				},
			},
			expected: map[string]int64{
				"ingester-1": 3000000 + (int64(math.MaxUint32) + 1 - 9000000),
				"ingester-2": 4000000,
				"ingester-3": 2000000,
			},
		},
	}

	for testName, testData := range tests {
		t.Run(testName, func(t *testing.T) {
			assert.Equal(t, testData.expected, testData.ring.CountTokens())
		})
	}
}

func TestCountTokensMultiZones(t *testing.T) {
	tests := map[string]struct {