package healthcheck import ( "context" "net/http" "net/http/httptest" "net/url" "sync" "testing" "time" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" ptypes "github.com/traefik/paerser/types" "github.com/traefik/traefik/v3/pkg/config/dynamic" "github.com/traefik/traefik/v3/pkg/config/runtime" "github.com/traefik/traefik/v3/pkg/testhelpers" healthpb "google.golang.org/grpc/health/grpc_health_v1" ) func TestServiceHealthChecker_newRequest(t *testing.T) { testCases := []struct { desc string targetURL string config dynamic.ServerHealthCheck expTarget string expError bool expHostname string expHeader string expMethod string }{ { desc: "no port override", targetURL: "http://backend1:80", config: dynamic.ServerHealthCheck{ Path: "/test", Port: 0, }, expError: false, expTarget: "http://backend1:80/test", expHostname: "backend1:80", expMethod: http.MethodGet, }, { desc: "port override", targetURL: "http://backend2:80", config: dynamic.ServerHealthCheck{ Path: "/test", Port: 8080, }, expError: false, expTarget: "http://backend2:8080/test", expHostname: "backend2:8080", expMethod: http.MethodGet, }, { desc: "no port override with no port in server URL", targetURL: "http://backend1", config: dynamic.ServerHealthCheck{ Path: "/health", Port: 0, }, expError: false, expTarget: "http://backend1/health", expHostname: "backend1", expMethod: http.MethodGet, }, { desc: "port override with no port in server URL", targetURL: "http://backend2", config: dynamic.ServerHealthCheck{ Path: "/health", Port: 8080, }, expError: false, expTarget: "http://backend2:8080/health", expHostname: "backend2:8080", expMethod: http.MethodGet, }, { desc: "scheme override", targetURL: "https://backend1:80", config: dynamic.ServerHealthCheck{ Scheme: "http", Path: "/test", Port: 0, }, expError: false, expTarget: "http://backend1:80/test", expHostname: "backend1:80", expMethod: http.MethodGet, }, { desc: "path with param", targetURL: "http://backend1:80", config: dynamic.ServerHealthCheck{ Path: "/health?powpow=do", Port: 0, }, expError: false, expTarget: "http://backend1:80/health?powpow=do", expHostname: "backend1:80", expMethod: http.MethodGet, }, { desc: "path with params", targetURL: "http://backend1:80", config: dynamic.ServerHealthCheck{ Path: "/health?powpow=do&do=powpow", Port: 0, }, expError: false, expTarget: "http://backend1:80/health?powpow=do&do=powpow", expHostname: "backend1:80", expMethod: http.MethodGet, }, { desc: "path with invalid path", targetURL: "http://backend1:80", config: dynamic.ServerHealthCheck{ Path: ":", Port: 0, }, expError: true, expTarget: "", expHostname: "backend1:80", expMethod: http.MethodGet, }, { desc: "override hostname", targetURL: "http://backend1:80", config: dynamic.ServerHealthCheck{ Hostname: "myhost", Path: "/", }, expTarget: "http://backend1:80/", expHostname: "myhost", expHeader: "", expMethod: http.MethodGet, }, { desc: "not override hostname", targetURL: "http://backend1:80", config: dynamic.ServerHealthCheck{ Hostname: "", Path: "/", }, expTarget: "http://backend1:80/", expHostname: "backend1:80", expHeader: "", expMethod: http.MethodGet, }, { desc: "custom header", targetURL: "http://backend1:80", config: dynamic.ServerHealthCheck{ Headers: map[string]string{"Custom-Header": "foo"}, Hostname: "", Path: "/", }, expTarget: "http://backend1:80/", expHostname: "backend1:80", expHeader: "foo", expMethod: http.MethodGet, }, { desc: "custom header with hostname override", targetURL: "http://backend1:80", config: dynamic.ServerHealthCheck{ Headers: map[string]string{"Custom-Header": "foo"}, Hostname: "myhost", Path: "/", }, expTarget: "http://backend1:80/", expHostname: "myhost", expHeader: "foo", expMethod: http.MethodGet, }, { desc: "custom method", targetURL: "http://backend1:80", config: dynamic.ServerHealthCheck{ Path: "/", Method: http.MethodHead, }, expTarget: "http://backend1:80/", expHostname: "backend1:80", expMethod: http.MethodHead, }, } for _, test := range testCases { test := test t.Run(test.desc, func(t *testing.T) { t.Parallel() shc := ServiceHealthChecker{config: &test.config} u := testhelpers.MustParseURL(test.targetURL) req, err := shc.newRequest(context.Background(), u) if test.expError { require.Error(t, err) assert.Nil(t, req) } else { require.NoError(t, err, "failed to create new request") require.NotNil(t, req) assert.Equal(t, test.expTarget, req.URL.String()) assert.Equal(t, test.expHeader, req.Header.Get("Custom-Header")) assert.Equal(t, test.expHostname, req.Host) assert.Equal(t, test.expMethod, req.Method) } }) } } func TestServiceHealthChecker_checkHealthHTTP_NotFollowingRedirects(t *testing.T) { redirectServerCalled := false redirectTestServer := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) { redirectServerCalled = true })) defer redirectTestServer.Close() ctx, cancel := context.WithTimeout(context.Background(), time.Duration(dynamic.DefaultHealthCheckTimeout)) defer cancel() server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) { rw.Header().Add("location", redirectTestServer.URL) rw.WriteHeader(http.StatusSeeOther) })) defer server.Close() config := &dynamic.ServerHealthCheck{ Path: "/path", FollowRedirects: Bool(false), Interval: dynamic.DefaultHealthCheckInterval, Timeout: dynamic.DefaultHealthCheckTimeout, } healthChecker := NewServiceHealthChecker(ctx, nil, config, nil, nil, http.DefaultTransport, nil) err := healthChecker.checkHealthHTTP(ctx, testhelpers.MustParseURL(server.URL)) require.NoError(t, err) assert.False(t, redirectServerCalled, "HTTP redirect must not be followed") } func TestServiceHealthChecker_Launch(t *testing.T) { testCases := []struct { desc string mode string status int server StartTestServer expNumRemovedServers int expNumUpsertedServers int expGaugeValue float64 targetStatus string }{ { desc: "healthy server staying healthy", server: newHTTPServer(http.StatusOK), expNumRemovedServers: 0, expNumUpsertedServers: 1, expGaugeValue: 1, targetStatus: runtime.StatusUp, }, { desc: "healthy server staying healthy, with custom code status check", server: newHTTPServer(http.StatusNotFound), status: http.StatusNotFound, expNumRemovedServers: 0, expNumUpsertedServers: 1, expGaugeValue: 1, targetStatus: runtime.StatusUp, }, { desc: "healthy server staying healthy (StatusNoContent)", server: newHTTPServer(http.StatusNoContent), expNumRemovedServers: 0, expNumUpsertedServers: 1, expGaugeValue: 1, targetStatus: runtime.StatusUp, }, { desc: "healthy server staying healthy (StatusPermanentRedirect)", server: newHTTPServer(http.StatusPermanentRedirect), expNumRemovedServers: 0, expNumUpsertedServers: 1, expGaugeValue: 1, targetStatus: runtime.StatusUp, }, { desc: "healthy server becoming sick", server: newHTTPServer(http.StatusServiceUnavailable), expNumRemovedServers: 1, expNumUpsertedServers: 0, expGaugeValue: 0, targetStatus: runtime.StatusDown, }, { desc: "healthy server becoming sick, with custom code status check", server: newHTTPServer(http.StatusOK), status: http.StatusServiceUnavailable, expNumRemovedServers: 1, expNumUpsertedServers: 0, expGaugeValue: 0, targetStatus: runtime.StatusDown, }, { desc: "healthy server toggling to sick and back to healthy", server: newHTTPServer(http.StatusServiceUnavailable, http.StatusOK), expNumRemovedServers: 1, expNumUpsertedServers: 1, expGaugeValue: 1, targetStatus: runtime.StatusUp, }, { desc: "healthy server toggling to healthy and go to sick", server: newHTTPServer(http.StatusOK, http.StatusServiceUnavailable), expNumRemovedServers: 1, expNumUpsertedServers: 1, expGaugeValue: 0, targetStatus: runtime.StatusDown, }, { desc: "healthy grpc server staying healthy", mode: "grpc", server: newGRPCServer(healthpb.HealthCheckResponse_SERVING), expNumRemovedServers: 0, expNumUpsertedServers: 1, expGaugeValue: 1, targetStatus: runtime.StatusUp, }, { desc: "healthy grpc server becoming sick", mode: "grpc", server: newGRPCServer(healthpb.HealthCheckResponse_NOT_SERVING), expNumRemovedServers: 1, expNumUpsertedServers: 0, expGaugeValue: 0, targetStatus: runtime.StatusDown, }, { desc: "healthy grpc server toggling to sick and back to healthy", mode: "grpc", server: newGRPCServer(healthpb.HealthCheckResponse_NOT_SERVING, healthpb.HealthCheckResponse_SERVING), expNumRemovedServers: 1, expNumUpsertedServers: 1, expGaugeValue: 1, targetStatus: runtime.StatusUp, }, } for _, test := range testCases { test := test t.Run(test.desc, func(t *testing.T) { t.Parallel() // The context is passed to the health check and // canonically canceled by the test server once all expected requests have been received. ctx, cancel := context.WithCancel(context.Background()) t.Cleanup(cancel) targetURL, timeout := test.server.Start(t, cancel) lb := &testLoadBalancer{RWMutex: &sync.RWMutex{}} config := &dynamic.ServerHealthCheck{ Mode: test.mode, Status: test.status, Path: "/path", Interval: ptypes.Duration(500 * time.Millisecond), Timeout: ptypes.Duration(499 * time.Millisecond), } gauge := &testhelpers.CollectingGauge{} serviceInfo := &runtime.ServiceInfo{} hc := NewServiceHealthChecker(ctx, &MetricsMock{gauge}, config, lb, serviceInfo, http.DefaultTransport, map[string]*url.URL{"test": targetURL}) wg := sync.WaitGroup{} wg.Add(1) go func() { hc.Launch(ctx) wg.Done() }() select { case <-time.After(timeout): t.Fatal("test did not complete in time") case <-ctx.Done(): wg.Wait() } lb.Lock() defer lb.Unlock() assert.Equal(t, test.expNumRemovedServers, lb.numRemovedServers, "removed servers") assert.Equal(t, test.expNumUpsertedServers, lb.numUpsertedServers, "upserted servers") assert.Equal(t, test.expGaugeValue, gauge.GaugeValue, "ServerUp Gauge") assert.Equal(t, serviceInfo.GetAllStatus(), map[string]string{targetURL.String(): test.targetStatus}) }) } } func Bool(b bool) *bool { return &b }