Merge pull request #5 from twharmon/test-handler

More tests

golamb.go

@@ -16,8 +16,13 @@ type Config struct {
 }
 
 func Start(handlerFunc WrappedHandler, config ...*Config) {
+	h := getHandler(handlerFunc, config...)
+	lambda.StartHandler(h)
+}
+
+func getHandler(handlerFunc WrappedHandler, config ...*Config) lambda.Handler {
 	cfg := getConfig(config...)
-	h := &handler{
+	return &handler{
 		cfg: cfg,
 		handler: func(r *events.APIGatewayV2HTTPRequest) (resp *events.APIGatewayV2HTTPResponse, err error) {
 			ctx := &handlerContext{
@@ -32,7 +37,6 @@ func Start(handlerFunc WrappedHandler, config ...*Config) {
 			return handlerFunc(ctx).Respond()
 		},
 	}
-	lambda.StartHandler(h)
 }
 
 func getConfig(configs ...*Config) *Config {

golamb_test.go

@@ -1,6 +1,8 @@
 package golamb
 
 import (
+	"context"
+	"encoding/json"
 	"errors"
 	"net/http"
 	"reflect"
@@ -9,6 +11,31 @@ import (
 	"github.com/aws/aws-lambda-go/events"
 )
 
+func TestGetHandler(t *testing.T) {
+	f := func(c Context) Responder {
+		return c.Response(http.StatusOK)
+	}
+	h := getHandler(f)
+	payload, err := json.Marshal(&events.APIGatewayV2HTTPRequest{})
+	if err != nil {
+		t.Fatalf("unexpected err: %s", err)
+	}
+	resp, err := h.Invoke(context.Background(), payload)
+	if err != nil {
+		t.Fatalf("unexpected err: %s", err)
+	}
+	var got events.APIGatewayV2HTTPResponse
+	if err := json.Unmarshal(resp, &got); err != nil {
+		t.Fatalf("unexpected err: %s", err)
+	}
+	want := events.APIGatewayV2HTTPResponse{
+		StatusCode: http.StatusOK,
+	}
+	if !reflect.DeepEqual(want, got) {
+		t.Fatalf("want %v; got %v", want, got)
+	}
+}
+
 func TestGetConfigEmpty(t *testing.T) {
 	want := &Config{
 		AWSServiceProvider: &AWSServiceProviderConfig{},