Merge pull request #325 from linkernetworks/johnlin/move-assertresponse

[Hot-fix] move assert response code function
linkernetworks · Sep 20, 2018 · 6962fd5 · 6962fd5
2 parents 4c91312 + 494bded
commit 6962fd5
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 16 deletions.
diff --git a/src/server/handler_batch_test.go b/src/server/handler_batch_test.go
diff --git a/src/server/main_test.go b/src/server/main_test.go
@@ -72,3 +72,13 @@ func dropTestDatabase() bool {
 	}
 	return false
 }
+
+// Assert the response code
+func assertResponseCode(t *testing.T, expectedCode int, resp *httptest.ResponseRecorder) {
+	t.Helper()
+	t.Logf("code:%d", resp.Code)
+	if expectedCode != resp.Code {
+		t.Errorf("status code %d expected.", expectedCode)
+		t.Logf("Response:\n%s", resp.Body.String())
+	}
+}