Function TestRing_Get_Stability

pkg/ring/ring_test.go:508–574

(t *testing.T)

}

func TestRing_Get_Stability(t *testing.T) {
	const numOfInvocations = 10
	const numOfTokensToTest = 10
	g := NewRandomTokenGenerator()

	tests := map[string]struct {
		numOfZones        int
		replicationFactor int
	}{
		"should return same set for same operation when zone awareness is disabled": {
			numOfZones:        0,
			replicationFactor: 3,
		},
		"should return same set when RF is equal to number of zones": {
			numOfZones:        3,
			replicationFactor: 3,
		},
		"should return same set when RF is less than number of zones": {
			numOfZones:        3,
			replicationFactor: 2,
		},
		"should return same set when RF is greater than number of zones": {
			numOfZones:        3,
			replicationFactor: 9,
		},
		"should return same set when number of instances in each zone is inconsistent": {
			numOfZones:        3,
			replicationFactor: 8,
		},
	}

	for testName, testData := range tests {
		t.Run(testName, func(t *testing.T) {
			testValues := g.GenerateTokens(NewDesc(), "", "", numOfInvocations, true)
			bufDescs, bufHosts, bufZones := MakeBuffersForGet()

			ringDesc := &Desc{Ingesters: generateRingInstances(16, testData.numOfZones, 128)}
			ring := Ring{
				cfg: Config{
					HeartbeatTimeout:     time.Hour,
					ZoneAwarenessEnabled: testData.numOfZones > 0,
					ReplicationFactor:    testData.replicationFactor,
				},
				ringDesc:            ringDesc,
				ringTokens:          ringDesc.GetTokens(),
				ringTokensByZone:    ringDesc.getTokensByZone(),
				ringInstanceByToken: ringDesc.getTokensInfo(),
				ringZones:           getZones(ringDesc.getTokensByZone()),
				strategy:            NewDefaultReplicationStrategy(),
				KVClient:            &MockClient{},
			}

			for i := range numOfTokensToTest {
				expectedSet, err := ring.Get(testValues[i], Write, bufDescs, bufHosts, bufZones)
				assert.NoError(t, err)
				assert.Equal(t, testData.replicationFactor, len(expectedSet.Instances))

				for range numOfInvocations {