Add test to reply POST body

Author: bcomnes

go/porcelain/http/http_test.go

@@ -3,6 +3,7 @@ package http
 import (
 	"errors"
 	"fmt"
+	"io/ioutil"
 	"net/http"
 	"net/http/httptest"
 	"net/url"
@@ -151,7 +152,7 @@ func TestRetryableTransport_POST(t *testing.T) {
 	defer server.Close()
 
 	rwrtr := runtime.ClientRequestWriterFunc(func(req runtime.ClientRequest, _ strfmt.Registry) error {
-		return req.SetBodyParam("test result")
+		return req.SetBodyParam("test body")
 	})
 
 	hu, _ := url.Parse(server.URL)
@@ -178,3 +179,59 @@ func TestRetryableTransport_POST(t *testing.T) {
 	actual := result.(string)
 	require.EqualValues(t, "test result", actual)
 }
+
+func TestRetryableTransportWithRetry_POST(t *testing.T) {
+	attempts := 0
+
+	server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {
+		attempts++
+
+		if attempts == 1 {
+			reset := fmt.Sprintf("%d", time.Now().Add(1*time.Second).Unix())
+			body, err := ioutil.ReadAll(req.Body)
+			require.NoError(t, err)
+			require.Equal(t, "\"test body\"\n", string(body))
+			rw.Header().Set("X-RateLimit-Reset", reset)
+			rw.WriteHeader(http.StatusTooManyRequests)
+			rw.Write([]byte("rate limited"))
+		} else {
+			body, err := ioutil.ReadAll(req.Body)
+			require.NoError(t, err)
+			require.Equal(t, "\"test body\"\n", string(body))
+			rw.WriteHeader(http.StatusOK)
+			rw.Write([]byte("ok"))
+		}
+	}))
+	defer server.Close()
+
+	rwrtr := runtime.ClientRequestWriterFunc(func(req runtime.ClientRequest, _ strfmt.Registry) error {
+		return req.SetBodyParam("test body")
+	})
+
+	hu, _ := url.Parse(server.URL)
+	rt := NewRetryableTransport(httptransport.New(hu.Host, "/", []string{"http"}), 2)
+
+	res, err := rt.Submit(&runtime.ClientOperation{
+		ID:          "getSites",
+		Method:      "POST",
+		PathPattern: "/",
+		Params:      rwrtr,
+		Reader: runtime.ClientResponseReaderFunc(func(response runtime.ClientResponse, consumer runtime.Consumer) (interface{}, error) {
+			if response.Code() == 200 {
+				var result string
+				if err := consumer.Consume(response.Body(), &result); err != nil {
+					return nil, err
+				}
+				return result, nil
+			}
+			return nil, errors.New("Generic error")
+		}),
+	})
+
+	require.NoError(t, err)
+	require.Equal(t, 2, attempts)
+
+	require.IsType(t, "", res)
+	actual := res.(string)
+	require.EqualValues(t, "ok", actual)
+}