add options shortcut (#455)

* add options shortcut * add test for options shortcut
emicklei · Nov 10, 2020 · 8ed82c3 · 8ed82c3
1 parent 0603e93
commit 8ed82c3
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/web_service.go b/web_service.go
@@ -288,3 +288,8 @@ func (w *WebService) PATCH(subPath string) *RouteBuilder {
 func (w *WebService) DELETE(subPath string) *RouteBuilder {
 	return new(RouteBuilder).typeNameHandler(w.typeNameHandleFunc).servicePath(w.rootPath).Method("DELETE").Path(subPath)
 }
+
+// OPTIONS is a shortcut for .Method("OPTIONS").Path(subPath)
+func (w *WebService) OPTIONS(subPath string) *RouteBuilder {
+	return new(RouteBuilder).typeNameHandler(w.typeNameHandleFunc).servicePath(w.rootPath).Method("OPTIONS").Path(subPath)
+}
diff --git a/web_service_test.go b/web_service_test.go
@@ -299,6 +299,20 @@ func TestParameterDataTypeCustomization(t *testing.T) {
 	}
 }
 
+func TestOptionsShortcut(t *testing.T) {
+	tearDown()
+	ws := new(WebService).Path("")
+	ws.Route(ws.OPTIONS("/options").To(return200))
+	Add(ws)
+
+	httpRequest, _ := http.NewRequest("OPTIONS", "http://here.com/options", nil)
+	httpWriter := httptest.NewRecorder()
+	DefaultContainer.dispatch(httpWriter, httpRequest)
+	if got, want := httpWriter.Code, 200; got != want {
+		t.Errorf("got %v, want %v", got, want)
+	}
+}
+
 func newPanicingService() *WebService {
 	ws := new(WebService).Path("")
 	ws.Route(ws.GET("/fire").To(doPanic))
@@ -385,3 +399,7 @@ func doNothing(req *Request, resp *Response) {
 func return204(req *Request, resp *Response) {
 	resp.WriteHeader(204)
 }
+
+func return200(req *Request, resp *Response) {
+	resp.WriteHeader(200)
+}