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