private/apigen: add PUT method to apigen

Change-Id: I99e2f69855a50181e9302c162dec1bb82fabd15d
storj · Jan 10, 2024 · d0149a7 · d0149a7
1 parent 17a1628
commit d0149a7
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/private/apigen/endpoint.go b/private/apigen/endpoint.go
@@ -206,6 +206,12 @@ func (eg *EndpointGroup) Post(path string, endpoint *Endpoint) {
 	eg.addEndpoint(path, http.MethodPost, endpoint)
 }
 
+// Put adds new PUT endpoint to endpoints group.
+// It panics if path doesn't begin with '/'.
+func (eg *EndpointGroup) Put(path string, endpoint *Endpoint) {
+	eg.addEndpoint(path, http.MethodPut, endpoint)
+}
+
 // Delete adds new DELETE endpoint to endpoints group.
 // It panics if path doesn't begin with '/'.
 func (eg *EndpointGroup) Delete(path string, endpoint *Endpoint) {

diff --git a/private/apigen/endpoint_test.go b/private/apigen/endpoint_test.go
@@ -177,10 +177,11 @@ func TestEndpointGroup(t *testing.T) {
 		assert.NotPanics(t, func() { eg.Get(path, endpointFn(http.MethodGet)) }, "Get")
 		assert.NotPanics(t, func() { eg.Patch(path, endpointFn(http.MethodPatch)) }, "Patch")
 		assert.NotPanics(t, func() { eg.Post(path, endpointFn(http.MethodPost)) }, "Post")
+		assert.NotPanics(t, func() { eg.Put(path, endpointFn(http.MethodPut)) }, "Put")
 		assert.NotPanics(t, func() { eg.Delete(path, endpointFn(http.MethodDelete)) }, "Delete")
 
-		require.Len(t, eg.endpoints, 4, "Group endpoints count")
-		for i, m := range []string{http.MethodGet, http.MethodPatch, http.MethodPost, http.MethodDelete} {
+		require.Len(t, eg.endpoints, 5, "Group endpoints count")
+		for i, m := range []string{http.MethodGet, http.MethodPatch, http.MethodPost, http.MethodPut, http.MethodDelete} {
 			ep := eg.endpoints[i]
 			assert.Equal(t, m, ep.Method)
 			assert.Equal(t, path, ep.Path)
@@ -204,6 +205,7 @@ func TestEndpointGroup(t *testing.T) {
 		assert.Panics(t, func() { eg.Get(path, endpointFn(http.MethodGet)) }, "Get")
 		assert.Panics(t, func() { eg.Patch(path, endpointFn(http.MethodPatch)) }, "Patch")
 		assert.Panics(t, func() { eg.Post(path, endpointFn(http.MethodPost)) }, "Post")
+		assert.Panics(t, func() { eg.Put(path, endpointFn(http.MethodPut)) }, "Put")
 		assert.Panics(t, func() { eg.Delete(path, endpointFn(http.MethodDelete)) }, "Delete")
 	})
 
@@ -223,6 +225,7 @@ func TestEndpointGroup(t *testing.T) {
 		assert.Panics(t, func() { eg.Get(path, endpointFn(http.MethodGet)) }, "Get")
 		assert.Panics(t, func() { eg.Patch(path, endpointFn(http.MethodPatch)) }, "Patch")
 		assert.Panics(t, func() { eg.Post(path, endpointFn(http.MethodPost)) }, "Post")
+		assert.Panics(t, func() { eg.Put(path, endpointFn(http.MethodPut)) }, "Put")
 		assert.Panics(t, func() { eg.Delete(path, endpointFn(http.MethodDelete)) }, "Delete")
 	})
 
@@ -242,11 +245,13 @@ func TestEndpointGroup(t *testing.T) {
 		assert.NotPanics(t, func() { eg.Get(path, endpointFn(http.MethodGet)) }, "Get")
 		assert.NotPanics(t, func() { eg.Patch(path, endpointFn(http.MethodPatch)) }, "Patch")
 		assert.NotPanics(t, func() { eg.Post(path, endpointFn(http.MethodPost)) }, "Post")
+		assert.NotPanics(t, func() { eg.Put(path, endpointFn(http.MethodPut)) }, "Put")
 		assert.NotPanics(t, func() { eg.Delete(path, endpointFn(http.MethodDelete)) }, "Delete")
 
 		assert.Panics(t, func() { eg.Get(path, endpointFn(http.MethodGet)) }, "Get")
 		assert.Panics(t, func() { eg.Patch(path, endpointFn(http.MethodPatch)) }, "Patch")
 		assert.Panics(t, func() { eg.Post(path, endpointFn(http.MethodPost)) }, "Post")
+		assert.Panics(t, func() { eg.Put(path, endpointFn(http.MethodPut)) }, "Put")
 		assert.Panics(t, func() { eg.Delete(path, endpointFn(http.MethodDelete)) }, "Delete")
 	})
 
@@ -266,6 +271,7 @@ func TestEndpointGroup(t *testing.T) {
 		assert.NotPanics(t, func() { eg.Get(path, endpointFn(http.MethodGet)) }, "Get")
 		assert.Panics(t, func() { eg.Patch(path, endpointFn(http.MethodPatch)) }, "Patch")
 		assert.Panics(t, func() { eg.Post(path, endpointFn(http.MethodPost)) }, "Post")
+		assert.Panics(t, func() { eg.Put(path, endpointFn(http.MethodPut)) }, "Put")
 		assert.Panics(t, func() { eg.Delete(path, endpointFn(http.MethodDelete)) }, "Delete")
 	})
 
@@ -285,6 +291,7 @@ func TestEndpointGroup(t *testing.T) {
 		assert.NotPanics(t, func() { eg.Patch(path, endpointFn(http.MethodPatch)) }, "Patch")
 		assert.Panics(t, func() { eg.Get(path, endpointFn(http.MethodGet)) }, "Get")
 		assert.Panics(t, func() { eg.Post(path, endpointFn(http.MethodPost)) }, "Post")
+		assert.Panics(t, func() { eg.Put(path, endpointFn(http.MethodPut)) }, "Put")
 		assert.Panics(t, func() { eg.Delete(path, endpointFn(http.MethodDelete)) }, "Delete")
 	})
 }