Function TestCleanUnhealthy

ring/client/pool_test.go:132–172

(t *testing.T)

}

func TestCleanUnhealthy(t *testing.T) {
	tcs := []struct {
		maxConcurrent int
	}{
		{maxConcurrent: 0}, // if not set, defaults to 16
		{maxConcurrent: 1},
	}
	for _, tc := range tcs {
		t.Run(fmt.Sprintf("max concurrent %d", tc.maxConcurrent), func(t *testing.T) {
			goodAddrs := []string{"good1", "good2"}
			badAddrs := []string{"bad1", "bad2"}
			clients := map[string]*poolMember{}
			for _, addr := range goodAddrs {
				clients[addr] = &poolMember{client: mockClient{happy: true, status: grpc_health_v1.HealthCheckResponse_SERVING}}
			}
			for _, addr := range badAddrs {
				clients[addr] = &poolMember{client: mockClient{happy: false, status: grpc_health_v1.HealthCheckResponse_NOT_SERVING}}
			}

			cfg := PoolConfig{
				MaxConcurrentHealthChecks: tc.maxConcurrent,
				CheckInterval:             1 * time.Second,
				HealthCheckTimeout:        5 * time.Millisecond,
			}
			pool := NewPool("test", cfg, nil, nil, nil, log.NewNopLogger())
			pool.members = clients
			pool.cleanUnhealthy()

			for _, addr := range badAddrs {
				if _, ok := pool.members[addr]; ok {
					t.Errorf("Found bad client after clean: %s\n", addr)
				}
			}
			for _, addr := range goodAddrs {
				if _, ok := pool.members[addr]; !ok {
					t.Errorf("Could not find good client after clean: %s\n", addr)
				}
			}
		})
	}
}

func TestCleanUnhealthyWithGracePeriod(t *testing.T) {
	const alwaysGood = "always-good"