Merge pull request #11 from georgizhivankin/406-not-acceptable

Add a method for the 406 Not Acceptable HTTP code
nicklaw5 · Sep 1, 2021 · 5972f8a · 5972f8a
2 parents 6e1c899 + 2040669
commit 5972f8a
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -59,6 +59,7 @@ func main() {
 | 403 | Forbidden() |
 | 404 | NotFound() |
 | 405 | MethodNotAllowed() |
+| 406 | NotAcceptable() |
 | 409 | Conflict() |
 | 411 | LengthRequired() |
 | 412 | PreconditionFailed() |

diff --git a/error.go b/error.go
@@ -27,6 +27,11 @@ func (resp *Response) MethodNotAllowed(v interface{}) {
 	resp.writeResponse(http.StatusMethodNotAllowed, v)
 }
 
+// NotAcceptable returns a 406 Not Acceptable JSON response
+func (resp *Response) NotAcceptable(v interface{}) {
+	resp.writeResponse(http.StatusNotAcceptable, v)
+}
+
 // Conflict returns a 409 Conflict JSON response
 func (resp *Response) Conflict(v interface{}) {
 	resp.writeResponse(http.StatusConflict, v)

diff --git a/error_test.go b/error_test.go
@@ -121,6 +121,28 @@ func TestMethodNotAllowed(t *testing.T) {
 	}
 }
 
+func TestNotAcceptable(t *testing.T) {
+	t.Parallel()
+
+	req := newRequest(t, "GET")
+
+	rr := httptest.NewRecorder()
+	handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		NewResponse(w).
+			NotAcceptable(&Error{406, "Not acceptable"})
+	})
+	handler.ServeHTTP(rr, req)
+
+	if err := validateStatusCode(rr.Code, http.StatusNotAcceptable); err != nil {
+		t.Fatal(err.Error())
+	}
+
+	expected := `{"code":406,"message":"Not acceptable"}`
+	if err := validateResponseBody(rr.Body.String(), expected); err != nil {
+		t.Fatal(err.Error())
+	}
+}
+
 func TestConflict(t *testing.T) {
 	t.Parallel()