add unit testing case of status handler feature for ghttp.Server

This commit is contained in:
John 2020-05-05 00:01:09 +08:00
parent ec92b08f25
commit 0c021b6da7
2 changed files with 44 additions and 0 deletions

View File

@ -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)

View 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")
})
}