Method TestServeHTTP

server/router_test.go:160–180

TestServeHTTP tests ServeHTTP method

()

// TestServeHTTP tests ServeHTTP method
func (s *RouterTestSuite) TestServeHTTP() {
	handler := func(reqID string, rw server.ResponseWriter, req *http.Request) *server.Error {
		rw.Header().Set("Custom-Header", "test-value")
		rw.WriteHeader(http.StatusOK)
		rw.Write([]byte("success"))
		return nil
	}

	s.router.GET("/test", handler)

	req := httptest.NewRequest(http.MethodGet, "/api/test", nil)
	rw := httptest.NewRecorder()

	s.router.ServeHTTP(rw, req)

	s.Require().Equal(http.StatusOK, rw.Code)
	s.Require().Equal("success", rw.Body.String())
	s.Require().Equal("test-value", rw.Header().Get("Custom-Header"))
	s.Require().Equal(server.DefaultServerName, rw.Header().Get(httpheaders.Server))
	s.Require().NotEmpty(rw.Header().Get(httpheaders.XRequestID))
}

// TestRequestID checks request ID generation and validation
func (s *RouterTestSuite) TestRequestID() {