Function TestRing_ShuffleShard_Consistency

ring/ring_test.go:2131–2241

(t *testing.T)

}

func TestRing_ShuffleShard_Consistency(t *testing.T) {
	type change string

	type scenario struct {
		name         string
		numInstances int
		numZones     int
		shardSize    int
		ringChange   change
	}

	const (
		numTenants      = 100
		add             = change("add-instance")
		remove          = change("remove-instance")
		enableReadOnly  = change("enable-read-only")
		disableReadOnly = change("disable-read-only")
	)

	rnd := rand.New(rand.NewSource(time.Now().UnixNano()))

	// Generate all test scenarios.
	var scenarios []scenario
	for _, numInstances := range []int{20, 30, 40, 50} {
		for _, shardSize := range []int{3, 6, 9, 12, 15} {
			for _, c := range []change{add, remove, enableReadOnly, disableReadOnly} {
				scenarios = append(scenarios, scenario{
					name:         fmt.Sprintf("instances = %d, shard size = %d, ring operation = %s", numInstances, shardSize, c),
					numInstances: numInstances,
					numZones:     3,
					shardSize:    shardSize,
					ringChange:   c,
				})
			}
		}
	}

	for _, s := range scenarios {
		t.Run(s.name, func(t *testing.T) {
			gen := initTokenGenerator(t)
			// Initialise the ring.
			ringDesc := &Desc{Ingesters: generateRingInstances(gen, s.numInstances, s.numZones, 128)}

			// Mark some instances as read only
			for i := 0; i < len(ringDesc.Ingesters); i += 8 {
				instanceID := getRandomInstanceID(ringDesc.Ingesters, rnd)
				inst := ringDesc.Ingesters[instanceID]
				inst.ReadOnly = true
				inst.ReadOnlyUpdatedTimestamp = time.Now().Unix()
				ringDesc.Ingesters[instanceID] = inst
			}

			ring := newRingForTesting(Config{
				HeartbeatTimeout:     time.Hour,
				ZoneAwarenessEnabled: true,
			}, false)
			ring.setRingStateFromDesc(ringDesc, false, false, false)