Add test coverage for disallowed methods

csp_collector_test.go

@@ -0,0 +1,32 @@
+package main
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+)
+
+func TestHandlerForDisallowedMethods(t *testing.T) {
+	disallowedMethods := []string{"GET", "DELETE", "PUT", "TRACE", "PATCH"}
+	randomUrls := []string{"/", "/blah"}
+
+	for _, method := range disallowedMethods {
+		for _, url := range randomUrls {
+			t.Run(method+url, func(t *testing.T) {
+				request, err := http.NewRequest(method, "localhost:8080"+url, nil)
+				if err != nil {
+					t.Fatalf("Failed to create request: %v", err)
+				}
+				recorder := httptest.NewRecorder()
+				handleViolationReport(recorder, request)
+
+				response := recorder.Result()
+				defer response.Body.Close()
+
+				if response.StatusCode != http.StatusMethodNotAllowed {
+					t.Errorf("Expected HTTP status %v; got %v", http.StatusMethodNotAllowed, response.Status)
+				}
+			})
+		}
+	}
+}