Update middleware framework integration tests

Signed-off-by: Xabier Larrakoetxea <[email protected]>
slok · Mar 27, 2019 · 8ed45ff · 8ed45ff
1 parent a28f76c
commit 8ed45ff
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 12 deletions.
diff --git a/middleware/gorestful/gorestful_test.go b/middleware/gorestful/gorestful_test.go
@@ -47,10 +47,10 @@ func TestMiddlewareIntegration(t *testing.T) {
 
 			// Mocks.
 			mr := &mmetrics.Recorder{}
-			mr.On("ObserveHTTPRequestDuration", test.expHandlerID, mock.Anything, test.expMethod, test.expStatusCode).Once()
-			mr.On("ObserveHTTPResponseSize", test.expHandlerID, mock.Anything, test.expMethod, test.expStatusCode).Once()
-			mr.On("AddInflightRequests", test.expHandlerID, 1).Once()
-			mr.On("AddInflightRequests", test.expHandlerID, -1).Once()
+			mr.On("ObserveHTTPRequestDuration", mock.Anything, test.expHandlerID, mock.Anything, test.expMethod, test.expStatusCode).Once()
+			mr.On("ObserveHTTPResponseSize", mock.Anything, test.expHandlerID, mock.Anything, test.expMethod, test.expStatusCode).Once()
+			mr.On("AddInflightRequests", mock.Anything, test.expHandlerID, 1).Once()
+			mr.On("AddInflightRequests", mock.Anything, test.expHandlerID, -1).Once()
 
 			// Create our instance with the middleware.
 			mdlw := middleware.New(middleware.Config{Recorder: mr})

diff --git a/middleware/httprouter/httprouter_test.go b/middleware/httprouter/httprouter_test.go
@@ -47,10 +47,10 @@ func TestMiddlewareIntegration(t *testing.T) {
 
 			// Mocks.
 			mr := &mmetrics.Recorder{}
-			mr.On("ObserveHTTPRequestDuration", test.expHandlerID, mock.Anything, test.expMethod, test.expStatusCode).Once()
-			mr.On("ObserveHTTPResponseSize", test.expHandlerID, mock.Anything, test.expMethod, test.expStatusCode).Once()
-			mr.On("AddInflightRequests", test.expHandlerID, 1).Once()
-			mr.On("AddInflightRequests", test.expHandlerID, -1).Once()
+			mr.On("ObserveHTTPRequestDuration", mock.Anything, test.expHandlerID, mock.Anything, test.expMethod, test.expStatusCode).Once()
+			mr.On("ObserveHTTPResponseSize", mock.Anything, test.expHandlerID, mock.Anything, test.expMethod, test.expStatusCode).Once()
+			mr.On("AddInflightRequests", mock.Anything, test.expHandlerID, 1).Once()
+			mr.On("AddInflightRequests", mock.Anything, test.expHandlerID, -1).Once()
 
 			// Create our instance with the middleware.
 			mdlw := middleware.New(middleware.Config{Recorder: mr})

diff --git a/middleware/negroni/negroni_test.go b/middleware/negroni/negroni_test.go
@@ -47,10 +47,10 @@ func TestMiddlewareIntegration(t *testing.T) {
 
 			// Mocks.
 			mr := &mmetrics.Recorder{}
-			mr.On("ObserveHTTPRequestDuration", test.expHandlerID, mock.Anything, test.expMethod, test.expStatusCode).Once()
-			mr.On("ObserveHTTPResponseSize", test.expHandlerID, mock.Anything, test.expMethod, test.expStatusCode).Once()
-			mr.On("AddInflightRequests", test.expHandlerID, 1).Once()
-			mr.On("AddInflightRequests", test.expHandlerID, -1).Once()
+			mr.On("ObserveHTTPRequestDuration", mock.Anything, test.expHandlerID, mock.Anything, test.expMethod, test.expStatusCode).Once()
+			mr.On("ObserveHTTPResponseSize", mock.Anything, test.expHandlerID, mock.Anything, test.expMethod, test.expStatusCode).Once()
+			mr.On("AddInflightRequests", mock.Anything, test.expHandlerID, 1).Once()
+			mr.On("AddInflightRequests", mock.Anything, test.expHandlerID, -1).Once()
 
 			// Create our negroni instance with the middleware.
 			mdlw := middleware.New(middleware.Config{Recorder: mr})