Add FilterHTTPMethod and FilteringHTTPHandler

1 files changed, 97 insertions, 0 deletions

diff --git a/http_test.go b/http_test.go
new file mode 100644
index 0000000..a8c3cb5
--- /dev/null
+++ b/http_test.go
@@ -0,0 +1,97 @@
+// SPDX-FileCopyrightText: © 2022 Grégoire Duchêne <gduchene@awhk.org>
+// SPDX-License-Identifier: ISC
+
+package core_test
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"go.awhk.org/core"
+)
+
+func TestFilteringHTTPHandler(s *testing.T) {
+	t := core.T{T: s}
+
+	handler := core.FilteringHTTPHandler(
+		http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) { w.WriteHeader(http.StatusOK) }),
+		core.FilterHTTPMethod(http.MethodHead),
+	)
+	for _, tc := range []struct {
+		name   string
+		method string
+
+		expHeader     http.Header
+		expStatusCode int
+	}{
+		{
+			name:   "Success",
+			method: http.MethodHead,
+
+			expHeader:     http.Header{},
+			expStatusCode: http.StatusOK,
+		},
+		{
+			name:   "WhenFiltered",
+			method: http.MethodGet,
+
+			expHeader:     http.Header{"Allowed": {"HEAD"}},
+			expStatusCode: http.StatusMethodNotAllowed,
+		},
+	} {
+		t.Run(tc.name, func(t *core.T) {
+			var (
+				req = httptest.NewRequest(tc.method, "/", nil)
+				w   = httptest.NewRecorder()
+			)
+			handler.ServeHTTP(w, req)
+
+			res := w.Result()
+			t.AssertEqual(tc.expHeader, res.Header)
+			t.AssertEqual(tc.expStatusCode, res.StatusCode)
+		})
+	}
+}
+
+func TestFilterHTTPMethod(s *testing.T) {
+	t := core.T{T: s}
+
+	filter := core.FilterHTTPMethod(http.MethodPost, http.MethodGet)
+	for _, tc := range []struct {
+		name   string
+		method string
+
+		expAllowed    string
+		expFiltered   bool
+		expStatusCode int
+	}{
+		{
+			name:   "Success",
+			method: http.MethodPost,
+
+			expFiltered:   false,
+			expStatusCode: http.StatusOK,
+		},
+		{
+			name:   "WhenFiltered",
+			method: http.MethodHead,
+
+			expAllowed:    "GET, POST",
+			expFiltered:   true,
+			expStatusCode: http.StatusMethodNotAllowed,
+		},
+	} {
+		t.Run(tc.name, func(t *core.T) {
+			var (
+				req = httptest.NewRequest(tc.method, "/", nil)
+				w   = httptest.NewRecorder()
+			)
+			t.AssertEqual(tc.expFiltered, filter(w, req))
+
+			res := w.Result()
+			t.AssertEqual(tc.expAllowed, res.Header.Get("Allowed"))
+			t.AssertEqual(tc.expStatusCode, res.StatusCode)
+		})
+	}
+}