feat: add projectSlug query to enable/disable spike protection endpoint

Author: jianyuan

sentry/sentry.go

@@ -7,7 +7,6 @@ import (
 	"errors"
 	"fmt"
 	"io"
-	"io/ioutil"
 	"net/http"
 	"net/url"
 	"reflect"
@@ -357,7 +356,7 @@ func CheckResponse(r *http.Response) error {
 	}
 
 	errorResponse := &ErrorResponse{Response: r}
-	data, err := ioutil.ReadAll(r.Body)
+	data, err := io.ReadAll(r.Body)
 	if err == nil && data != nil {
 		apiError := new(APIError)
 		json.Unmarshal(data, apiError)
@@ -368,7 +367,7 @@ func CheckResponse(r *http.Response) error {
 		}
 	}
 	// Re-populate error response body.
-	r.Body = ioutil.NopCloser(bytes.NewBuffer(data))
+	r.Body = io.NopCloser(bytes.NewBuffer(data))
 
 	switch {
 	case r.StatusCode == http.StatusTooManyRequests &&

sentry/sentry_test.go

@@ -4,7 +4,7 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
-	"io/ioutil"
+	"io"
 	"net/http"
 	"net/http/httptest"
 	"net/url"
@@ -272,7 +272,7 @@ func TestCheckResponse(t *testing.T) {
 			res := &http.Response{
 				Request:    &http.Request{},
 				StatusCode: http.StatusBadRequest,
-				Body:       ioutil.NopCloser(strings.NewReader(tc.body)),
+				Body:       io.NopCloser(strings.NewReader(tc.body)),
 			}
 
 			err := CheckResponse(res)
@@ -313,7 +313,7 @@ func TestCheckResponse_rateLimit(t *testing.T) {
 				Request:    &http.Request{},
 				StatusCode: http.StatusTooManyRequests,
 				Header:     http.Header{},
-				Body:       ioutil.NopCloser(strings.NewReader(`{"detail": "Rate limit exceeded"}`)),
+				Body:       io.NopCloser(strings.NewReader(`{"detail": "Rate limit exceeded"}`)),
 			}
 			tc.addHeaders(res)
 

sentry/spike_protections.go

@@ -9,11 +9,17 @@ import (
 type SpikeProtectionsService service
 
 type SpikeProtectionParams struct {
-	Projects []string `json:"projects"`
+	ProjectSlug string   `json:"-" url:"projectSlug,omitempty"`
+	Projects    []string `json:"projects" url:"-"`
 }
 
 func (s *SpikeProtectionsService) Enable(ctx context.Context, organizationSlug string, params *SpikeProtectionParams) (*Response, error) {
 	u := fmt.Sprintf("0/organizations/%v/spike-protections/", organizationSlug)
+	u, err := addQuery(u, params)
+	if err != nil {
+		return nil, err
+	}
+
 	req, err := s.client.NewRequest(http.MethodPost, u, params)
 	if err != nil {
 		return nil, err
@@ -24,6 +30,11 @@ func (s *SpikeProtectionsService) Enable(ctx context.Context, organizationSlug s
 
 func (s *SpikeProtectionsService) Disable(ctx context.Context, organizationSlug string, params *SpikeProtectionParams) (*Response, error) {
 	u := fmt.Sprintf("0/organizations/%v/spike-protections/", organizationSlug)
+	u, err := addQuery(u, params)
+	if err != nil {
+		return nil, err
+	}
+
 	req, err := s.client.NewRequest(http.MethodDelete, u, params)
 	if err != nil {
 		return nil, err