Merge pull request #8 from twharmon/fakes

Fix responer interface implementation

fakes/response.go

@@ -1,9 +1,16 @@
 package fakes
 
-import "github.com/aws/aws-lambda-go/events"
+import (
+	"net/http"
+
+	"github.com/aws/aws-lambda-go/events"
+	"github.com/twharmon/golamb"
+)
 
 type Response struct {
 	response *events.APIGatewayV2HTTPResponse
+	cookie   []*http.Cookie
+	header   map[string]string
 	err      error
 }
 
@@ -17,3 +24,13 @@ func NewResponse(response *events.APIGatewayV2HTTPResponse, err error) *Response
 func (r *Response) Respond() (*events.APIGatewayV2HTTPResponse, error) {
 	return r.response, r.err
 }
+
+func (r *Response) SetCookie(cookie *http.Cookie) golamb.Responder {
+	r.cookie = append(r.cookie, cookie)
+	return r
+}
+
+func (r *Response) SetHeader(key string, val string) golamb.Responder {
+	r.header[key] = val
+	return r
+}