lambda_execute: allow function arn instead of name (#1275)

lambda_execute: allow function arn instead of name

SUMMARY
Fixes #1268
ISSUE TYPE

Bugfix Pull Request

COMPONENT NAME
lambda_execute
ADDITIONAL INFORMATION

Reviewed-by: Mark Chappell <None>

changelogs/fragments/1268-lambda-execute-arn.yml

@@ -0,0 +1,2 @@
+bugfixes:
+  - lambda_execute - Fix waiter error when ``function_arn`` is passed instead of ``name``(https://github.com/ansible-collections/amazon.aws/issues/1268).

plugins/modules/lambda_execute.py

@@ -212,7 +212,7 @@ def main():
         module.exit_json(changed=True)
 
     try:
-        wait_for_lambda(client, module, name)
+        wait_for_lambda(client, module, name or function_arn)
         response = client.invoke(**invoke_params, aws_retry=True)
     except is_boto3_error_code('ResourceNotFoundException') as nfe:
         module.fail_json_aws(nfe, msg="Could not find Lambda to execute. Make sure "
@@ -265,12 +265,12 @@ def main():
     module.exit_json(changed=True, result=results)
 
 
-def wait_for_lambda(client, module, name):
+def wait_for_lambda(client, module, name_or_arn):
     try:
         client_active_waiter = client.get_waiter('function_active')
         client_updated_waiter = client.get_waiter('function_updated')
-        client_active_waiter.wait(FunctionName=name)
-        client_updated_waiter.wait(FunctionName=name)
+        client_active_waiter.wait(FunctionName=name_or_arn)
+        client_updated_waiter.wait(FunctionName=name_or_arn)
     except botocore.exceptions.WaiterError as e:
         module.fail_json_aws(e, msg='Timeout while waiting on lambda to be Active')
     except (botocore.exceptions.ClientError, botocore.exceptions.BotoCoreError) as e:

tests/integration/targets/lambda/tasks/main.yml

@@ -184,6 +184,10 @@
       - result.configuration.tracing_config.mode == "PassThrough"
       - result.configuration.architectures == ['arm64']
 
+  - name: Save Lambda ARN
+    ansible.builtin.set_fact:
+      lambda_function_arn: "{{ result['configuration']['function_arn'] }}"
+
   - include_tasks: tagging.yml
 
   # Test basic operation of Uploaded lambda
@@ -212,6 +216,18 @@
       - result is not failed
       - result.result.output.message == "hello Mr Ansible Tests"
 
+  - name: test execute lambda with function arn
+    lambda_execute:
+      function_arn: "{{ lambda_function_arn }}"
+      payload:
+        name: Mr Ansible Tests
+    register: result
+  - name: assert lambda manages to respond as expected
+    assert:
+      that:
+      - result is not failed
+      - result.result.output.message == "hello Mr Ansible Tests"
+
   # Test updating Lambda
   - name: test lambda config updates (check mode)
     lambda:
@@ -644,7 +660,7 @@
     assert:
       that:
       - result is not failed
-  
+
   # Test creation with layers
   - name: Create temporary directory for testing
     tempfile: