2020-05-07 19:28:21 +02:00
|
|
|
// ⚡️ Fiber is an Express inspired web framework written in Go with ☕️
|
2020-05-07 20:22:26 +02:00
|
|
|
// 🤖 Github Repository: https://github.com/gofiber/fiber
|
2020-05-07 19:28:21 +02:00
|
|
|
// 📌 API Documentation: https://docs.gofiber.io
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-02-21 01:54:50 +01:00
|
|
|
package fiber
|
|
|
|
|
|
|
|
import (
|
2020-07-29 08:46:36 +08:00
|
|
|
"bytes"
|
2020-06-06 07:27:01 +02:00
|
|
|
"errors"
|
2020-06-07 14:43:25 +02:00
|
|
|
"fmt"
|
2020-07-16 16:42:36 +08:00
|
|
|
"io"
|
2021-07-11 11:45:46 +02:00
|
|
|
"mime/multipart"
|
2020-07-15 18:50:28 +02:00
|
|
|
"net"
|
2020-07-29 08:46:36 +08:00
|
|
|
"net/http"
|
2020-05-07 19:28:21 +02:00
|
|
|
"net/http/httptest"
|
2020-07-15 10:20:53 +08:00
|
|
|
"reflect"
|
2020-06-07 14:43:25 +02:00
|
|
|
"regexp"
|
2022-08-08 14:18:57 +08:00
|
|
|
"runtime"
|
2020-05-29 10:22:54 +02:00
|
|
|
"strings"
|
2020-02-21 01:54:50 +01:00
|
|
|
"testing"
|
2020-04-13 09:01:27 +02:00
|
|
|
"time"
|
2020-05-23 09:25:49 +02:00
|
|
|
|
2022-08-22 13:57:10 +08:00
|
|
|
"github.com/stretchr/testify/require"
|
2022-08-15 18:27:06 +03:00
|
|
|
"github.com/valyala/fasthttp"
|
2020-02-21 01:54:50 +01:00
|
|
|
)
|
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
var testEmptyHandler = func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
func testStatus200(t *testing.T, app *App, url string, method string) {
|
2021-11-05 08:00:03 +01:00
|
|
|
t.Helper()
|
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
req := httptest.NewRequest(method, url, nil)
|
2020-02-21 01:54:50 +01:00
|
|
|
|
2020-02-21 18:07:43 +01:00
|
|
|
resp, err := app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
2020-02-21 18:07:43 +01:00
|
|
|
}
|
2020-05-07 19:28:21 +02:00
|
|
|
|
2022-04-05 08:40:22 +02:00
|
|
|
func testErrorResponse(t *testing.T, err error, resp *http.Response, expectedBodyError string) {
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 500, resp.StatusCode, "Status code")
|
2022-04-05 08:40:22 +02:00
|
|
|
|
2022-06-01 18:34:22 +03:00
|
|
|
body, err := io.ReadAll(resp.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, expectedBodyError, string(body), "Response body")
|
2022-04-05 08:40:22 +02:00
|
|
|
}
|
|
|
|
|
2020-06-20 17:26:48 +02:00
|
|
|
func Test_App_MethodNotAllowed(t *testing.T) {
|
|
|
|
app := New()
|
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Use(func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
return c.Next()
|
|
|
|
})
|
2020-07-07 23:25:02 +08:00
|
|
|
|
2020-09-13 11:20:11 +02:00
|
|
|
app.Post("/", testEmptyHandler)
|
2020-06-20 17:26:48 +02:00
|
|
|
|
2020-09-13 11:20:11 +02:00
|
|
|
app.Options("/", testEmptyHandler)
|
2020-07-10 14:15:58 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodPost, "/", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, 200, resp.StatusCode)
|
|
|
|
require.Equal(t, "", resp.Header.Get(HeaderAllow))
|
2020-07-17 09:34:27 +08:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, 405, resp.StatusCode)
|
|
|
|
require.Equal(t, "POST, OPTIONS", resp.Header.Get(HeaderAllow))
|
2020-06-20 17:26:48 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err = app.Test(httptest.NewRequest(MethodPatch, "/", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, 405, resp.StatusCode)
|
|
|
|
require.Equal(t, "POST, OPTIONS", resp.Header.Get(HeaderAllow))
|
2020-06-20 17:26:48 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err = app.Test(httptest.NewRequest(MethodPut, "/", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, 405, resp.StatusCode)
|
|
|
|
require.Equal(t, "POST, OPTIONS", resp.Header.Get(HeaderAllow))
|
2020-06-21 11:02:17 +02:00
|
|
|
|
2020-09-13 11:20:11 +02:00
|
|
|
app.Get("/", testEmptyHandler)
|
2020-06-21 11:02:17 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err = app.Test(httptest.NewRequest(MethodTrace, "/", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, 405, resp.StatusCode)
|
2022-09-05 17:41:25 +03:00
|
|
|
require.Equal(t, "GET, POST, OPTIONS", resp.Header.Get(HeaderAllow))
|
2020-06-21 11:02:17 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err = app.Test(httptest.NewRequest(MethodPatch, "/", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, 405, resp.StatusCode)
|
2022-09-05 17:41:25 +03:00
|
|
|
require.Equal(t, "GET, POST, OPTIONS", resp.Header.Get(HeaderAllow))
|
|
|
|
|
|
|
|
app.Head("/", testEmptyHandler)
|
2020-06-21 11:02:17 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err = app.Test(httptest.NewRequest(MethodPut, "/", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, 405, resp.StatusCode)
|
|
|
|
require.Equal(t, "GET, HEAD, POST, OPTIONS", resp.Header.Get(HeaderAllow))
|
2020-06-20 17:26:48 +02:00
|
|
|
}
|
|
|
|
|
2020-07-08 22:40:44 +08:00
|
|
|
func Test_App_Custom_Middleware_404_Should_Not_SetMethodNotAllowed(t *testing.T) {
|
|
|
|
app := New()
|
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Use(func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
return c.SendStatus(404)
|
2020-07-08 22:40:44 +08:00
|
|
|
})
|
|
|
|
|
2020-09-13 11:20:11 +02:00
|
|
|
app.Post("/", testEmptyHandler)
|
2020-07-08 22:40:44 +08:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, 404, resp.StatusCode)
|
2020-07-08 22:40:44 +08:00
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
g := app.Group("/with-next", func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
return c.Status(404).Next()
|
2020-07-08 22:40:44 +08:00
|
|
|
})
|
|
|
|
|
2020-09-13 11:20:11 +02:00
|
|
|
g.Post("/", testEmptyHandler)
|
2020-07-08 22:40:44 +08:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/with-next", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, 404, resp.StatusCode)
|
2020-07-08 22:40:44 +08:00
|
|
|
}
|
|
|
|
|
2020-06-07 14:43:25 +02:00
|
|
|
func Test_App_ServerErrorHandler_SmallReadBuffer(t *testing.T) {
|
|
|
|
expectedError := regexp.MustCompile(
|
|
|
|
`error when reading request headers: small read buffer\. Increase ReadBufferSize\. Buffer size=4096, contents: "GET / HTTP/1.1\\r\\nHost: example\.com\\r\\nVery-Long-Header: -+`,
|
|
|
|
)
|
|
|
|
app := New()
|
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Get("/", func(c Ctx) error {
|
2020-06-08 02:55:19 +02:00
|
|
|
panic(errors.New("should never called"))
|
2020-06-07 14:43:25 +02:00
|
|
|
})
|
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
request := httptest.NewRequest(MethodGet, "/", nil)
|
2020-10-03 13:20:54 -04:00
|
|
|
logHeaderSlice := make([]string, 5000)
|
2020-06-07 14:43:25 +02:00
|
|
|
request.Header.Set("Very-Long-Header", strings.Join(logHeaderSlice, "-"))
|
|
|
|
_, err := app.Test(request)
|
|
|
|
|
|
|
|
if err == nil {
|
|
|
|
t.Error("Expect an error at app.Test(request)")
|
|
|
|
}
|
|
|
|
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Regexp(t, expectedError, err.Error())
|
2020-06-07 14:43:25 +02:00
|
|
|
}
|
|
|
|
|
2020-09-13 11:20:11 +02:00
|
|
|
func Test_App_Errors(t *testing.T) {
|
|
|
|
app := New(Config{
|
2020-07-15 10:20:53 +08:00
|
|
|
BodyLimit: 4,
|
|
|
|
})
|
2020-06-03 17:16:10 +02:00
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Get("/", func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
return errors.New("hi, i'm an error")
|
2020-06-06 07:27:01 +02:00
|
|
|
})
|
2020-06-03 17:16:10 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 500, resp.StatusCode, "Status code")
|
2020-06-03 17:16:10 +02:00
|
|
|
|
2022-06-01 18:34:22 +03:00
|
|
|
body, err := io.ReadAll(resp.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, "hi, i'm an error", string(body))
|
2020-06-03 17:16:10 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
_, err = app.Test(httptest.NewRequest(MethodGet, "/", strings.NewReader("big body")))
|
2020-07-15 10:20:53 +08:00
|
|
|
if err != nil {
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "body size exceeds the given limit", err.Error(), "app.Test(req)")
|
2020-07-15 10:20:53 +08:00
|
|
|
}
|
2020-06-06 07:27:01 +02:00
|
|
|
}
|
2020-06-03 17:16:10 +02:00
|
|
|
|
2020-06-06 07:27:01 +02:00
|
|
|
func Test_App_ErrorHandler_Custom(t *testing.T) {
|
2020-09-13 11:20:11 +02:00
|
|
|
app := New(Config{
|
2022-07-13 08:48:29 +03:00
|
|
|
ErrorHandler: func(c Ctx, err error) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
return c.Status(200).SendString("hi, i'm an custom error")
|
2020-06-06 07:27:01 +02:00
|
|
|
},
|
|
|
|
})
|
2020-06-03 17:16:10 +02:00
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Get("/", func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
return errors.New("hi, i'm an error")
|
2020-06-06 07:27:01 +02:00
|
|
|
})
|
2020-06-03 17:16:10 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
2020-06-03 17:16:10 +02:00
|
|
|
|
2022-06-01 18:34:22 +03:00
|
|
|
body, err := io.ReadAll(resp.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, "hi, i'm an custom error", string(body))
|
2020-06-06 07:27:01 +02:00
|
|
|
}
|
2020-06-03 17:16:10 +02:00
|
|
|
|
2020-09-18 20:44:00 +02:00
|
|
|
func Test_App_ErrorHandler_HandlerStack(t *testing.T) {
|
|
|
|
app := New(Config{
|
2022-07-13 08:48:29 +03:00
|
|
|
ErrorHandler: func(c Ctx, err error) error {
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "1: USE error", err.Error())
|
2020-09-18 20:44:00 +02:00
|
|
|
return DefaultErrorHandler(c, err)
|
|
|
|
},
|
|
|
|
})
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Use("/", func(c Ctx) error {
|
2020-09-18 20:44:00 +02:00
|
|
|
err := c.Next() // call next USE
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "2: USE error", err.Error())
|
2020-09-18 20:44:00 +02:00
|
|
|
return errors.New("1: USE error")
|
2022-07-13 08:48:29 +03:00
|
|
|
}, func(c Ctx) error {
|
2020-09-18 20:44:00 +02:00
|
|
|
err := c.Next() // call [0] GET
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "0: GET error", err.Error())
|
2020-09-18 20:44:00 +02:00
|
|
|
return errors.New("2: USE error")
|
|
|
|
})
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Get("/", func(c Ctx) error {
|
2020-09-18 20:44:00 +02:00
|
|
|
return errors.New("0: GET error")
|
|
|
|
})
|
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 500, resp.StatusCode, "Status code")
|
2020-09-18 20:44:00 +02:00
|
|
|
|
2022-06-01 18:34:22 +03:00
|
|
|
body, err := io.ReadAll(resp.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, "1: USE error", string(body))
|
2020-09-18 20:44:00 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
func Test_App_ErrorHandler_RouteStack(t *testing.T) {
|
|
|
|
app := New(Config{
|
2022-07-13 08:48:29 +03:00
|
|
|
ErrorHandler: func(c Ctx, err error) error {
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "1: USE error", err.Error())
|
2020-09-18 20:44:00 +02:00
|
|
|
return DefaultErrorHandler(c, err)
|
|
|
|
},
|
|
|
|
})
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Use("/", func(c Ctx) error {
|
2020-09-18 20:44:00 +02:00
|
|
|
err := c.Next()
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "0: GET error", err.Error())
|
2020-09-18 20:44:00 +02:00
|
|
|
return errors.New("1: USE error") // [2] call ErrorHandler
|
|
|
|
})
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Get("/test", func(c Ctx) error {
|
2020-09-18 20:44:00 +02:00
|
|
|
return errors.New("0: GET error") // [1] return to USE
|
|
|
|
})
|
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/test", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 500, resp.StatusCode, "Status code")
|
2020-09-18 20:44:00 +02:00
|
|
|
|
2022-06-01 18:34:22 +03:00
|
|
|
body, err := io.ReadAll(resp.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, "1: USE error", string(body))
|
2020-09-18 20:44:00 +02:00
|
|
|
}
|
|
|
|
|
2021-12-28 21:05:35 +08:00
|
|
|
func Test_App_ErrorHandler_GroupMount(t *testing.T) {
|
|
|
|
micro := New(Config{
|
2022-07-13 08:48:29 +03:00
|
|
|
ErrorHandler: func(c Ctx, err error) error {
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "0: GET error", err.Error())
|
2021-12-28 21:05:35 +08:00
|
|
|
return c.Status(500).SendString("1: custom error")
|
|
|
|
},
|
|
|
|
})
|
2022-07-13 08:48:29 +03:00
|
|
|
micro.Get("/doe", func(c Ctx) error {
|
2021-12-28 21:05:35 +08:00
|
|
|
return errors.New("0: GET error")
|
|
|
|
})
|
|
|
|
|
|
|
|
app := New()
|
|
|
|
v1 := app.Group("/v1")
|
|
|
|
v1.Mount("/john", micro)
|
|
|
|
|
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/v1/john/doe", nil))
|
2022-04-05 08:40:22 +02:00
|
|
|
testErrorResponse(t, err, resp, "1: custom error")
|
|
|
|
}
|
2021-12-28 21:05:35 +08:00
|
|
|
|
2022-04-05 08:40:22 +02:00
|
|
|
func Test_App_ErrorHandler_GroupMountRootLevel(t *testing.T) {
|
|
|
|
micro := New(Config{
|
2022-07-13 08:48:29 +03:00
|
|
|
ErrorHandler: func(c Ctx, err error) error {
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "0: GET error", err.Error())
|
2022-04-05 08:40:22 +02:00
|
|
|
return c.Status(500).SendString("1: custom error")
|
|
|
|
},
|
|
|
|
})
|
2022-07-13 08:48:29 +03:00
|
|
|
micro.Get("/john/doe", func(c Ctx) error {
|
2022-04-05 08:40:22 +02:00
|
|
|
return errors.New("0: GET error")
|
|
|
|
})
|
|
|
|
|
|
|
|
app := New()
|
|
|
|
v1 := app.Group("/v1")
|
|
|
|
v1.Mount("/", micro)
|
|
|
|
|
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/v1/john/doe", nil))
|
|
|
|
testErrorResponse(t, err, resp, "1: custom error")
|
2021-12-28 21:05:35 +08:00
|
|
|
}
|
|
|
|
|
2020-05-12 19:24:04 +02:00
|
|
|
func Test_App_Nested_Params(t *testing.T) {
|
2020-05-11 13:42:42 +02:00
|
|
|
app := New()
|
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Get("/test", func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
return c.Status(400).Send([]byte("Should move on"))
|
2020-05-11 13:42:42 +02:00
|
|
|
})
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Get("/test/:param", func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
return c.Status(400).Send([]byte("Should move on"))
|
2020-05-11 13:42:42 +02:00
|
|
|
})
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Get("/test/:param/test", func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
return c.Status(400).Send([]byte("Should move on"))
|
2020-05-11 13:42:42 +02:00
|
|
|
})
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Get("/test/:param/test/:param2", func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
return c.Status(200).Send([]byte("Good job"))
|
2020-05-11 13:42:42 +02:00
|
|
|
})
|
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
req := httptest.NewRequest(MethodGet, "/test/john/test/doe", nil)
|
2020-05-11 13:42:42 +02:00
|
|
|
resp, err := app.Test(req)
|
|
|
|
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
2020-05-11 13:42:42 +02:00
|
|
|
}
|
|
|
|
|
2020-09-26 13:46:14 +02:00
|
|
|
// go test -run Test_App_Mount
|
|
|
|
func Test_App_Mount(t *testing.T) {
|
|
|
|
micro := New()
|
2022-07-13 08:48:29 +03:00
|
|
|
micro.Get("/doe", func(c Ctx) error {
|
2020-09-26 13:46:14 +02:00
|
|
|
return c.SendStatus(StatusOK)
|
|
|
|
})
|
|
|
|
|
|
|
|
app := New()
|
|
|
|
app.Mount("/john", micro)
|
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/john/doe", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
2022-09-05 17:41:25 +03:00
|
|
|
require.Equal(t, uint32(1), app.handlersCount)
|
2020-09-26 13:46:14 +02:00
|
|
|
}
|
2020-09-13 11:20:11 +02:00
|
|
|
|
2020-05-12 19:24:04 +02:00
|
|
|
func Test_App_Use_Params(t *testing.T) {
|
2020-05-09 15:15:34 +02:00
|
|
|
app := New()
|
2020-05-12 19:24:04 +02:00
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Use("/prefix/:param", func(c Ctx) error {
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "john", c.Params("param"))
|
2020-09-13 11:20:11 +02:00
|
|
|
return nil
|
2020-05-09 15:15:34 +02:00
|
|
|
})
|
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Use("/foo/:bar?", func(c Ctx) error {
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "foobar", c.Params("bar", "foobar"))
|
2020-09-13 11:20:11 +02:00
|
|
|
return nil
|
2020-07-04 10:11:23 +02:00
|
|
|
})
|
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Use("/:param/*", func(c Ctx) error {
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "john", c.Params("param"))
|
|
|
|
require.Equal(t, "doe", c.Params("*"))
|
2020-09-13 11:20:11 +02:00
|
|
|
return nil
|
2020-05-12 19:24:04 +02:00
|
|
|
})
|
2020-05-09 15:15:34 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/prefix/john", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
2020-05-12 19:24:04 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/john/doe", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
2020-07-04 10:11:23 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/foo", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
2020-07-15 17:43:30 +02:00
|
|
|
|
|
|
|
defer func() {
|
|
|
|
if err := recover(); err != nil {
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "use: invalid handler func()\n", fmt.Sprintf("%v", err))
|
2020-07-15 17:43:30 +02:00
|
|
|
}
|
|
|
|
}()
|
|
|
|
|
|
|
|
app.Use("/:param/*", func() {
|
|
|
|
// this should panic
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2021-01-06 13:21:54 +01:00
|
|
|
func Test_App_Use_UnescapedPath(t *testing.T) {
|
|
|
|
app := New(Config{UnescapePath: true, CaseSensitive: true})
|
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Use("/cRéeR/:param", func(c Ctx) error {
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "/cRéeR/اختبار", c.Path())
|
2021-01-06 13:21:54 +01:00
|
|
|
return c.SendString(c.Params("param"))
|
|
|
|
})
|
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Use("/abc", func(c Ctx) error {
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "/AbC", c.Path())
|
2021-01-06 13:46:50 +01:00
|
|
|
return nil
|
|
|
|
})
|
|
|
|
|
2021-01-06 13:52:34 +01:00
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/cR%C3%A9eR/%D8%A7%D8%AE%D8%AA%D8%A8%D8%A7%D8%B1", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, StatusOK, resp.StatusCode, "Status code")
|
2021-01-06 13:21:54 +01:00
|
|
|
|
2022-06-01 18:34:22 +03:00
|
|
|
body, err := io.ReadAll(resp.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
2021-01-06 13:21:54 +01:00
|
|
|
// check the param result
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "اختبار", app.getString(body))
|
2021-01-06 13:21:54 +01:00
|
|
|
|
|
|
|
// with lowercase letters
|
2021-01-06 13:52:34 +01:00
|
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/cr%C3%A9er/%D8%A7%D8%AE%D8%AA%D8%A8%D8%A7%D8%B1", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, StatusNotFound, resp.StatusCode, "Status code")
|
2021-01-06 13:21:54 +01:00
|
|
|
}
|
|
|
|
|
2021-01-06 13:46:50 +01:00
|
|
|
func Test_App_Use_CaseSensitive(t *testing.T) {
|
|
|
|
app := New(Config{CaseSensitive: true})
|
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Use("/abc", func(c Ctx) error {
|
2021-01-06 13:46:50 +01:00
|
|
|
return c.SendString(c.Path())
|
|
|
|
})
|
|
|
|
|
|
|
|
// wrong letters in the requested route -> 404
|
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/AbC", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, StatusNotFound, resp.StatusCode, "Status code")
|
2021-01-06 13:46:50 +01:00
|
|
|
|
|
|
|
// right letters in the requrested route -> 200
|
|
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/abc", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, StatusOK, resp.StatusCode, "Status code")
|
2021-01-06 13:46:50 +01:00
|
|
|
|
|
|
|
// check the detected path when the case insensitive recognition is activated
|
|
|
|
app.config.CaseSensitive = false
|
|
|
|
// check the case sensitive feature
|
|
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/AbC", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, StatusOK, resp.StatusCode, "Status code")
|
2021-01-06 13:46:50 +01:00
|
|
|
|
2022-06-01 18:34:22 +03:00
|
|
|
body, err := io.ReadAll(resp.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
2021-01-06 13:46:50 +01:00
|
|
|
// check the detected path result
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "/AbC", app.getString(body))
|
2021-01-06 13:46:50 +01:00
|
|
|
}
|
|
|
|
|
2020-07-15 17:43:30 +02:00
|
|
|
func Test_App_Add_Method_Test(t *testing.T) {
|
|
|
|
app := New()
|
|
|
|
defer func() {
|
|
|
|
if err := recover(); err != nil {
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "add: invalid http method JOHN\n", fmt.Sprintf("%v", err))
|
2020-07-15 17:43:30 +02:00
|
|
|
}
|
|
|
|
}()
|
2020-09-13 11:20:11 +02:00
|
|
|
app.Add("JOHN", "/doe", testEmptyHandler)
|
2020-05-12 19:24:04 +02:00
|
|
|
}
|
2020-05-13 20:21:49 +02:00
|
|
|
|
2020-09-13 11:20:11 +02:00
|
|
|
// go test -run Test_App_GETOnly
|
|
|
|
func Test_App_GETOnly(t *testing.T) {
|
|
|
|
app := New(Config{
|
|
|
|
GETOnly: true,
|
|
|
|
})
|
2020-07-15 18:50:28 +02:00
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Post("/", func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
return c.SendString("Hello 👋!")
|
|
|
|
})
|
2020-07-15 18:56:42 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
req := httptest.NewRequest(MethodPost, "/", nil)
|
2020-09-13 11:20:11 +02:00
|
|
|
resp, err := app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, StatusMethodNotAllowed, resp.StatusCode, "Status code")
|
2020-07-15 18:50:28 +02:00
|
|
|
}
|
2020-09-13 11:20:11 +02:00
|
|
|
|
2020-05-13 20:21:49 +02:00
|
|
|
func Test_App_Use_Params_Group(t *testing.T) {
|
|
|
|
app := New()
|
|
|
|
|
|
|
|
group := app.Group("/prefix/:param/*")
|
2022-07-13 08:48:29 +03:00
|
|
|
group.Use("/", func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
return c.Next()
|
2020-05-13 20:21:49 +02:00
|
|
|
})
|
2022-07-13 08:48:29 +03:00
|
|
|
group.Get("/test", func(c Ctx) error {
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "john", c.Params("param"))
|
|
|
|
require.Equal(t, "doe", c.Params("*"))
|
2020-09-13 11:20:11 +02:00
|
|
|
return nil
|
2020-05-13 20:21:49 +02:00
|
|
|
})
|
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/prefix/john/doe/test", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
2020-05-13 20:21:49 +02:00
|
|
|
}
|
|
|
|
|
2020-05-27 22:53:40 +02:00
|
|
|
func Test_App_Chaining(t *testing.T) {
|
2022-07-13 08:48:29 +03:00
|
|
|
n := func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
return c.Next()
|
2020-05-27 22:53:40 +02:00
|
|
|
}
|
|
|
|
app := New()
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Use("/john", n, n, n, n, func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
return c.SendStatus(202)
|
2020-05-27 22:53:40 +02:00
|
|
|
})
|
2020-07-20 19:29:54 +02:00
|
|
|
// check handler count for registered HEAD route
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, 5, len(app.stack[methodInt(MethodHead)][0].Handlers), "app.Test(req)")
|
2020-05-27 22:53:40 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
req := httptest.NewRequest(MethodPost, "/john", nil)
|
2020-05-27 22:53:40 +02:00
|
|
|
|
|
|
|
resp, err := app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 202, resp.StatusCode, "Status code")
|
2020-05-27 22:53:40 +02:00
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Get("/test", n, n, n, n, func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
return c.SendStatus(203)
|
2020-05-27 22:53:40 +02:00
|
|
|
})
|
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
req = httptest.NewRequest(MethodGet, "/test", nil)
|
2020-05-27 22:53:40 +02:00
|
|
|
|
|
|
|
resp, err = app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 203, resp.StatusCode, "Status code")
|
2020-05-27 22:53:40 +02:00
|
|
|
}
|
|
|
|
|
2020-05-12 19:24:04 +02:00
|
|
|
func Test_App_Order(t *testing.T) {
|
2020-05-09 15:15:34 +02:00
|
|
|
app := New()
|
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Get("/test", func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
c.Write([]byte("1"))
|
|
|
|
return c.Next()
|
2020-05-07 19:28:21 +02:00
|
|
|
})
|
2020-05-09 15:15:34 +02:00
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.All("/test", func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
c.Write([]byte("2"))
|
|
|
|
return c.Next()
|
2020-05-07 19:28:21 +02:00
|
|
|
})
|
2020-05-09 15:15:34 +02:00
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Use(func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
c.Write([]byte("3"))
|
|
|
|
return nil
|
2020-05-07 19:28:21 +02:00
|
|
|
})
|
2020-05-09 15:15:34 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
req := httptest.NewRequest(MethodGet, "/test", nil)
|
2020-05-09 15:15:34 +02:00
|
|
|
|
2020-05-07 19:28:21 +02:00
|
|
|
resp, err := app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
2020-05-09 15:15:34 +02:00
|
|
|
|
2022-06-01 18:34:22 +03:00
|
|
|
body, err := io.ReadAll(resp.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, "123", string(body))
|
2020-05-07 19:28:21 +02:00
|
|
|
}
|
2021-11-05 08:00:03 +01:00
|
|
|
|
2020-05-12 19:24:04 +02:00
|
|
|
func Test_App_Methods(t *testing.T) {
|
2021-11-05 08:00:03 +01:00
|
|
|
dummyHandler := testEmptyHandler
|
2020-05-09 15:15:34 +02:00
|
|
|
|
|
|
|
app := New()
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
app.Connect("/:john?/:doe?", dummyHandler)
|
|
|
|
testStatus200(t, app, "/john/doe", "CONNECT")
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
app.Put("/:john?/:doe?", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/john/doe", MethodPut)
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
app.Post("/:john?/:doe?", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/john/doe", MethodPost)
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
app.Delete("/:john?/:doe?", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/john/doe", MethodDelete)
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
app.Head("/:john?/:doe?", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/john/doe", MethodHead)
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
app.Patch("/:john?/:doe?", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/john/doe", MethodPatch)
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
app.Options("/:john?/:doe?", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/john/doe", MethodOptions)
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
app.Trace("/:john?/:doe?", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/john/doe", MethodTrace)
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
app.Get("/:john?/:doe?", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/john/doe", MethodGet)
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
app.All("/:john?/:doe?", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/john/doe", MethodPost)
|
2020-05-09 15:15:34 +02:00
|
|
|
|
|
|
|
app.Use("/:john?/:doe?", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/john/doe", MethodGet)
|
2020-02-21 01:54:50 +01:00
|
|
|
}
|
|
|
|
|
2021-12-28 16:04:04 +03:00
|
|
|
func Test_App_Route_Naming(t *testing.T) {
|
|
|
|
app := New()
|
2022-07-13 08:48:29 +03:00
|
|
|
handler := func(c Ctx) error {
|
2021-12-28 16:04:04 +03:00
|
|
|
return c.SendStatus(StatusOK)
|
|
|
|
}
|
|
|
|
app.Get("/john", handler).Name("john")
|
|
|
|
app.Delete("/doe", handler)
|
|
|
|
app.Name("doe")
|
|
|
|
|
|
|
|
jane := app.Group("/jane").Name("jane.")
|
|
|
|
jane.Get("/test", handler).Name("test")
|
|
|
|
jane.Trace("/trace", handler).Name("trace")
|
|
|
|
|
|
|
|
group := app.Group("/group")
|
|
|
|
group.Get("/test", handler).Name("test")
|
|
|
|
|
|
|
|
app.Post("/post", handler).Name("post")
|
|
|
|
|
2022-01-29 17:26:01 +03:00
|
|
|
subGroup := jane.Group("/sub-group").Name("sub.")
|
|
|
|
subGroup.Get("/done", handler).Name("done")
|
|
|
|
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "post", app.GetRoute("post").Name)
|
|
|
|
require.Equal(t, "john", app.GetRoute("john").Name)
|
|
|
|
require.Equal(t, "jane.test", app.GetRoute("jane.test").Name)
|
|
|
|
require.Equal(t, "jane.trace", app.GetRoute("jane.trace").Name)
|
|
|
|
require.Equal(t, "jane.sub.done", app.GetRoute("jane.sub.done").Name)
|
|
|
|
require.Equal(t, "test", app.GetRoute("test").Name)
|
2021-12-28 16:04:04 +03:00
|
|
|
}
|
|
|
|
|
2020-05-12 19:24:04 +02:00
|
|
|
func Test_App_New(t *testing.T) {
|
2020-05-09 15:15:34 +02:00
|
|
|
app := New()
|
2020-09-13 11:20:11 +02:00
|
|
|
app.Get("/", testEmptyHandler)
|
2020-05-09 15:15:34 +02:00
|
|
|
|
2020-09-13 11:20:11 +02:00
|
|
|
appConfig := New(Config{
|
2020-04-13 09:01:27 +02:00
|
|
|
Immutable: true,
|
|
|
|
})
|
2020-09-13 11:20:11 +02:00
|
|
|
appConfig.Get("/", testEmptyHandler)
|
2020-04-13 09:01:27 +02:00
|
|
|
}
|
|
|
|
|
2020-09-15 09:02:24 +08:00
|
|
|
func Test_App_Config(t *testing.T) {
|
2020-09-13 11:20:11 +02:00
|
|
|
app := New(Config{
|
2020-04-28 21:34:34 +02:00
|
|
|
DisableStartupMessage: true,
|
|
|
|
})
|
2022-08-22 13:57:10 +08:00
|
|
|
require.True(t, app.Config().DisableStartupMessage)
|
2020-09-15 09:02:24 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func Test_App_Shutdown(t *testing.T) {
|
|
|
|
t.Run("success", func(t *testing.T) {
|
|
|
|
app := New(Config{
|
|
|
|
DisableStartupMessage: true,
|
|
|
|
})
|
2022-08-22 13:57:10 +08:00
|
|
|
require.True(t, app.Shutdown() == nil)
|
2020-09-15 09:02:24 +08:00
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("no server", func(t *testing.T) {
|
|
|
|
app := &App{}
|
|
|
|
if err := app.Shutdown(); err != nil {
|
|
|
|
if err.Error() != "shutdown: server is not running" {
|
|
|
|
t.Fatal()
|
|
|
|
}
|
2020-07-06 17:51:25 +02:00
|
|
|
}
|
2020-09-15 09:02:24 +08:00
|
|
|
})
|
2020-04-13 09:01:27 +02:00
|
|
|
}
|
|
|
|
|
2020-06-03 17:16:10 +02:00
|
|
|
// go test -run Test_App_Static_Index_Default
|
|
|
|
func Test_App_Static_Index_Default(t *testing.T) {
|
2020-05-29 10:22:54 +02:00
|
|
|
app := New()
|
|
|
|
|
2020-06-01 12:54:32 +02:00
|
|
|
app.Static("/prefix", "./.github/workflows")
|
2020-07-14 15:23:41 +08:00
|
|
|
app.Static("", "./.github/")
|
|
|
|
app.Static("test", "", Static{Index: "index.html"})
|
2020-05-29 10:22:54 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, MIMETextHTMLCharsetUTF8, resp.Header.Get(HeaderContentType))
|
2020-05-29 10:22:54 +02:00
|
|
|
|
2022-06-01 18:34:22 +03:00
|
|
|
body, err := io.ReadAll(resp.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.True(t, strings.Contains(string(body), "Hello, World!"))
|
2020-06-01 12:54:32 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/not-found", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 404, resp.StatusCode, "Status code")
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, MIMETextPlainCharsetUTF8, resp.Header.Get(HeaderContentType))
|
2020-07-14 15:23:41 +08:00
|
|
|
|
2022-06-01 18:34:22 +03:00
|
|
|
body, err = io.ReadAll(resp.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, "Cannot GET /not-found", string(body))
|
2020-05-29 10:22:54 +02:00
|
|
|
}
|
2020-06-03 17:16:10 +02:00
|
|
|
|
|
|
|
// go test -run Test_App_Static_Index
|
|
|
|
func Test_App_Static_Direct(t *testing.T) {
|
|
|
|
app := New()
|
|
|
|
|
|
|
|
app.Static("/", "./.github")
|
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/index.html", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, MIMETextHTMLCharsetUTF8, resp.Header.Get(HeaderContentType))
|
2020-06-03 17:16:10 +02:00
|
|
|
|
2022-06-01 18:34:22 +03:00
|
|
|
body, err := io.ReadAll(resp.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.True(t, strings.Contains(string(body), "Hello, World!"))
|
2020-06-03 17:16:10 +02:00
|
|
|
|
2021-08-22 09:48:07 +02:00
|
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/testdata/testRoutes.json", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, MIMEApplicationJSON, resp.Header.Get("Content-Type"))
|
|
|
|
require.Equal(t, "", resp.Header.Get(HeaderCacheControl), "CacheControl Control")
|
2020-10-07 10:20:19 +02:00
|
|
|
|
2022-06-01 18:34:22 +03:00
|
|
|
body, err = io.ReadAll(resp.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.True(t, strings.Contains(string(body), "testRoutes"))
|
2020-06-03 17:16:10 +02:00
|
|
|
}
|
2020-10-07 10:20:19 +02:00
|
|
|
|
|
|
|
// go test -run Test_App_Static_MaxAge
|
|
|
|
func Test_App_Static_MaxAge(t *testing.T) {
|
|
|
|
app := New()
|
|
|
|
|
|
|
|
app.Static("/", "./.github", Static{MaxAge: 100})
|
|
|
|
|
|
|
|
resp, err := app.Test(httptest.NewRequest("GET", "/index.html", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, "text/html; charset=utf-8", resp.Header.Get(HeaderContentType))
|
|
|
|
require.Equal(t, "public, max-age=100", resp.Header.Get(HeaderCacheControl), "CacheControl Control")
|
2020-10-07 10:20:19 +02:00
|
|
|
}
|
|
|
|
|
2022-02-03 08:41:45 -05:00
|
|
|
// go test -run Test_App_Static_Download
|
|
|
|
func Test_App_Static_Download(t *testing.T) {
|
|
|
|
app := New()
|
|
|
|
|
|
|
|
app.Static("/fiber.png", "./.github/testdata/fs/img/fiber.png", Static{Download: true})
|
|
|
|
|
|
|
|
resp, err := app.Test(httptest.NewRequest("GET", "/fiber.png", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, "image/png", resp.Header.Get(HeaderContentType))
|
|
|
|
require.Equal(t, `attachment`, resp.Header.Get(HeaderContentDisposition))
|
2022-02-03 08:41:45 -05:00
|
|
|
}
|
|
|
|
|
2020-10-07 10:20:19 +02:00
|
|
|
// go test -run Test_App_Static_Group
|
2020-05-27 22:53:40 +02:00
|
|
|
func Test_App_Static_Group(t *testing.T) {
|
2020-05-09 15:15:34 +02:00
|
|
|
app := New()
|
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
grp := app.Group("/v1", func(c Ctx) error {
|
2020-05-27 22:53:40 +02:00
|
|
|
c.Set("Test-Header", "123")
|
2020-09-13 11:20:11 +02:00
|
|
|
return c.Next()
|
2020-05-27 22:53:40 +02:00
|
|
|
})
|
2020-05-09 15:15:34 +02:00
|
|
|
|
2021-08-22 09:48:07 +02:00
|
|
|
grp.Static("/v2", "./.github/index.html")
|
2020-05-09 15:15:34 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
req := httptest.NewRequest(MethodGet, "/v1/v2", nil)
|
2020-03-04 12:30:29 +01:00
|
|
|
resp, err := app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, MIMETextHTMLCharsetUTF8, resp.Header.Get(HeaderContentType))
|
|
|
|
require.Equal(t, "123", resp.Header.Get("Test-Header"))
|
2020-05-27 22:53:40 +02:00
|
|
|
|
|
|
|
grp = app.Group("/v2")
|
2021-08-22 09:48:07 +02:00
|
|
|
grp.Static("/v3*", "./.github/index.html")
|
2020-05-09 15:15:34 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
req = httptest.NewRequest(MethodGet, "/v2/v3/john/doe", nil)
|
2020-03-04 12:30:29 +01:00
|
|
|
resp, err = app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, MIMETextHTMLCharsetUTF8, resp.Header.Get(HeaderContentType))
|
2020-05-27 22:53:40 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
func Test_App_Static_Wildcard(t *testing.T) {
|
|
|
|
app := New()
|
|
|
|
|
2021-08-22 09:48:07 +02:00
|
|
|
app.Static("*", "./.github/index.html")
|
2020-05-27 22:53:40 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
req := httptest.NewRequest(MethodGet, "/yesyes/john/doe", nil)
|
2020-05-27 22:53:40 +02:00
|
|
|
resp, err := app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, MIMETextHTMLCharsetUTF8, resp.Header.Get(HeaderContentType))
|
2020-05-27 22:53:40 +02:00
|
|
|
|
2022-06-01 18:34:22 +03:00
|
|
|
body, err := io.ReadAll(resp.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.True(t, strings.Contains(string(body), "Test file"))
|
2020-05-27 22:53:40 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
func Test_App_Static_Prefix_Wildcard(t *testing.T) {
|
|
|
|
app := New()
|
|
|
|
|
2021-08-22 09:48:07 +02:00
|
|
|
app.Static("/test/*", "./.github/index.html")
|
2020-05-27 22:53:40 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
req := httptest.NewRequest(MethodGet, "/test/john/doe", nil)
|
2020-05-27 22:53:40 +02:00
|
|
|
resp, err := app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, MIMETextHTMLCharsetUTF8, resp.Header.Get(HeaderContentType))
|
2020-06-03 17:16:10 +02:00
|
|
|
|
2021-08-22 09:48:07 +02:00
|
|
|
app.Static("/my/nameisjohn*", "./.github/index.html")
|
2020-06-03 17:16:10 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/my/nameisjohn/no/its/not", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, MIMETextHTMLCharsetUTF8, resp.Header.Get(HeaderContentType))
|
2020-06-03 17:16:10 +02:00
|
|
|
|
2022-06-01 18:34:22 +03:00
|
|
|
body, err := io.ReadAll(resp.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.True(t, strings.Contains(string(body), "Test file"))
|
2020-05-27 22:53:40 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
func Test_App_Static_Prefix(t *testing.T) {
|
|
|
|
app := New()
|
|
|
|
app.Static("/john", "./.github")
|
|
|
|
|
2021-08-22 09:48:07 +02:00
|
|
|
req := httptest.NewRequest(MethodGet, "/john/index.html", nil)
|
2020-05-27 22:53:40 +02:00
|
|
|
resp, err := app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, MIMETextHTMLCharsetUTF8, resp.Header.Get(HeaderContentType))
|
2020-05-27 22:53:40 +02:00
|
|
|
|
2021-08-22 09:48:07 +02:00
|
|
|
app.Static("/prefix", "./.github/testdata")
|
2020-05-27 22:53:40 +02:00
|
|
|
|
2022-02-13 00:52:24 +03:00
|
|
|
req = httptest.NewRequest(MethodGet, "/prefix/index.html", nil)
|
2020-03-04 12:30:29 +01:00
|
|
|
resp, err = app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, MIMETextHTMLCharsetUTF8, resp.Header.Get(HeaderContentType))
|
2020-05-27 22:53:40 +02:00
|
|
|
|
2021-08-22 09:48:07 +02:00
|
|
|
app.Static("/single", "./.github/testdata/testRoutes.json")
|
2020-05-09 15:15:34 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
req = httptest.NewRequest(MethodGet, "/single", nil)
|
2020-03-04 12:30:29 +01:00
|
|
|
resp, err = app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, MIMEApplicationJSON, resp.Header.Get(HeaderContentType))
|
2020-03-04 12:30:29 +01:00
|
|
|
}
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-09-20 21:48:45 +02:00
|
|
|
func Test_App_Static_Trailing_Slash(t *testing.T) {
|
|
|
|
app := New()
|
|
|
|
app.Static("/john", "./.github")
|
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
req := httptest.NewRequest(MethodGet, "/john/", nil)
|
2020-09-20 21:48:45 +02:00
|
|
|
resp, err := app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, MIMETextHTMLCharsetUTF8, resp.Header.Get(HeaderContentType))
|
2021-10-01 18:51:48 +08:00
|
|
|
|
|
|
|
app.Static("/john_without_index", "./.github/testdata/fs/css")
|
|
|
|
|
|
|
|
req = httptest.NewRequest(MethodGet, "/john_without_index/", nil)
|
|
|
|
resp, err = app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 404, resp.StatusCode, "Status code")
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, MIMETextPlainCharsetUTF8, resp.Header.Get(HeaderContentType))
|
2021-10-05 16:55:11 +08:00
|
|
|
|
|
|
|
app.Static("/john/", "./.github")
|
|
|
|
|
|
|
|
req = httptest.NewRequest(MethodGet, "/john/", nil)
|
|
|
|
resp, err = app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, MIMETextHTMLCharsetUTF8, resp.Header.Get(HeaderContentType))
|
2021-10-05 16:55:11 +08:00
|
|
|
|
|
|
|
req = httptest.NewRequest(MethodGet, "/john", nil)
|
|
|
|
resp, err = app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, MIMETextHTMLCharsetUTF8, resp.Header.Get(HeaderContentType))
|
2021-10-05 16:55:11 +08:00
|
|
|
|
|
|
|
app.Static("/john_without_index/", "./.github/testdata/fs/css")
|
|
|
|
|
|
|
|
req = httptest.NewRequest(MethodGet, "/john_without_index/", nil)
|
|
|
|
resp, err = app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 404, resp.StatusCode, "Status code")
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, MIMETextPlainCharsetUTF8, resp.Header.Get(HeaderContentType))
|
2020-09-20 21:48:45 +02:00
|
|
|
}
|
|
|
|
|
2021-01-23 08:31:45 +01:00
|
|
|
func Test_App_Static_Next(t *testing.T) {
|
|
|
|
app := New()
|
|
|
|
app.Static("/", ".github", Static{
|
2022-07-13 08:48:29 +03:00
|
|
|
Next: func(c Ctx) bool {
|
2021-01-23 08:31:45 +01:00
|
|
|
// If value of the header is any other from "skip"
|
|
|
|
// c.Next() will be invoked
|
|
|
|
return c.Get("X-Custom-Header") == "skip"
|
|
|
|
},
|
|
|
|
})
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Get("/", func(c Ctx) error {
|
2021-01-23 08:31:45 +01:00
|
|
|
return c.SendString("You've skipped app.Static")
|
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("app.Static is skipped: invoking Get handler", func(t *testing.T) {
|
|
|
|
req := httptest.NewRequest("GET", "/", nil)
|
|
|
|
req.Header.Set("X-Custom-Header", "skip")
|
|
|
|
resp, err := app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, 200, resp.StatusCode)
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, MIMETextPlainCharsetUTF8, resp.Header.Get(HeaderContentType))
|
2021-01-23 08:31:45 +01:00
|
|
|
|
2022-06-01 18:34:22 +03:00
|
|
|
body, err := io.ReadAll(resp.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.True(t, strings.Contains(string(body), "You've skipped app.Static"))
|
2021-01-23 08:31:45 +01:00
|
|
|
})
|
|
|
|
|
|
|
|
t.Run("app.Static is not skipped: serving index.html", func(t *testing.T) {
|
|
|
|
req := httptest.NewRequest("GET", "/", nil)
|
|
|
|
req.Header.Set("X-Custom-Header", "don't skip")
|
|
|
|
resp, err := app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, 200, resp.StatusCode)
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, MIMETextHTMLCharsetUTF8, resp.Header.Get(HeaderContentType))
|
2021-01-23 08:31:45 +01:00
|
|
|
|
2022-06-01 18:34:22 +03:00
|
|
|
body, err := io.ReadAll(resp.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.True(t, strings.Contains(string(body), "Hello, World!"))
|
2021-01-23 08:31:45 +01:00
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2020-06-02 23:25:06 +02:00
|
|
|
// go test -run Test_App_Mixed_Routes_WithSameLen
|
|
|
|
func Test_App_Mixed_Routes_WithSameLen(t *testing.T) {
|
|
|
|
app := New()
|
|
|
|
|
|
|
|
// middleware
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Use(func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
c.Set("TestHeader", "TestValue")
|
|
|
|
return c.Next()
|
2020-06-02 23:25:06 +02:00
|
|
|
})
|
|
|
|
// routes with the same length
|
|
|
|
app.Static("/tesbar", "./.github")
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Get("/foobar", func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
c.Type("html")
|
|
|
|
return c.Send([]byte("FOO_BAR"))
|
2020-06-02 23:25:06 +02:00
|
|
|
})
|
|
|
|
|
|
|
|
// match get route
|
2020-10-07 11:07:12 +02:00
|
|
|
req := httptest.NewRequest(MethodGet, "/foobar", nil)
|
2020-06-02 23:25:06 +02:00
|
|
|
resp, err := app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, "TestValue", resp.Header.Get("TestHeader"))
|
|
|
|
require.Equal(t, "text/html", resp.Header.Get(HeaderContentType))
|
2020-06-02 23:25:06 +02:00
|
|
|
|
2022-06-01 18:34:22 +03:00
|
|
|
body, err := io.ReadAll(resp.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, "FOO_BAR", string(body))
|
2020-06-02 23:25:06 +02:00
|
|
|
|
|
|
|
// match static route
|
2020-10-07 11:07:12 +02:00
|
|
|
req = httptest.NewRequest(MethodGet, "/tesbar", nil)
|
2020-06-02 23:25:06 +02:00
|
|
|
resp, err = app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
|
|
|
require.False(t, resp.Header.Get(HeaderContentLength) == "")
|
|
|
|
require.Equal(t, "TestValue", resp.Header.Get("TestHeader"))
|
|
|
|
require.Equal(t, "text/html; charset=utf-8", resp.Header.Get(HeaderContentType))
|
2020-06-02 23:25:06 +02:00
|
|
|
|
2022-06-01 18:34:22 +03:00
|
|
|
body, err = io.ReadAll(resp.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.True(t, strings.Contains(string(body), "Hello, World!"), "Response: "+string(body))
|
|
|
|
require.True(t, strings.HasPrefix(string(body), "<!DOCTYPE html>"), "Response: "+string(body))
|
2020-06-02 23:25:06 +02:00
|
|
|
}
|
|
|
|
|
2020-07-16 16:42:36 +08:00
|
|
|
func Test_App_Group_Invalid(t *testing.T) {
|
|
|
|
defer func() {
|
|
|
|
if err := recover(); err != nil {
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "use: invalid handler int\n", fmt.Sprintf("%v", err))
|
2020-07-16 16:42:36 +08:00
|
|
|
}
|
|
|
|
}()
|
|
|
|
New().Group("/").Use(1)
|
|
|
|
}
|
|
|
|
|
2020-10-04 00:36:11 +08:00
|
|
|
// go test -run Test_App_Group_Mount
|
|
|
|
func Test_App_Group_Mount(t *testing.T) {
|
|
|
|
micro := New()
|
2022-07-13 08:48:29 +03:00
|
|
|
micro.Get("/doe", func(c Ctx) error {
|
2020-10-04 00:36:11 +08:00
|
|
|
return c.SendStatus(StatusOK)
|
|
|
|
})
|
|
|
|
|
|
|
|
app := New()
|
|
|
|
v1 := app.Group("/v1")
|
|
|
|
v1.Mount("/john", micro)
|
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/v1/john/doe", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
2022-09-05 17:41:25 +03:00
|
|
|
require.Equal(t, uint32(1), app.handlersCount)
|
2020-10-04 00:36:11 +08:00
|
|
|
}
|
|
|
|
|
2020-05-12 19:24:04 +02:00
|
|
|
func Test_App_Group(t *testing.T) {
|
2021-11-05 08:00:03 +01:00
|
|
|
dummyHandler := testEmptyHandler
|
2020-05-09 15:15:34 +02:00
|
|
|
|
|
|
|
app := New()
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-02-21 01:54:50 +01:00
|
|
|
grp := app.Group("/test")
|
2020-05-09 15:15:34 +02:00
|
|
|
grp.Get("/", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/test", MethodGet)
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
grp.Get("/:demo?", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/test/john", MethodGet)
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
grp.Connect("/CONNECT", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/test/CONNECT", MethodConnect)
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
grp.Put("/PUT", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/test/PUT", MethodPut)
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
grp.Post("/POST", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/test/POST", MethodPost)
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
grp.Delete("/DELETE", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/test/DELETE", MethodDelete)
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
grp.Head("/HEAD", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/test/HEAD", MethodHead)
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
grp.Patch("/PATCH", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/test/PATCH", MethodPatch)
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
grp.Options("/OPTIONS", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/test/OPTIONS", MethodOptions)
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
grp.Trace("/TRACE", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/test/TRACE", MethodTrace)
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
grp.All("/ALL", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/test/ALL", MethodPost)
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-12-28 13:13:24 +08:00
|
|
|
grp.Use(dummyHandler)
|
|
|
|
testStatus200(t, app, "/test/oke", MethodGet)
|
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
grp.Use("/USE", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/test/USE/oke", MethodGet)
|
2020-02-21 18:07:43 +01:00
|
|
|
|
|
|
|
api := grp.Group("/v1")
|
2020-05-09 15:15:34 +02:00
|
|
|
api.Post("/", dummyHandler)
|
2020-05-23 09:25:49 +02:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodPost, "/test/v1/", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
2021-11-05 08:00:03 +01:00
|
|
|
// utils.AssertEqual(t, "/test/v1", resp.Header.Get("Location"), "Location")
|
2020-02-21 18:07:43 +01:00
|
|
|
|
2020-05-09 15:15:34 +02:00
|
|
|
api.Get("/users", dummyHandler)
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/test/v1/UsErS", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
2021-11-05 08:00:03 +01:00
|
|
|
// utils.AssertEqual(t, "/test/v1/users", resp.Header.Get("Location"), "Location")
|
2020-02-21 01:54:50 +01:00
|
|
|
}
|
|
|
|
|
2022-01-24 18:37:28 +03:00
|
|
|
func Test_App_Route(t *testing.T) {
|
|
|
|
dummyHandler := testEmptyHandler
|
|
|
|
|
|
|
|
app := New()
|
|
|
|
|
|
|
|
grp := app.Route("/test", func(grp Router) {
|
|
|
|
grp.Get("/", dummyHandler)
|
|
|
|
grp.Get("/:demo?", dummyHandler)
|
|
|
|
grp.Connect("/CONNECT", dummyHandler)
|
|
|
|
grp.Put("/PUT", dummyHandler)
|
|
|
|
grp.Post("/POST", dummyHandler)
|
|
|
|
grp.Delete("/DELETE", dummyHandler)
|
|
|
|
grp.Head("/HEAD", dummyHandler)
|
|
|
|
grp.Patch("/PATCH", dummyHandler)
|
|
|
|
grp.Options("/OPTIONS", dummyHandler)
|
|
|
|
grp.Trace("/TRACE", dummyHandler)
|
|
|
|
grp.All("/ALL", dummyHandler)
|
|
|
|
grp.Use(dummyHandler)
|
|
|
|
grp.Use("/USE", dummyHandler)
|
|
|
|
})
|
|
|
|
|
|
|
|
testStatus200(t, app, "/test", MethodGet)
|
|
|
|
testStatus200(t, app, "/test/john", MethodGet)
|
|
|
|
testStatus200(t, app, "/test/CONNECT", MethodConnect)
|
|
|
|
testStatus200(t, app, "/test/PUT", MethodPut)
|
|
|
|
testStatus200(t, app, "/test/POST", MethodPost)
|
|
|
|
testStatus200(t, app, "/test/DELETE", MethodDelete)
|
|
|
|
testStatus200(t, app, "/test/HEAD", MethodHead)
|
|
|
|
testStatus200(t, app, "/test/PATCH", MethodPatch)
|
|
|
|
testStatus200(t, app, "/test/OPTIONS", MethodOptions)
|
|
|
|
testStatus200(t, app, "/test/TRACE", MethodTrace)
|
|
|
|
testStatus200(t, app, "/test/ALL", MethodPost)
|
|
|
|
testStatus200(t, app, "/test/oke", MethodGet)
|
|
|
|
testStatus200(t, app, "/test/USE/oke", MethodGet)
|
|
|
|
|
|
|
|
grp.Route("/v1", func(grp Router) {
|
|
|
|
grp.Post("/", dummyHandler)
|
|
|
|
grp.Get("/users", dummyHandler)
|
|
|
|
})
|
|
|
|
|
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodPost, "/test/v1/", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
2022-01-24 18:37:28 +03:00
|
|
|
|
|
|
|
resp, err = app.Test(httptest.NewRequest(MethodGet, "/test/v1/UsErS", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
2022-01-24 18:37:28 +03:00
|
|
|
}
|
|
|
|
|
2020-06-19 10:54:15 +02:00
|
|
|
func Test_App_Deep_Group(t *testing.T) {
|
|
|
|
runThroughCount := 0
|
2022-07-13 08:48:29 +03:00
|
|
|
dummyHandler := func(c Ctx) error {
|
2020-06-19 10:54:15 +02:00
|
|
|
runThroughCount++
|
2020-09-13 11:20:11 +02:00
|
|
|
return c.Next()
|
2020-06-19 10:54:15 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
app := New()
|
2020-07-12 16:26:08 +08:00
|
|
|
gAPI := app.Group("/api", dummyHandler)
|
|
|
|
gV1 := gAPI.Group("/v1", dummyHandler)
|
2020-06-19 10:54:15 +02:00
|
|
|
gUser := gV1.Group("/user", dummyHandler)
|
2022-07-13 08:48:29 +03:00
|
|
|
gUser.Get("/authenticate", func(c Ctx) error {
|
2020-06-19 10:54:15 +02:00
|
|
|
runThroughCount++
|
2020-09-13 11:20:11 +02:00
|
|
|
return c.SendStatus(200)
|
2020-06-19 10:54:15 +02:00
|
|
|
})
|
2020-10-07 11:07:12 +02:00
|
|
|
testStatus200(t, app, "/api/v1/user/authenticate", MethodGet)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, 4, runThroughCount, "Loop count")
|
2020-06-19 10:54:15 +02:00
|
|
|
}
|
|
|
|
|
2020-07-10 13:57:53 +02:00
|
|
|
// go test -run Test_App_Next_Method
|
|
|
|
func Test_App_Next_Method(t *testing.T) {
|
|
|
|
app := New()
|
2020-09-13 11:20:11 +02:00
|
|
|
app.config.DisableStartupMessage = true
|
2020-07-10 13:57:53 +02:00
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Use(func(c Ctx) error {
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, MethodGet, c.Method())
|
2022-04-05 07:39:53 +01:00
|
|
|
err := c.Next()
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, MethodGet, c.Method())
|
2022-04-05 07:39:53 +01:00
|
|
|
return err
|
2020-07-10 13:57:53 +02:00
|
|
|
})
|
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 404, resp.StatusCode, "Status code")
|
2020-07-10 13:57:53 +02:00
|
|
|
}
|
|
|
|
|
2020-09-13 11:20:11 +02:00
|
|
|
// go test -v -run=^$ -bench=Benchmark_AcquireCtx -benchmem -count=4
|
|
|
|
func Benchmark_AcquireCtx(b *testing.B) {
|
|
|
|
app := New()
|
|
|
|
for n := 0; n < b.N; n++ {
|
2022-07-13 08:48:29 +03:00
|
|
|
c := app.AcquireCtx()
|
2022-08-15 18:27:06 +03:00
|
|
|
c.Reset(&fasthttp.RequestCtx{})
|
|
|
|
|
2020-09-13 11:20:11 +02:00
|
|
|
app.ReleaseCtx(c)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-08-08 09:21:00 +03:00
|
|
|
// go test -v -run=^$ -bench=Benchmark_NewError -benchmem -count=4
|
|
|
|
func Benchmark_NewError(b *testing.B) {
|
|
|
|
for n := 0; n < b.N; n++ {
|
|
|
|
NewError(200, "test")
|
|
|
|
}
|
|
|
|
}
|
2022-06-12 19:37:23 +03:00
|
|
|
|
2022-08-08 09:21:00 +03:00
|
|
|
// go test -run Test_NewError
|
|
|
|
func Test_NewError(t *testing.T) {
|
|
|
|
e := NewError(StatusForbidden, "permission denied")
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, StatusForbidden, e.Code)
|
|
|
|
require.Equal(t, "permission denied", e.Message)
|
2022-01-10 21:22:34 +08:00
|
|
|
}
|
|
|
|
|
2020-11-14 03:31:46 +01:00
|
|
|
// go test -run Test_Test_Timeout
|
2020-07-15 10:20:53 +08:00
|
|
|
func Test_Test_Timeout(t *testing.T) {
|
|
|
|
app := New()
|
2020-09-13 11:20:11 +02:00
|
|
|
app.config.DisableStartupMessage = true
|
2020-07-15 10:20:53 +08:00
|
|
|
|
2020-09-13 11:20:11 +02:00
|
|
|
app.Get("/", testEmptyHandler)
|
2020-07-15 10:20:53 +08:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/", nil), -1)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
2020-07-15 10:20:53 +08:00
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Get("timeout", func(c Ctx) error {
|
2022-08-02 08:31:22 +02:00
|
|
|
time.Sleep(200 * time.Millisecond)
|
2020-09-13 11:20:11 +02:00
|
|
|
return nil
|
2020-07-15 10:20:53 +08:00
|
|
|
})
|
|
|
|
|
2022-08-02 08:31:22 +02:00
|
|
|
_, err = app.Test(httptest.NewRequest(MethodGet, "/timeout", nil), 20)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.True(t, err != nil, "app.Test(req)")
|
2020-07-15 10:20:53 +08:00
|
|
|
}
|
|
|
|
|
2020-07-29 09:14:15 +08:00
|
|
|
type errorReader int
|
|
|
|
|
|
|
|
func (errorReader) Read([]byte) (int, error) {
|
|
|
|
return 0, errors.New("errorReader")
|
|
|
|
}
|
|
|
|
|
2020-11-14 03:31:46 +01:00
|
|
|
// go test -run Test_Test_DumpError
|
2020-07-29 09:14:15 +08:00
|
|
|
func Test_Test_DumpError(t *testing.T) {
|
|
|
|
app := New()
|
2020-09-13 11:20:11 +02:00
|
|
|
app.config.DisableStartupMessage = true
|
2020-07-29 09:14:15 +08:00
|
|
|
|
2020-09-13 11:20:11 +02:00
|
|
|
app.Get("/", testEmptyHandler)
|
2020-07-29 09:14:15 +08:00
|
|
|
|
2020-10-07 11:07:12 +02:00
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/", errorReader(0)))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.True(t, resp == nil)
|
|
|
|
require.Equal(t, "errorReader", err.Error())
|
2020-07-29 09:14:15 +08:00
|
|
|
}
|
|
|
|
|
2020-11-14 03:31:46 +01:00
|
|
|
// go test -run Test_App_Handler
|
2020-07-15 10:20:53 +08:00
|
|
|
func Test_App_Handler(t *testing.T) {
|
|
|
|
h := New().Handler()
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "fasthttp.RequestHandler", reflect.TypeOf(h).String())
|
2020-07-15 10:20:53 +08:00
|
|
|
}
|
2020-07-16 16:42:36 +08:00
|
|
|
|
|
|
|
type invalidView struct{}
|
|
|
|
|
|
|
|
func (invalidView) Load() error { return errors.New("invalid view") }
|
|
|
|
|
2022-05-31 17:41:38 +03:00
|
|
|
func (i invalidView) Render(io.Writer, string, any, ...string) error { panic("implement me") }
|
2020-07-16 16:42:36 +08:00
|
|
|
|
2020-11-14 03:31:46 +01:00
|
|
|
// go test -run Test_App_Init_Error_View
|
2020-07-16 16:42:36 +08:00
|
|
|
func Test_App_Init_Error_View(t *testing.T) {
|
2020-09-13 11:20:11 +02:00
|
|
|
app := New(Config{Views: invalidView{}})
|
2020-07-16 16:42:36 +08:00
|
|
|
|
|
|
|
defer func() {
|
|
|
|
if err := recover(); err != nil {
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, "implement me", fmt.Sprintf("%v", err))
|
2020-07-16 16:42:36 +08:00
|
|
|
}
|
|
|
|
}()
|
2020-09-13 11:20:11 +02:00
|
|
|
_ = app.config.Views.Render(nil, "", nil)
|
2020-07-16 16:42:36 +08:00
|
|
|
}
|
2020-07-28 16:48:05 +08:00
|
|
|
|
2020-11-14 03:31:46 +01:00
|
|
|
// go test -run Test_App_Stack
|
2020-07-28 16:48:05 +08:00
|
|
|
func Test_App_Stack(t *testing.T) {
|
|
|
|
app := New()
|
|
|
|
|
2020-09-13 11:20:11 +02:00
|
|
|
app.Use("/path0", testEmptyHandler)
|
|
|
|
app.Get("/path1", testEmptyHandler)
|
|
|
|
app.Get("/path2", testEmptyHandler)
|
|
|
|
app.Post("/path3", testEmptyHandler)
|
2020-07-28 16:48:05 +08:00
|
|
|
|
|
|
|
stack := app.Stack()
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, 9, len(stack))
|
|
|
|
require.Equal(t, 3, len(stack[methodInt(MethodGet)]))
|
2022-09-05 17:41:25 +03:00
|
|
|
require.Equal(t, 1, len(stack[methodInt(MethodHead)]))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, 2, len(stack[methodInt(MethodPost)]))
|
|
|
|
require.Equal(t, 1, len(stack[methodInt(MethodPut)]))
|
|
|
|
require.Equal(t, 1, len(stack[methodInt(MethodPatch)]))
|
|
|
|
require.Equal(t, 1, len(stack[methodInt(MethodDelete)]))
|
|
|
|
require.Equal(t, 1, len(stack[methodInt(MethodConnect)]))
|
|
|
|
require.Equal(t, 1, len(stack[methodInt(MethodOptions)]))
|
|
|
|
require.Equal(t, 1, len(stack[methodInt(MethodTrace)]))
|
2020-07-28 16:48:05 +08:00
|
|
|
}
|
2020-07-29 08:46:36 +08:00
|
|
|
|
2021-12-29 09:23:14 +03:00
|
|
|
// go test -run Test_App_HandlersCount
|
|
|
|
func Test_App_HandlersCount(t *testing.T) {
|
|
|
|
app := New()
|
|
|
|
|
|
|
|
app.Use("/path0", testEmptyHandler)
|
|
|
|
app.Get("/path2", testEmptyHandler)
|
|
|
|
app.Post("/path3", testEmptyHandler)
|
|
|
|
|
|
|
|
count := app.HandlersCount()
|
2022-09-05 17:41:25 +03:00
|
|
|
require.Equal(t, uint32(3), count)
|
2021-12-29 09:23:14 +03:00
|
|
|
}
|
|
|
|
|
2020-07-29 08:46:36 +08:00
|
|
|
// go test -run Test_App_ReadTimeout
|
2020-09-13 11:20:11 +02:00
|
|
|
func Test_App_ReadTimeout(t *testing.T) {
|
|
|
|
app := New(Config{
|
|
|
|
ReadTimeout: time.Nanosecond,
|
|
|
|
IdleTimeout: time.Minute,
|
|
|
|
DisableStartupMessage: true,
|
|
|
|
DisableKeepalive: true,
|
|
|
|
})
|
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Get("/read-timeout", func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
return c.SendString("I should not be sent")
|
|
|
|
})
|
|
|
|
|
|
|
|
go func() {
|
|
|
|
time.Sleep(500 * time.Millisecond)
|
|
|
|
|
2021-02-08 11:35:17 +08:00
|
|
|
conn, err := net.Dial(NetworkTCP4, "127.0.0.1:4004")
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
2020-09-13 11:20:11 +02:00
|
|
|
defer conn.Close()
|
|
|
|
|
|
|
|
_, err = conn.Write([]byte("HEAD /read-timeout HTTP/1.1\r\n"))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
2020-09-13 11:20:11 +02:00
|
|
|
|
|
|
|
buf := make([]byte, 1024)
|
|
|
|
var n int
|
|
|
|
n, err = conn.Read(buf)
|
|
|
|
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.True(t, bytes.Contains(buf[:n], []byte("408 Request Timeout")))
|
2020-09-13 11:20:11 +02:00
|
|
|
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Nil(t, app.Shutdown())
|
2020-09-13 11:20:11 +02:00
|
|
|
}()
|
|
|
|
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Nil(t, app.Listen(":4004"))
|
2020-09-13 11:20:11 +02:00
|
|
|
}
|
2020-07-29 08:46:36 +08:00
|
|
|
|
|
|
|
// go test -run Test_App_BadRequest
|
|
|
|
func Test_App_BadRequest(t *testing.T) {
|
2020-09-13 11:20:11 +02:00
|
|
|
app := New(Config{
|
2020-07-29 08:46:36 +08:00
|
|
|
DisableStartupMessage: true,
|
|
|
|
})
|
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Get("/bad-request", func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
return c.SendString("I should not be sent")
|
2020-07-29 08:46:36 +08:00
|
|
|
})
|
|
|
|
|
|
|
|
go func() {
|
|
|
|
time.Sleep(500 * time.Millisecond)
|
2021-02-08 11:35:17 +08:00
|
|
|
conn, err := net.Dial(NetworkTCP4, "127.0.0.1:4005")
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
2020-07-29 08:46:36 +08:00
|
|
|
defer conn.Close()
|
|
|
|
|
|
|
|
_, err = conn.Write([]byte("BadRequest\r\n"))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
2020-07-29 08:46:36 +08:00
|
|
|
|
|
|
|
buf := make([]byte, 1024)
|
|
|
|
var n int
|
|
|
|
n, err = conn.Read(buf)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
2020-07-29 08:46:36 +08:00
|
|
|
|
2022-08-22 13:57:10 +08:00
|
|
|
require.True(t, bytes.Contains(buf[:n], []byte("400 Bad Request")))
|
2020-07-29 08:46:36 +08:00
|
|
|
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Nil(t, app.Shutdown())
|
2020-07-29 08:46:36 +08:00
|
|
|
}()
|
|
|
|
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Nil(t, app.Listen(":4005"))
|
2020-07-29 08:46:36 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// go test -run Test_App_SmallReadBuffer
|
|
|
|
func Test_App_SmallReadBuffer(t *testing.T) {
|
2020-09-13 11:20:11 +02:00
|
|
|
app := New(Config{
|
2020-07-29 08:46:36 +08:00
|
|
|
ReadBufferSize: 1,
|
|
|
|
DisableStartupMessage: true,
|
|
|
|
})
|
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Get("/small-read-buffer", func(c Ctx) error {
|
2020-09-13 11:20:11 +02:00
|
|
|
return c.SendString("I should not be sent")
|
2020-07-29 08:46:36 +08:00
|
|
|
})
|
|
|
|
|
|
|
|
go func() {
|
|
|
|
time.Sleep(500 * time.Millisecond)
|
2020-07-29 14:52:13 +08:00
|
|
|
resp, err := http.Get("http://127.0.0.1:4006/small-read-buffer")
|
2020-07-29 08:46:36 +08:00
|
|
|
if resp != nil {
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, 431, resp.StatusCode)
|
2020-07-29 08:46:36 +08:00
|
|
|
}
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Nil(t, app.Shutdown())
|
2020-07-29 08:46:36 +08:00
|
|
|
}()
|
|
|
|
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Nil(t, app.Listen(":4006"))
|
2020-07-29 08:46:36 +08:00
|
|
|
}
|
2020-09-15 09:02:24 +08:00
|
|
|
|
2020-10-04 00:05:28 +08:00
|
|
|
func Test_App_Server(t *testing.T) {
|
|
|
|
app := New()
|
|
|
|
|
2022-08-22 13:57:10 +08:00
|
|
|
require.False(t, app.Server() == nil)
|
2020-10-04 00:05:28 +08:00
|
|
|
}
|
2020-10-24 16:24:48 +08:00
|
|
|
|
|
|
|
func Test_App_Error_In_Fasthttp_Server(t *testing.T) {
|
|
|
|
app := New()
|
2022-07-13 08:48:29 +03:00
|
|
|
app.config.ErrorHandler = func(c Ctx, err error) error {
|
2020-10-24 16:24:48 +08:00
|
|
|
return errors.New("fake error")
|
|
|
|
}
|
|
|
|
app.server.GetOnly = true
|
|
|
|
|
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodPost, "/", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, 500, resp.StatusCode)
|
2020-10-24 16:24:48 +08:00
|
|
|
}
|
2021-05-05 16:43:07 -03:00
|
|
|
|
|
|
|
// go test -race -run Test_App_New_Test_Parallel
|
|
|
|
func Test_App_New_Test_Parallel(t *testing.T) {
|
|
|
|
t.Run("Test_App_New_Test_Parallel_1", func(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
app := New(Config{Immutable: true})
|
|
|
|
app.Test(httptest.NewRequest("GET", "/", nil))
|
|
|
|
})
|
|
|
|
t.Run("Test_App_New_Test_Parallel_2", func(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
app := New(Config{Immutable: true})
|
|
|
|
app.Test(httptest.NewRequest("GET", "/", nil))
|
|
|
|
})
|
|
|
|
}
|
2021-07-11 11:45:46 +02:00
|
|
|
|
|
|
|
func Test_App_ReadBodyStream(t *testing.T) {
|
|
|
|
app := New(Config{StreamRequestBody: true})
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Post("/", func(c Ctx) error {
|
2021-07-11 11:45:46 +02:00
|
|
|
// Calling c.Body() automatically reads the entire stream.
|
|
|
|
return c.SendString(fmt.Sprintf("%v %s", c.Request().IsBodyStream(), c.Body()))
|
|
|
|
})
|
|
|
|
testString := "this is a test"
|
|
|
|
resp, err := app.Test(httptest.NewRequest("POST", "/", bytes.NewBufferString(testString)))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
2022-06-01 18:34:22 +03:00
|
|
|
body, err := io.ReadAll(resp.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "io.ReadAll(resp.Body)")
|
|
|
|
require.Equal(t, fmt.Sprintf("true %s", testString), string(body))
|
2021-07-11 11:45:46 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
func Test_App_DisablePreParseMultipartForm(t *testing.T) {
|
|
|
|
// Must be used with both otherwise there is no point.
|
|
|
|
testString := "this is a test"
|
|
|
|
|
|
|
|
app := New(Config{DisablePreParseMultipartForm: true, StreamRequestBody: true})
|
2022-07-13 08:48:29 +03:00
|
|
|
app.Post("/", func(c Ctx) error {
|
2021-07-11 11:45:46 +02:00
|
|
|
req := c.Request()
|
|
|
|
mpf, err := req.MultipartForm()
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if !req.IsBodyStream() {
|
|
|
|
return fmt.Errorf("not a body stream")
|
|
|
|
}
|
|
|
|
file, err := mpf.File["test"][0].Open()
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
buffer := make([]byte, len(testString))
|
|
|
|
n, err := file.Read(buffer)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if n != len(testString) {
|
|
|
|
return fmt.Errorf("bad read length")
|
|
|
|
}
|
|
|
|
return c.Send(buffer)
|
|
|
|
})
|
|
|
|
b := &bytes.Buffer{}
|
|
|
|
w := multipart.NewWriter(b)
|
|
|
|
writer, err := w.CreateFormFile("test", "test")
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "w.CreateFormFile")
|
2021-07-11 11:45:46 +02:00
|
|
|
n, err := writer.Write([]byte(testString))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "writer.Write")
|
|
|
|
require.Equal(t, len(testString), n, "writer n")
|
|
|
|
require.Nil(t, w.Close(), "w.Close()")
|
2021-07-11 11:45:46 +02:00
|
|
|
|
|
|
|
req := httptest.NewRequest("POST", "/", b)
|
|
|
|
req.Header.Set("Content-Type", w.FormDataContentType())
|
|
|
|
resp, err := app.Test(req)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "app.Test(req)")
|
2022-06-01 18:34:22 +03:00
|
|
|
body, err := io.ReadAll(resp.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "io.ReadAll(resp.Body)")
|
2021-07-11 11:45:46 +02:00
|
|
|
|
2022-08-22 13:57:10 +08:00
|
|
|
require.Equal(t, testString, string(body))
|
2021-07-11 11:45:46 +02:00
|
|
|
}
|
2021-10-05 08:03:20 -04:00
|
|
|
|
|
|
|
func Test_App_UseMountedErrorHandler(t *testing.T) {
|
|
|
|
app := New()
|
|
|
|
|
|
|
|
fiber := New(Config{
|
2022-07-13 08:48:29 +03:00
|
|
|
ErrorHandler: func(c Ctx, err error) error {
|
|
|
|
return c.Status(500).SendString("hi, i'm a custom error")
|
2021-10-05 08:03:20 -04:00
|
|
|
},
|
|
|
|
})
|
2022-07-13 08:48:29 +03:00
|
|
|
fiber.Get("/", func(c Ctx) error {
|
2021-10-05 08:03:20 -04:00
|
|
|
return errors.New("something happened")
|
|
|
|
})
|
|
|
|
|
|
|
|
app.Mount("/api", fiber)
|
|
|
|
|
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/api", nil))
|
2022-04-05 08:40:22 +02:00
|
|
|
testErrorResponse(t, err, resp, "hi, i'm a custom error")
|
|
|
|
}
|
|
|
|
|
|
|
|
func Test_App_UseMountedErrorHandlerRootLevel(t *testing.T) {
|
|
|
|
app := New()
|
2021-10-05 08:03:20 -04:00
|
|
|
|
2022-04-05 08:40:22 +02:00
|
|
|
fiber := New(Config{
|
2022-07-13 08:48:29 +03:00
|
|
|
ErrorHandler: func(c Ctx, err error) error {
|
|
|
|
return c.Status(500).SendString("hi, i'm a custom error")
|
2022-04-05 08:40:22 +02:00
|
|
|
},
|
|
|
|
})
|
2022-07-13 08:48:29 +03:00
|
|
|
fiber.Get("/api", func(c Ctx) error {
|
2022-04-05 08:40:22 +02:00
|
|
|
return errors.New("something happened")
|
|
|
|
})
|
|
|
|
|
|
|
|
app.Mount("/", fiber)
|
|
|
|
|
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/api", nil))
|
|
|
|
testErrorResponse(t, err, resp, "hi, i'm a custom error")
|
2021-10-05 08:03:20 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
func Test_App_UseMountedErrorHandlerForBestPrefixMatch(t *testing.T) {
|
|
|
|
app := New()
|
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
tsf := func(c Ctx, err error) error {
|
|
|
|
return c.Status(200).SendString("hi, i'm a custom sub sub fiber error")
|
2021-10-05 08:03:20 -04:00
|
|
|
}
|
|
|
|
tripleSubFiber := New(Config{
|
|
|
|
ErrorHandler: tsf,
|
|
|
|
})
|
2022-07-13 08:48:29 +03:00
|
|
|
tripleSubFiber.Get("/", func(c Ctx) error {
|
2021-10-05 08:03:20 -04:00
|
|
|
return errors.New("something happened")
|
|
|
|
})
|
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
sf := func(c Ctx, err error) error {
|
|
|
|
return c.Status(200).SendString("hi, i'm a custom sub fiber error")
|
2021-10-05 08:03:20 -04:00
|
|
|
}
|
|
|
|
subfiber := New(Config{
|
|
|
|
ErrorHandler: sf,
|
|
|
|
})
|
2022-07-13 08:48:29 +03:00
|
|
|
subfiber.Get("/", func(c Ctx) error {
|
2021-10-05 08:03:20 -04:00
|
|
|
return errors.New("something happened")
|
|
|
|
})
|
|
|
|
subfiber.Mount("/third", tripleSubFiber)
|
|
|
|
|
2022-07-13 08:48:29 +03:00
|
|
|
f := func(c Ctx, err error) error {
|
|
|
|
return c.Status(200).SendString("hi, i'm a custom error")
|
2021-10-05 08:03:20 -04:00
|
|
|
}
|
|
|
|
fiber := New(Config{
|
|
|
|
ErrorHandler: f,
|
|
|
|
})
|
2022-07-13 08:48:29 +03:00
|
|
|
fiber.Get("/", func(c Ctx) error {
|
2021-10-05 08:03:20 -04:00
|
|
|
return errors.New("something happened")
|
|
|
|
})
|
|
|
|
fiber.Mount("/sub", subfiber)
|
|
|
|
|
|
|
|
app.Mount("/api", fiber)
|
|
|
|
|
|
|
|
resp, err := app.Test(httptest.NewRequest(MethodGet, "/api/sub", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "/api/sub req")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
2021-10-05 08:03:20 -04:00
|
|
|
|
2022-06-01 18:34:22 +03:00
|
|
|
b, err := io.ReadAll(resp.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "iotuil.ReadAll()")
|
|
|
|
require.Equal(t, "hi, i'm a custom sub fiber error", string(b), "Response body")
|
2021-10-05 08:03:20 -04:00
|
|
|
|
|
|
|
resp2, err := app.Test(httptest.NewRequest(MethodGet, "/api/sub/third", nil))
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "/api/sub/third req")
|
|
|
|
require.Equal(t, 200, resp.StatusCode, "Status code")
|
2021-10-05 08:03:20 -04:00
|
|
|
|
2022-06-01 18:34:22 +03:00
|
|
|
b, err = io.ReadAll(resp2.Body)
|
2022-08-22 13:57:10 +08:00
|
|
|
require.NoError(t, err, "iotuil.ReadAll()")
|
|
|
|
require.Equal(t, "hi, i'm a custom sub sub fiber error", string(b), "Third fiber Response body")
|
2021-10-05 08:03:20 -04:00
|
|
|
}
|
2021-12-30 21:13:31 +08:00
|
|
|
|
2022-08-08 14:18:57 +08:00
|
|
|
func Test_App_Test_no_timeout_infinitely(t *testing.T) {
|
|
|
|
var err error
|
|
|
|
c := make(chan int)
|
|
|
|
|
|
|
|
go func() {
|
|
|
|
defer func() { c <- 0 }()
|
|
|
|
app := New()
|
2022-08-19 14:33:31 +03:00
|
|
|
app.Get("/", func(c Ctx) error {
|
2022-08-08 14:18:57 +08:00
|
|
|
runtime.Goexit()
|
|
|
|
return nil
|
|
|
|
})
|
|
|
|
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/", http.NoBody)
|
|
|
|
_, err = app.Test(req, -1)
|
|
|
|
}()
|
|
|
|
|
|
|
|
tk := time.NewTimer(5 * time.Second)
|
|
|
|
defer tk.Stop()
|
|
|
|
|
|
|
|
select {
|
|
|
|
case <-tk.C:
|
|
|
|
t.Error("hanging test")
|
|
|
|
t.FailNow()
|
|
|
|
case <-c:
|
|
|
|
}
|
|
|
|
|
|
|
|
if err == nil {
|
|
|
|
t.Error("unexpected success request")
|
|
|
|
t.FailNow()
|
|
|
|
}
|
2021-12-30 21:13:31 +08:00
|
|
|
}
|