From 6aaa0e87e4392fa09090b75351141f55d348ce88 Mon Sep 17 00:00:00 2001
From: Amit Karsale <amit.karsale@perforce.com>
Date: Wed, 5 Jun 2024 19:41:54 +0530
Subject: [PATCH] (PA-6276) Update puppet_agent module task acceptance for AL2

---
 task_spec/spec/acceptance/init_spec.rb | 1 +
 tasks/install_shell.sh                 | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/task_spec/spec/acceptance/init_spec.rb b/task_spec/spec/acceptance/init_spec.rb
index 87dc8f22..b686a4be 100644
--- a/task_spec/spec/acceptance/init_spec.rb
+++ b/task_spec/spec/acceptance/init_spec.rb
@@ -46,6 +46,7 @@ def latest_platform_list
       el-9-aarch64|
       ubuntu-22\.04-aarch64|
       amazon-2023|
+      amazon-2|
       osx-14|
       debian-12|
       debian-11-aarch64|
diff --git a/tasks/install_shell.sh b/tasks/install_shell.sh
index 3750f374..1a55b62a 100644
--- a/tasks/install_shell.sh
+++ b/tasks/install_shell.sh
@@ -663,7 +663,7 @@ case $platform in
     filetype="rpm"
     platform_package="el"
     # For Amazon Linux 2023 and onwards we can use the 'amazon' packages created instead of 'el' packages
-    if (( $platform_version >= 2023 )); then
+    if (( $platform_version == 2023 || $platform_version == 7 )); then
       platform_package="amazon"
     fi
     filename="${collection}-release-${platform_package}-${platform_version}.noarch.rpm"