Function TestPartitionRingGetTokenRangesForPartition

ring/partition_ring_test.go:1025–1113

(t *testing.T)

}

func TestPartitionRingGetTokenRangesForPartition(t *testing.T) {
	gen := initTokenGenerator(t)

	tests := map[string]struct {
		partitionTokens map[int][]uint32
		expected        map[int]TokenRanges
	}{
		"single instance in zone": {
			partitionTokens: map[int][]uint32{
				0: gen.GenerateTokens(512, nil),
			},
			expected: map[int]TokenRanges{
				0: {0, math.MaxUint32},
			},
		},
		"simple ranges": {
			partitionTokens: map[int][]uint32{
				0: {25, 75},
				1: {10, 50, 100},
			},
			expected: map[int]TokenRanges{
				0: {10, 24, 50, 74},
				1: {0, 9, 25, 49, 75, math.MaxUint32},
			},
		},
		"grouped tokens": {
			partitionTokens: map[int][]uint32{
				0: {10, 20, 30, 40, 50},
				1: {1000, 2000, 3000, 4000},
			},
			expected: map[int]TokenRanges{
				0: {0, 49, 4000, math.MaxUint32},
				1: {50, 3999},
			},
		},
		"consecutive tokens": {
			partitionTokens: map[int][]uint32{
				0: {99},
				1: {100},
			},
			expected: map[int]TokenRanges{
				0: {0, 98, 100, math.MaxUint32},
				1: {99, 99},
			},
		},
		"extremes": {
			partitionTokens: map[int][]uint32{
				0: {0},
				1: {math.MaxUint32},
			},
			expected: map[int]TokenRanges{
				0: {math.MaxUint32, math.MaxUint32},
				1: {0, math.MaxUint32 - 1},
			},
		},
	}

	for testName, testData := range tests {