GOMODULES
/
fiber
mirror of https://github.com/gofiber/fiber.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update test

pull/455/head
Fenny 2020-06-07 22:17:10 +02:00
parent b4ffb82493
commit b4244de892
3 changed files with 3 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ctx.go
View File

@ -743,9 +743,9 @@ func (ctx *Ctx) Render(name string, bind interface{}) (err error) {
buf := bytebufferpool.Get() buf := bytebufferpool.Get()
defer bytebufferpool.Put(buf) defer bytebufferpool.Put(buf)
// Use ViewEngine if exist // Use Templates engine if exist
if ctx.app.Settings.Templates != nil { if ctx.app.Settings.Templates != nil {
// Render template from ViewEngine // Render template from Engine
if err := ctx.app.Settings.Templates.Render(buf, name, bind); err != nil { if err := ctx.app.Settings.Templates.Render(buf, name, bind); err != nil {
return err return err
} }

2
ctx_test.go
View File

@ -893,7 +893,7 @@ func Test_Ctx_Download(t *testing.T) {
expect, err := ioutil.ReadAll(f) expect, err := ioutil.ReadAll(f)
utils.AssertEqual(t, nil, err) utils.AssertEqual(t, nil, err)
utils.AssertEqual(t, true, bytes.Equal(expect, ctx.Fasthttp.Response.Body())) utils.AssertEqual(t, expect, ctx.Fasthttp.Response.Body())
} }
// go test -run Test_Ctx_JSON // go test -run Test_Ctx_JSON

9
middleware/helmet.go
View File

@ -2,7 +2,6 @@ package middleware
import ( import (
"github.com/gofiber/fiber" "github.com/gofiber/fiber"
"github.com/gofiber/utils"
) )
// Middleware types // Middleware types
@ -33,14 +32,6 @@ type (
// Helmet helps secure your apps by setting various HTTP headers // Helmet helps secure your apps by setting various HTTP headers
func Helmet() fiber.Handler { func Helmet() fiber.Handler {
return func(ctx *fiber.Ctx) { return func(ctx *fiber.Ctx) {
// Get id from request
rid := ctx.Get(fiber.HeaderXRequestID)
// Create new UUID if empty
if len(rid) <= 0 {
rid = utils.UUID()
}
// Set new id to response
ctx.Set(fiber.HeaderXRequestID, rid)
// Continue stack // Continue stack
ctx.Next() ctx.Next()
} }