(t *testing.T)
| 24 | ) |
| 25 | |
| 26 | func TestBasicLifecycler_GetTokenGenerator(t *testing.T) { |
| 27 | cfg := prepareBasicLifecyclerConfig() |
| 28 | |
| 29 | spreadMinimizingTokenGenerator, err := NewSpreadMinimizingTokenGenerator(cfg.ID, cfg.Zone, []string{zone(1), zone(2), zone(3)}, true) |
| 30 | require.NoError(t, err) |
| 31 | |
| 32 | tests := []TokenGenerator{nil, NewRandomTokenGenerator(), spreadMinimizingTokenGenerator} |
| 33 | |
| 34 | for _, testData := range tests { |
| 35 | cfg.RingTokenGenerator = testData |
| 36 | lifecycler, _, _, err := prepareBasicLifecycler(t, cfg) |
| 37 | require.NoError(t, err) |
| 38 | if testData == nil { |
| 39 | // If cfg.RingTokenGenerator is empty, RandomTokenGenerator is used |
| 40 | tokenGenerator, ok := lifecycler.tokenGenerator.(*RandomTokenGenerator) |
| 41 | require.True(t, ok) |
| 42 | require.NotNil(t, tokenGenerator) |
| 43 | } else { |
| 44 | // If cfg.RingTokenGenerator is not empty, it is used |
| 45 | require.NotNil(t, lifecycler.tokenGenerator) |
| 46 | require.Equal(t, testData, lifecycler.tokenGenerator) |
| 47 | } |
| 48 | } |
| 49 | } |
| 50 | |
| 51 | func TestBasicLifecycler_RegisterOnStart(t *testing.T) { |
| 52 | tests := map[string]struct { |
nothing calls this directly
no test coverage detected