Merge pull request #48 from Clever/enable-mock-assertions

mocks issue fixed, enable mocks assertions
Clever · Jun 9, 2015 · 50d7988 · 50d7988
2 parents 77198fd + 05e0452
commit 50d7988
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/handlers/http_test.go b/handlers/http_test.go
@@ -165,8 +165,7 @@ func TestHandleWhenNotFull(t *testing.T) {
 	limiter.ServeHTTP(w, r)
 
 	compareStatusesToHeader(t, w.Header(), statuses)
-	// commented out until https://github.com/stretchr/testify/issues/31 is resolved
-	// limitMock.AssertExpectations(t)
+	limitMock.AssertExpectations(t)
 	proxyMock.AssertExpectations(t)
 }
 
@@ -188,8 +187,7 @@ func TestHandleWhenFull(t *testing.T) {
 	if w.Code != 429 {
 		t.Fatalf("expected status 429, received %d", w.Code)
 	}
-	// commented out until https://github.com/stretchr/testify/issues/31 is resolved
-	// limitMock.AssertExpectations(t)
+	limitMock.AssertExpectations(t)
 }
 
 func TestHandleWhenErrWithStatus(t *testing.T) {
@@ -210,8 +208,7 @@ func TestHandleWhenErrWithStatus(t *testing.T) {
 	if w.Code != 500 {
 		t.Fatalf("expected status 500, received %d", w.Code)
 	}
-	// commented out until https://github.com/stretchr/testify/issues/31 is resolved
-	// limitMock.AssertExpectations(t)
+	limitMock.AssertExpectations(t)
 }
 
 func TestHandleWhenErrWithoutStatus(t *testing.T) {
@@ -232,6 +229,5 @@ func TestHandleWhenErrWithoutStatus(t *testing.T) {
 	if w.Code != 500 {
 		t.Fatalf("expected status 500, received %d", w.Code)
 	}
-	// commented out until https://github.com/stretchr/testify/issues/31 is resolved
-	// limitMock.AssertExpectations(t)
+	limitMock.AssertExpectations(t)
 }