replace rr.HeaderMap by rr.Header() (#443)

gorilla · Jan 25, 2019 · a796238 · a796238
1 parent 797e653
commit a796238
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git 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
@@ -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{