Function prepare

pkg/distributor/distributor_test.go:3271–3434

(tb testing.TB, cfg prepConfig)

}

func prepare(tb testing.TB, cfg prepConfig) ([]*Distributor, []*mockIngester, []*prometheus.Registry, *ring.Ring) {
	// Strings to be used for get labels values/Names
	var unusedStrings []string
	if cfg.lblValuesPerIngester > 0 {
		unusedStrings = make([]string, min(len(randomStrings), cfg.numIngesters*cfg.lblValuesPerIngester))
		copy(unusedStrings, randomStrings)
	}
	s := &prepState{
		unusedStrings: unusedStrings,
	}
	ingesters := []*mockIngester{}
	for i := 0; i < cfg.happyIngesters; i++ {
		ingesters = append(ingesters, newMockIngester(i, s, cfg))
	}
	for i := cfg.happyIngesters; i < cfg.numIngesters; i++ {
		miError := errFail
		if cfg.errFail != nil {
			miError = cfg.errFail
		}

		ingesters = append(ingesters, &mockIngester{
			queryDelay: cfg.queryDelay,
			failResp:   *atomic.NewError(miError),
		})
	}

	// Use a real ring with a mock KV store to test ring RF logic.
	ingesterDescs := map[string]ring.InstanceDesc{}
	ingestersByAddr := map[string]*mockIngester{}
	for i := range ingesters {
		var tokens []uint32
		if len(cfg.tokens) > i {
			tokens = cfg.tokens[i]
		} else {
			tokens = []uint32{uint32((math.MaxUint32 / cfg.numIngesters) * i)}
		}
		ingester := fmt.Sprintf("ingester-%d", i)
		addr := fmt.Sprintf("ip-ingester-%d", i)
		ingesterDescs[ingester] = ring.InstanceDesc{
			Addr:                addr,
			Zone:                "",
			State:               ring.ACTIVE,
			Timestamp:           time.Now().Unix(),
			RegisteredTimestamp: time.Now().Add(-2 * time.Hour).Unix(),
			Tokens:              tokens,
		}
		ingestersByAddr[addr] = ingesters[i]
	}

	kvStore, closer := consul.NewInMemoryClient(ring.GetCodec(), log.NewNopLogger(), nil)
	tb.Cleanup(func() { assert.NoError(tb, closer.Close()) })

	err := kvStore.CAS(context.Background(), ingester.RingKey,
		func(_ any) (any, bool, error) {
			return &ring.Desc{
				Ingesters: ingesterDescs,
			}, true, nil
		},