Skip to content

Commit

Permalink
add integration test
Browse files Browse the repository at this point in the history
  • Loading branch information
mandar242 committed Aug 10, 2023
1 parent 92b0562 commit 62b3bcf
Showing 1 changed file with 50 additions and 0 deletions.
50 changes: 50 additions & 0 deletions tests/integration/targets/ec2_ami/tasks/main.yml
Original file line number Diff line number Diff line change
Expand Up @@ -697,6 +697,48 @@
- ami_facts_result_boot_tpm.images[0].boot_mode == 'uefi'
- ami_facts_result_boot_tpm.images[0].tpm_support == 'v2.0'

# === Test modify launch permissions org_arns and org_unit_arns=========================

- name: create an image from the instance
ec2_ami:
instance_id: '{{ ec2_instance_id }}'
state: present
name: '{{ ec2_ami_name }}_permissions'
description: '{{ ec2_ami_description }}'
tags:
Name: '{{ ec2_ami_name }}_permissions'
wait: yes
root_device_name: '{{ ec2_ami_root_disk }}'
register: permissions_create_result

- name: modify the AMI launch permissions
ec2_ami:
state: present
image_id: '{{ permissions_create_result.image_id }}'
name: '{{ ec2_ami_name }}_permissions'
tags:
Name: '{{ ec2_ami_name }}_permissions'
launch_permissions:
org_arns: ['arn:aws:organizations::123456789012:organization/o-123ab4cdef']
org_unit_arns: ['arn:aws:organizations::123456789012:ou/o-123example/ou-1234-5exampld']
register: permissions_update_result

- name: Get ami info
amazon.aws.ec2_ami_info:
image_ids: '{{ permissions_create_result.image_id }}'
describe_image_attributes: true
register: permissions_info_result

- name: assert that launch permissions have changed
assert:
that:
- "permissions_create_result.changed"
- "'organization_arn' in permissions_info_result.images[0].launch_permissions[0]"
- "permissions_info_result.images[0].launch_permissions[0]['organization_arn'] == 'arn:aws:organizations::123456789012:organization/o-123ab4cdef'"
- "'organizational_unit_arn' in permissions_info_result.images[0].launch_permissions[1]"
- "permissions_info_result.images[0].launch_permissions[1]['organizational_unit_arn'] == 'arn:aws:organizations::123456789012:ou/o-123example/ou-1234-5exampld'"


# ============================================================

always:
Expand Down Expand Up @@ -737,6 +779,14 @@
wait: yes
ignore_errors: yes

- name: delete ami
ec2_ami:
state: absent
image_id: "{{ ec2_ami_image_id }}"
name: '{{ ec2_ami_name }}_permissions'
wait: yes
ignore_errors: yes

- name: remove setup snapshot of ec2 instance
ec2_snapshot:
state: absent
Expand Down

0 comments on commit 62b3bcf

Please sign in to comment.