update test helpers for amazon 2023 and ubuntu 2404 (#661)

* update test helpers for amazon 2023 and ubuntu 2404

* f

Author: laverya

tgrun/pkg/runner/vmi/embed/testhelpers.sh

@@ -596,6 +596,26 @@ function is_rhel_9_variant() {
     return 1
 }
 
+# is_amazon_2023 returns 0 if the current distro is Amazon 2023.
+function is_amazon_2023() {
+    if grep '^ID=' /etc/os-release | grep -qE '(amzn)' ; then
+        if grep '^VERSION_ID=' /etc/os-release | grep -q '=*"2023' ; then
+            return 0
+        fi
+    fi
+    return 1
+}
+
+# is_ubuntu_2404 returns 0 if the current distro is Ubuntu 24.04.
+function is_ubuntu_2404() {
+    if grep '^ID=' /etc/os-release | grep -qE '(ubuntu)' ; then
+        if grep '^VERSION_ID=' /etc/os-release | grep -q '=*"24.04' ; then
+            return 0
+        fi
+    fi
+    return 1
+}
+
 function rhel_9_install_host_packages() {
     local packages=("$@")
     if is_rhel_9_variant ; then
@@ -604,6 +624,23 @@ function rhel_9_install_host_packages() {
     fi
 }
 
+function amazon_2023_install_host_packages() {
+    local packages=("$@")
+    if is_amazon_2023 ; then
+      # install required host packages
+      yum install -y "${packages[@]}"
+    fi
+}
+
+function ubuntu_2024_install_host_packages() {
+    local packages=("$@")
+    if is_ubuntu_2404 ; then
+      # install required host packages
+      apt update
+      apt install -y "${packages[@]}"
+    fi
+}
+
 function installer_host() {
     if echo "$1" | grep -q "staging" ; then
         echo "staging.kurl.sh"