Function TestLifecycler_TokenGenerator

ring/lifecycler_test.go:115–147

(t *testing.T)

}

func TestLifecycler_TokenGenerator(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

	cfg := testLifecyclerConfig(ringConfig, "instance-1")

	spreadMinimizingTokenGenerator, err := NewSpreadMinimizingTokenGenerator(cfg.ID, cfg.Zone, []string{zone(1), zone(2), zone(3)}, true)
	require.NoError(t, err)

	tests := []TokenGenerator{nil, initTokenGenerator(t), spreadMinimizingTokenGenerator}

	for _, testData := range tests {
		cfg.RingTokenGenerator = testData
		lifecycler, err := NewLifecycler(cfg, &nopFlushTransferer{}, "ingester", ringKey, true, log.NewNopLogger(), nil)
		require.NoError(t, err)
		if testData == nil {
			// If cfg.RingTokenGenerator is empty, RandomTokenGenerator is used
			tokenGenerator, ok := lifecycler.tokenGenerator.(*RandomTokenGenerator)
			require.True(t, ok)
			require.NotNil(t, tokenGenerator)
		} else {
			// If cfg.RingTokenGenerator is not empty, it is used
			require.NotNil(t, lifecycler.tokenGenerator)
			require.Equal(t, testData, lifecycler.tokenGenerator)
		}
	}
}

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