From 4a59e11606574e48be2948dd26785fb9b06ec5e6 Mon Sep 17 00:00:00 2001
From: Andrea Tartaglia <me@andreatartaglia.com>
Date: Thu, 10 Oct 2024 09:50:02 +0200
Subject: [PATCH] add imdsv2 test image cleanup

---
 .../targets/ec2_ami_instance/tasks/main.yml          | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/tests/integration/targets/ec2_ami_instance/tasks/main.yml b/tests/integration/targets/ec2_ami_instance/tasks/main.yml
index 0950a64267..e9130bd735 100644
--- a/tests/integration/targets/ec2_ami_instance/tasks/main.yml
+++ b/tests/integration/targets/ec2_ami_instance/tasks/main.yml
@@ -371,6 +371,10 @@
         that:
           - imdsv2_info.images[0].imdsv2_enabled
 
+    - name: set image id fact for deletion later
+      ansible.builtin.set_fact:
+        ec2_ami_image_id_imdsv2: "{{ imdsv2_image.image_id }}"
+
   always:
     # ============================================================
 
@@ -403,6 +407,14 @@
         wait: true
       ignore_errors: true
 
+    - name: delete imdsv2 ami
+      amazon.aws.ec2_ami:
+        state: absent
+        image_id: "{{ ec2_ami_image_id_imdsv2 }}"
+        name: "{{ ec2_ami_name }}_imdsv2"
+        wait: true
+      ignore_errors: true
+
     - name: delete ami
       amazon.aws.ec2_ami:
         state: absent