fix: adding more robust testing of consumable request body

Author: dotkas

pkg/networking/middleware/response.go

@@ -31,7 +31,6 @@ func NewReponseMiddleware(roundTriper http.RoundTripper, config configuration.Co
 
 func (rm ResponseMiddleware) RoundTrip(req *http.Request) (*http.Response, error) {
 	res, err := rm.next.RoundTrip(req)
-
 	if err != nil {
 		return res, err
 	}
@@ -43,7 +42,7 @@ func (rm ResponseMiddleware) RoundTrip(req *http.Request) (*http.Response, error
 	return res, err
 }
 
-// HandleResponse maps the response param to the eror catalog error.
+// HandleResponse maps the response param to the error catalog error.
 func HandleResponse(res *http.Response, config configuration.Configuration) error {
 	if res == nil {
 		return nil
@@ -75,6 +74,7 @@ func getErrorList(res *http.Response) []snyk_errors.Error {
 	if err != nil {
 		return []snyk_errors.Error{}
 	}
+
 	res.Body = io.NopCloser(bytes.NewBuffer(bodyBytes))
 
 	errorList, err := snyk_errors.FromJSONAPIErrorBytes(bodyBytes)

pkg/networking/middleware/response_test.go

@@ -3,6 +3,7 @@ package middleware_test
 import (
 	"context"
 	"fmt"
+	"io"
 	"net/http"
 	"net/http/httptest"
 	"testing"
@@ -204,6 +205,50 @@ func Test_ResponseMiddleware(t *testing.T) {
 		assert.NoError(t, err)
 		assert.Equal(t, http.StatusOK, res.StatusCode)
 	})
+
+	t.Run("response body should be consumable after roundtrip", func(t *testing.T) {
+		testCases := []struct {
+			name        string
+			urlPath     string
+			expectError bool
+		}{
+			{
+				name:        "no error",
+				urlPath:     "",
+				expectError: false,
+			},
+			{
+				name:        "with error",
+				urlPath:     "/error-catalog",
+				expectError: true,
+			},
+		}
+
+		for _, tc := range testCases {
+			t.Run(tc.name, func(t *testing.T) {
+				config := getBaseConfig()
+				config.Set(configuration.AUTHENTICATION_ADDITIONAL_URLS, []string{server.URL})
+
+				rt := middleware.NewReponseMiddleware(http.DefaultTransport, config, errHandler)
+				req := buildRequest(server.URL + tc.urlPath)
+				res, err := rt.RoundTrip(req)
+
+				assert.NotNil(t, res)
+				if tc.expectError {
+					assert.Error(t, err)
+				} else {
+					assert.NoError(t, err)
+				}
+
+				bodyBytes, err := io.ReadAll(res.Body)
+				assert.NoError(t, err, "Body should be readable")
+				assert.NotNil(t, bodyBytes, "Should be able to read body")
+
+				err = res.Body.Close()
+				assert.NoError(t, err, "Body should close without errors")
+			})
+		}
+	})
 }
 
 func buildRequest(url string) *http.Request {