Merge pull request #11 from twharmon/response-fake

Response-fake
twharmon · Apr 11, 2022 · ef8810b · ef8810b
2 parents c7a3610 + 575cd24
commit ef8810b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/fakes/response.go b/fakes/response.go
@@ -1,6 +1,7 @@
 package fakes
 
 import (
+	"encoding/json"
 	"net/http"
 
 	"github.com/aws/aws-lambda-go/events"
@@ -20,6 +21,12 @@ func NewResponse() *Response {
 	}
 }
 
+func (r *Response) WithBody(v interface{}) {
+	b, err := json.Marshal(v)
+	r.response.Body = string(b)
+	r.err = err
+}
+
 func (r *Response) Respond() (*events.APIGatewayV2HTTPResponse, error) {
 	return r.response, r.err
 }