Function TestCleanUnhealthyWithGracePeriod

ring/client/pool_test.go:174–214

(t *testing.T)

}

func TestCleanUnhealthyWithGracePeriod(t *testing.T) {
	const alwaysGood = "always-good"
	const flapping = "flapping"
	const neverGood = "never-good"
	flappingClient := &mockClient{happy: true, status: grpc_health_v1.HealthCheckResponse_NOT_SERVING}

	clients := map[string]*poolMember{
		alwaysGood: {client: &mockClient{happy: true, status: grpc_health_v1.HealthCheckResponse_SERVING}},
		flapping:   {client: flappingClient},
		neverGood:  {client: &mockClient{happy: true, status: grpc_health_v1.HealthCheckResponse_NOT_SERVING}},
	}

	cfg := PoolConfig{
		CheckInterval:          500 * time.Millisecond,
		HealthCheckTimeout:     5 * time.Millisecond,
		HealthCheckGracePeriod: time.Second,
	}

	pool := NewPool("test", cfg, nil, nil, nil, log.NewNopLogger())
	pool.members = clients

	pool.cleanUnhealthy()
	require.ElementsMatch(t, []string{alwaysGood, flapping, neverGood}, pool.RegisteredAddresses(), "no clients should be removed before minimum failure period has expired")

	time.Sleep(cfg.HealthCheckGracePeriod / 2)
	pool.cleanUnhealthy()
	require.ElementsMatch(t, []string{alwaysGood, flapping, neverGood}, pool.RegisteredAddresses(), "no clients should be removed before minimum failure period has expired, even after a subsequent failure")

	flappingClient.status = grpc_health_v1.HealthCheckResponse_SERVING
	time.Sleep((cfg.HealthCheckGracePeriod / 2) + time.Millisecond)
	pool.cleanUnhealthy()
	require.ElementsMatch(t, []string{alwaysGood, flapping}, pool.RegisteredAddresses(), "only client that has consistently failed all health checks during minimum failure period should be removed")

	flappingClient.status = grpc_health_v1.HealthCheckResponse_NOT_SERVING
	pool.cleanUnhealthy()
	require.ElementsMatch(t, []string{alwaysGood, flapping}, pool.RegisteredAddresses(), "no clients should be removed if they have just become unhealthy")

	time.Sleep(cfg.HealthCheckGracePeriod + time.Millisecond)
	pool.cleanUnhealthy()
	require.ElementsMatch(t, []string{alwaysGood}, pool.RegisteredAddresses(), "client should be removed if still unhealthy after minimum failure period has expired")
}

func TestRemoveClient(t *testing.T) {
	const (