Function TestWaitBeforeJoining

ring/lifecycler_test.go:1733–1847

(t *testing.T)

}

func TestWaitBeforeJoining(t *testing.T) {
	t.Parallel()

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

	var ringConfig Config
	flagext.DefaultValues(&ringConfig)
	ringConfig.KVStore.Mock = ringStore

	r, err := New(ringConfig, "ingester", ringKey, log.NewNopLogger(), nil)
	require.NoError(t, err)
	require.NoError(t, services.StartAndAwaitRunning(context.Background(), r))
	defer services.StopAndAwaitTerminated(context.Background(), r) //nolint:errcheck

	err = r.KVClient.CAS(context.Background(), ringKey, func(interface{}) (interface{}, bool, error) {
		r := &Desc{
			Ingesters: map[string]InstanceDesc{
				instanceName(0, 1): {
					State:  ACTIVE,
					Tokens: []uint32{1, 2, 3},
				},
				instanceName(1, 1): {
					State:  ACTIVE,
					Tokens: []uint32{4, 5, 6},
				},
			},
		}

		return r, true, nil
	})
	require.NoError(t, err)

	targetZone := zone(1)
	spreadMinimizingZones := []string{zone(1), zone(2), zone(3)}
	canJoinTimeout := 5 * time.Second
	spreadMinimizingTokenGenerator := func(targetInstanceID string, canJoinEnabled bool) TokenGenerator {
		tokenGenerator, err := NewSpreadMinimizingTokenGenerator(targetInstanceID, targetZone, spreadMinimizingZones, canJoinEnabled)
		require.NoError(t, err)
		return tokenGenerator
	}

	tests := map[string]struct {
		targetInstanceID string
		tokenGenerator   TokenGenerator
		timeoutRequired  bool
		errorRequired    bool
	}{
		"RandomTokenGenerator never returns errors": {
			targetInstanceID: instanceName(3, 1),
			tokenGenerator:   initTokenGenerator(t),
		},
		"SpreadMinimizingTokenGenerator with CanJoinEnabled=false never returns errors": {
			targetInstanceID: instanceName(3, 1),
			tokenGenerator:   spreadMinimizingTokenGenerator(instanceName(2, 1), false),
		},
		"SpreadMinimizingTokenGenerator with CanJoinEnabled=true returns nil when the first instance joins": {
			targetInstanceID: instanceName(0, 1),