Add missing part of previous commit; restore InvokeFunction

gruntwork-io · Mar 29, 2021 · 0afc01b · 0afc01b
1 parent 4ce8177
commit 0afc01b
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/modules/aws/lambda.go b/modules/aws/lambda.go
@@ -45,10 +45,9 @@ type LambdaOutput struct {
 
 // InvokeFunction invokes a lambda function.
 func InvokeFunction(t testing.TestingT, region, functionName string, payload interface{}) []byte {
-	input := &LambdaOptions{Payload: &payload}
-	out, err := InvokeFunctionWithParams(t, region, functionName, input)
+	out, err := InvokeFunctionE(t, region, functionName, payload)
 	require.NoError(t, err)
-	return out.Payload
+	return out
 }
 
 // InvokeFunctionE invokes a lambda function.
@@ -93,7 +92,10 @@ func InvokeFunctionWithParams(t testing.TestingT, region, functionName string, i
 		}
 	}
 
-	invokeInput := &lambda.InvokeInput{InvocationType: &invocationType}
+	invokeInput := &lambda.InvokeInput{
+		FunctionName:   &functionName,
+		InvocationType: &invocationType,
+	}
 
 	if input.Payload != nil {
 		payloadJson, err := json.Marshal(input.Payload)