Function TestLifecycler_StartingWithReadOnlyInstanceInRing

ring/lifecycler_test.go:1071–1111

(t *testing.T)

}

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

	ctx := context.Background()

	createLifecyclerFn := func(id string) *Lifecycler {
		// Add the first ingester to the ring
		lifecyclerConfig1 := testLifecyclerConfig(ringConfig, id)
		lifecyclerConfig1.HeartbeatPeriod = 100 * time.Millisecond
		lifecyclerConfig1.JoinAfter = 100 * time.Millisecond

		lifecycler, err := NewLifecycler(lifecyclerConfig1, &nopFlushTransferer{}, "ingester", ringKey, true, log.NewNopLogger(), nil)
		require.NoError(t, err)
		assert.Equal(t, 0, lifecycler.HealthyInstancesCount())
		require.NoError(t, services.StartAndAwaitRunning(ctx, lifecycler))
		t.Cleanup(func() {
			assert.NoError(t, services.StopAndAwaitTerminated(ctx, lifecycler))
		})
		return lifecycler
	}

	lifecycler1 := createLifecyclerFn("ing1")
	require.NoError(t, lifecycler1.ChangeReadOnlyState(ctx, true))
	// Assert the ingester has changed to read only
	test.Poll(t, time.Second, true, func() interface{} {
		return lifecycler1.HealthyInstancesCount() == 1 && lifecycler1.ReadOnlyInstancesCount() == 1
	})

	// Assert the second ingester joins the ring and sees the first ingester as read only
	lifecycler2 := createLifecyclerFn("ing2")
	test.Poll(t, time.Second, true, func() interface{} {
		return lifecycler2.HealthyInstancesCount() == 2 && lifecycler2.ReadOnlyInstancesCount() == 1
	})
}

// Test Lifecycler when decreasing tokens and instance is already in the ring in leaving state.
func TestLifecycler_DecreasingTokensLeavingInstanceInTheRing(t *testing.T) {