Body not repeat issue resolved

config_test.go

@@ -110,7 +110,10 @@ func TestConfig(t *testing.T) {
 					w.Header().Add("intercepted", "true")
 					w.Header().Add(headers.ContentType, mimetypes.TextPlain)
 					w.WriteHeader(http.StatusBadRequest)
-					w.Write([]byte(msg))
+					_, err := w.Write([]byte(msg))
+					if err != nil {
+						log.Fatal(err)
+					}
 				})
 		}
 

cors/cors_test.go

@@ -19,7 +19,10 @@ func TestCORS(t *testing.T) {
 	handler := func(w http.ResponseWriter, r *http.Request) {
 		w.Header().Add("content-type", "text/plain")
 		w.WriteHeader(http.StatusOK)
-		w.Write([]byte(msg))
+		_, err := w.Write([]byte(msg))
+		if err != nil {
+			log.Fatal(err)
+		}
 	}
 
 	t.Run("should allow request", func(t *testing.T) {

http_handler.go

@@ -2,7 +2,9 @@ package mocha
 
 import (
 	"bufio"
+	"bytes"
 	"fmt"
+	"io/ioutil"
 	"net/http"
 	"strings"
 	"time"
@@ -100,7 +102,13 @@ func (h *mockHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 		respondError(w, r, h.evt, err)
 		return
 	}
-
+	copy, err := ioutil.ReadAll(res.Body)
+	if err != nil {
+		h.t.Logf(err.Error())
+		respondError(w, r, h.evt, err)
+		return
+	}
+	res.Body = ioutil.NopCloser(bytes.NewBuffer(copy))
 	// map the response using mock mappers.
 	mapperArgs := reply.ResponseMapperArgs{Request: r, Parameters: h.params}
 	for _, mapper := range res.Mappers {
@@ -127,12 +135,16 @@ func (h *mockHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	if res.Body != nil {
 		scanner := bufio.NewScanner(res.Body)
 		for scanner.Scan() {
-			w.Write(scanner.Bytes())
+			_, err := w.Write(scanner.Bytes())
+			if err != nil {
+				h.t.Logf("error writing reponse body: error=%v", err.Error())
+			}
 		}
 
 		if scanner.Err() != nil {
 			h.t.Logf("error writing response body: error=%v", scanner.Err())
 		}
+		res.Body = ioutil.NopCloser(bytes.NewBuffer(copy))
 	}
 
 	// run post actions.
@@ -183,7 +195,7 @@ func respondNonMatched(w http.ResponseWriter, r *http.Request, result *findResul
 
 	w.Header().Add(headers.ContentType, mimetypes.TextPlain)
 	w.WriteHeader(http.StatusTeapot)
-	w.Write([]byte(builder.String()))
+	_, _ = w.Write([]byte(builder.String()))
 }
 
 func respondError(w http.ResponseWriter, r *http.Request, evt *hooks.Emitter, err error) {
@@ -192,6 +204,6 @@ func respondError(w http.ResponseWriter, r *http.Request, evt *hooks.Emitter, er
 	w.Header().Add(headers.ContentType, mimetypes.TextPlain)
 	w.WriteHeader(http.StatusTeapot)
 
-	w.Write([]byte("Request did not match. An error occurred.\n"))
-	w.Write([]byte(fmt.Sprintf("%v", err)))
+	_, _ = w.Write([]byte("Request did not match. An error occurred.\n"))
+	_, _ = w.Write([]byte(fmt.Sprintf("%v", err)))
 }

internal/middleware/compose_test.go

@@ -20,7 +20,10 @@ func TestMiddlewaresComposition(t *testing.T) {
 		w.Header().Add("x-two", r.Header.Get("x-two"))
 		w.Header().Add("content-type", "text/plain")
 		w.WriteHeader(http.StatusOK)
-		w.Write([]byte(msg))
+		_, err := w.Write([]byte(msg))
+		if err != nil {
+			log.Fatal(err)
+		}
 	}
 
 	ts := httptest.NewServer(Compose(one, two, recover.Recover).Root(http.HandlerFunc(fn)))

internal/middleware/recover/recover.go

@@ -21,8 +21,8 @@ func Recover(next http.Handler) http.Handler {
 
 				w.Header().Set(headers.ContentType, mimetypes.TextPlain)
 				w.WriteHeader(http.StatusTeapot)
-				w.Write([]byte(fmt.Sprintf("%s - an unexpected error has occurred", http.StatusText(http.StatusTeapot))))
-				w.Write([]byte(fmt.Sprintf("%v", recovery)))
+				_, _ = w.Write([]byte(fmt.Sprintf("%s - an unexpected error has occurred", http.StatusText(http.StatusTeapot))))
+				_, _ = w.Write([]byte(fmt.Sprintf("%v", recovery)))
 			}
 		}()
 

reply/proxied_test.go

@@ -23,7 +23,10 @@ func TestForward(t *testing.T) {
 			assert.Equal(t, "proxied", r.Header.Get("x-proxy"))
 
 			w.WriteHeader(http.StatusCreated)
-			w.Write([]byte("hello world"))
+			_, err := w.Write([]byte("hello world"))
+			if err != nil {
+				t.Fatal(err)
+			}
 		}))
 
 		defer dest.Close()
@@ -61,7 +64,7 @@ func TestForward(t *testing.T) {
 				t.Fatal(err)
 			}
 
-			w.Write(b)
+			_, _ = w.Write(b)
 		}))
 
 		defer dest.Close()

test/proxied_responses_test.go

@@ -29,7 +29,10 @@ func TestForward(t *testing.T) {
 		}
 
 		w.WriteHeader(http.StatusOK)
-		w.Write(b)
+		_, err = w.Write(b)
+		if err != nil {
+			t.Fatal(err)
+		}
 	}))
 
 	defer dest.Close()