add more tests

master
kevin 4 years ago
parent 9c9f80518f
commit 2b1466e41e

@ -209,6 +209,6 @@ func (s *engine) use(middleware Middleware) {
func convertMiddleware(ware Middleware) func(http.Handler) http.Handler {
return func(next http.Handler) http.Handler {
return http.HandlerFunc(ware(next.ServeHTTP))
return ware(next.ServeHTTP)
}
}

@ -1,6 +1,7 @@
package httpx
import (
"errors"
"net/http"
"strings"
"testing"
@ -17,6 +18,24 @@ func init() {
logx.Disable()
}
func TestError(t *testing.T) {
const body = "foo"
w := tracedResponseWriter{
headers: make(map[string][]string),
}
Error(&w, errors.New(body))
assert.Equal(t, http.StatusBadRequest, w.code)
assert.Equal(t, body, strings.TrimSpace(w.builder.String()))
}
func TestOk(t *testing.T) {
w := tracedResponseWriter{
headers: make(map[string][]string),
}
Ok(&w)
assert.Equal(t, http.StatusOK, w.code)
}
func TestOkJson(t *testing.T) {
w := tracedResponseWriter{
headers: make(map[string][]string),

Loading…
Cancel
Save