Fix test execution

PascalMinder · Mar 14, 2021 · 9629fa8 · 9629fa8
1 parent 18fc1ee
commit 9629fa8
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/.vscode/launch.json b/.vscode/launch.json
@@ -0,0 +1,7 @@
+{
+    // Use IntelliSense to learn about possible attributes.
+    // Hover to view descriptions of existing attributes.
+    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+    "version": "0.2.0",
+    "configurations": []
+}
diff --git a/geoblock_test.go b/geoblock_test.go
@@ -39,7 +39,7 @@ func TestAllowedContry(t *testing.T) {
 
 	handler.ServeHTTP(recorder, req)
 
-	assertStatusCode(t, req, http.StatusOK)
+	assertStatusCode(t, recorder.Result(), http.StatusOK)
 }
 
 func TestDeniedContry(t *testing.T) {
@@ -66,13 +66,13 @@ func TestDeniedContry(t *testing.T) {
 
 	handler.ServeHTTP(recorder, req)
 
-	assertStatusCode(t, req, http.StatusOK)
+	assertStatusCode(t, recorder.Result(), http.StatusForbidden)
 }
 
-func assertStatusCode(t *testing.T, req *http.Request, expected int) {
+func assertStatusCode(t *testing.T, req *http.Response, expected int) {
 	t.Helper()
 
-	received := req.Response.StatusCode
+	received := req.StatusCode
 
 	if received != expected {
 		t.Errorf("invalid status code: %d <> %d", expected, received)