diff --git a/middleware_test.go b/middleware_test.go index b708be09..24016cbb 100644 --- a/middleware_test.go +++ b/middleware_test.go @@ -151,7 +151,7 @@ func TestMiddlewareExecution(t *testing.T) { // Test handler-only call router.ServeHTTP(rw, req) - if bytes.Compare(rw.Body.Bytes(), handlerStr) != 0 { + if !bytes.Equal(rw.Body.Bytes(), handlerStr) { t.Fatal("Handler response is not what it should be") } @@ -166,7 +166,7 @@ func TestMiddlewareExecution(t *testing.T) { }) router.ServeHTTP(rw, req) - if bytes.Compare(rw.Body.Bytes(), append(mwStr, handlerStr...)) != 0 { + if !bytes.Equal(rw.Body.Bytes(), append(mwStr, handlerStr...)) { t.Fatal("Middleware + handler response is not what it should be") } } @@ -368,7 +368,7 @@ func TestCORSMethodMiddleware(t *testing.T) { t.Errorf("Expected body '%s', found '%s'", tt.response, rr.Body.String()) } - allowedMethods := rr.HeaderMap.Get("Access-Control-Allow-Methods") + allowedMethods := rr.Header().Get("Access-Control-Allow-Methods") if allowedMethods != tt.expectedAllowedMethods { t.Errorf("Expected Access-Control-Allow-Methods '%s', found '%s'", tt.expectedAllowedMethods, allowedMethods) diff --git a/mux_test.go b/mux_test.go index 5198024d..61653d7f 100644 --- a/mux_test.go +++ b/mux_test.go @@ -1123,10 +1123,7 @@ func TestSchemes(t *testing.T) { func TestMatcherFunc(t *testing.T) { m := func(r *http.Request, m *RouteMatch) bool { - if r.URL.Host == "aaa.bbb.ccc" { - return true - } - return false + return r.URL.Host == "aaa.bbb.ccc" } tests := []routeTest{