Added tests

Author: cloudshiftchris

demux/event.go

@@ -54,7 +54,10 @@ func processEvent(
 	args[1] = reflect.ValueOf(event)
 	returnValues := handlerFn.Call(args)
 
-	return returnValues[0].Interface(), returnValues[1].Interface().(error)
+	if errVal, ok := returnValues[1].Interface().(error); ok && errVal != nil {
+		return nil, errVal
+	}
+	return returnValues[0].Interface(), nil
 }
 
 func createEvent(rawEvent RawEvent, eventFactories []Factory) any {

demux/event_test.go

@@ -47,3 +47,36 @@ func TestCreateEventNoFactory(t *testing.T) {
 	event := createEvent(nil, eventFactories)
 	assert.Nil(t, event)
 }
+
+func TestProcessEventWorks(t *testing.T) {
+	var handlerCalled = false
+	handlerMap, err := createHandlerMap(
+		[]any{
+			func(ctx context.Context, event *events.APIGatewayProxyRequest) (
+				*events.APIGatewayProxyResponse,
+				error) {
+				handlerCalled = true
+				return &events.APIGatewayProxyResponse{}, nil
+			}})
+	assert.NoError(t, err)
+	assert.NotNil(t, handlerMap)
+
+	var factoryCalled = false
+	cfg := &demuxCfg{
+		factories: []Factory{
+			func(ctx *EventContext) any {
+				factoryCalled = true
+				return &events.APIGatewayProxyRequest{}
+			},
+		},
+		handlerMap: handlerMap,
+	}
+
+	rawEvent := map[string]any{}
+
+	resp, err := processEvent(cfg, context.TODO(), rawEvent)
+	assert.NoError(t, err)
+	assert.NotNil(t, resp)
+	assert.True(t, handlerCalled)
+	assert.True(t, factoryCalled)
+}