mirror of
https://gitee.com/johng/gf.git
synced 2024-11-30 03:07:45 +08:00
add unit testing case of status handler feature for ghttp.Server
This commit is contained in:
parent
ec92b08f25
commit
0c021b6da7
@ -32,6 +32,7 @@ func Test_Log(t *testing.T) {
|
||||
s.BindHandler("/error", func(r *ghttp.Request) {
|
||||
panic("custom error")
|
||||
})
|
||||
s.SetDumpRouterMap(false)
|
||||
s.SetLogPath(logDir)
|
||||
s.SetAccessLogEnabled(true)
|
||||
s.SetErrorLogEnabled(true)
|
||||
|
43
net/ghttp/ghttp_unit_status_test.go
Normal file
43
net/ghttp/ghttp_unit_status_test.go
Normal file
@ -0,0 +1,43 @@
|
||||
// Copyright 2018 gf Author(https://github.com/gogf/gf). All Rights Reserved.
|
||||
//
|
||||
// This Source Code Form is subject to the terms of the MIT License.
|
||||
// If a copy of the MIT was not distributed with this file,
|
||||
// You can obtain one at https://github.com/gogf/gf.
|
||||
|
||||
// static service testing.
|
||||
|
||||
package ghttp_test
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/gogf/gf/frame/g"
|
||||
"github.com/gogf/gf/net/ghttp"
|
||||
"github.com/gogf/gf/test/gtest"
|
||||
)
|
||||
|
||||
func Test_StatusHandler(t *testing.T) {
|
||||
gtest.C(t, func(t *gtest.T) {
|
||||
p, _ := ports.PopRand()
|
||||
s := g.Server(p)
|
||||
s.BindStatusHandlerByMap(map[int]ghttp.HandlerFunc{
|
||||
404: func(r *ghttp.Request) { r.Response.WriteOver("404") },
|
||||
502: func(r *ghttp.Request) { r.Response.WriteOver("502") },
|
||||
})
|
||||
s.BindHandler("/502", func(r *ghttp.Request) {
|
||||
r.Response.WriteStatusExit(502)
|
||||
})
|
||||
s.SetDumpRouterMap(false)
|
||||
s.SetPort(p)
|
||||
s.Start()
|
||||
defer s.Shutdown()
|
||||
time.Sleep(100 * time.Millisecond)
|
||||
client := ghttp.NewClient()
|
||||
client.SetPrefix(fmt.Sprintf("http://127.0.0.1:%d", p))
|
||||
|
||||
t.Assert(client.GetContent("/404"), "404")
|
||||
t.Assert(client.GetContent("/502"), "502")
|
||||
})
|
||||
}
|
Loading…
Reference in New Issue
Block a user