package integration import ( "net/http" "net/http/httptest" "testing" "time" "github.com/stretchr/testify/require" "github.com/stretchr/testify/suite" "github.com/traefik/traefik/v2/integration/try" ) // ErrorPagesSuite test suites. type ErrorPagesSuite struct { BaseSuite ErrorPageIP string BackendIP string } func TestErrorPagesSuite(t *testing.T) { suite.Run(t, new(ErrorPagesSuite)) } func (s *ErrorPagesSuite) SetupSuite() { s.BaseSuite.SetupSuite() s.createComposeProject("error_pages") s.composeUp() s.ErrorPageIP = s.getComposeServiceIP("nginx2") s.BackendIP = s.getComposeServiceIP("nginx1") } func (s *ErrorPagesSuite) TearDownSuite() { s.BaseSuite.TearDownSuite() } func (s *ErrorPagesSuite) TestSimpleConfiguration() { file := s.adaptFile("fixtures/error_pages/simple.toml", struct { Server1 string Server2 string }{"http://" + s.BackendIP + ":80", s.ErrorPageIP}) s.traefikCmd(withConfigFile(file)) frontendReq, err := http.NewRequest(http.MethodGet, "http://127.0.0.1:8080", nil) require.NoError(s.T(), err) frontendReq.Host = "test.local" err = try.Request(frontendReq, 2*time.Second, try.BodyContains("nginx")) require.NoError(s.T(), err) } func (s *ErrorPagesSuite) TestErrorPage() { // error.toml contains a mis-configuration of the backend host file := s.adaptFile("fixtures/error_pages/error.toml", struct { Server1 string Server2 string }{s.BackendIP, s.ErrorPageIP}) s.traefikCmd(withConfigFile(file)) frontendReq, err := http.NewRequest(http.MethodGet, "http://127.0.0.1:8080", nil) require.NoError(s.T(), err) frontendReq.Host = "test.local" err = try.Request(frontendReq, 2*time.Second, try.BodyContains("An error occurred.")) require.NoError(s.T(), err) } func (s *ErrorPagesSuite) TestErrorPageFlush() { srv := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) { rw.Header().Add("Transfer-Encoding", "chunked") rw.WriteHeader(http.StatusInternalServerError) _, _ = rw.Write([]byte("KO")) })) file := s.adaptFile("fixtures/error_pages/simple.toml", struct { Server1 string Server2 string }{srv.URL, s.ErrorPageIP}) s.traefikCmd(withConfigFile(file)) frontendReq, err := http.NewRequest(http.MethodGet, "http://127.0.0.1:8080", nil) require.NoError(s.T(), err) frontendReq.Host = "test.local" err = try.Request(frontendReq, 2*time.Second, try.BodyContains("An error occurred."), try.HasHeaderValue("Content-Type", "text/html", true), ) require.NoError(s.T(), err) }