From 3ed24b09a3d9e560f6f4b6ed3385ea22f54cb376 Mon Sep 17 00:00:00 2001
From: tbaev <teodor.baev@suse.com>
Date: Mon, 30 Sep 2024 17:10:31 +0300
Subject: [PATCH] Enable 12sp5 ltss and ltss_es virt tests

---
 data/virtualization/autoyast/guest_12.xml.ep     | 11 ++++++++++-
 data/virtualization/autoyast/host_12.xml.ep      |  9 ++++++++-
 lib/main_common.pm                               |  2 ++
 lib/virt_autotest/common.pm                      | 10 ++++++++--
 tests/virtualization/universal/prepare_guests.pm |  2 ++
 5 files changed, 30 insertions(+), 4 deletions(-)

diff --git a/data/virtualization/autoyast/guest_12.xml.ep b/data/virtualization/autoyast/guest_12.xml.ep
index 17448717863b..38e55feadff6 100644
--- a/data/virtualization/autoyast/guest_12.xml.ep
+++ b/data/virtualization/autoyast/guest_12.xml.ep
@@ -51,7 +51,7 @@
     <arch>{{ARCH}}</arch>
     <release_type/>
     </addon>
-    % if ($ltss_code) {
+    % if ($ltss_code && $check_var->('EXTENDED_SECURITY', '0') || $ltss_code && !$ltss_es_code && $check_var->('EXTENDED_SECURITY', '1')) {
     <addon>
       <name>SLES-LTSS</name>
       <version>{{VERSION}}</version>
@@ -60,6 +60,15 @@
       <release_type/>
     </addon>
     % }
+    % if ($ltss_es_code && $check_var->('EXTENDED_SECURITY', '1')) {
+    <addon>
+      <name>SLES-LTSS-Extended-Security</name>
+      <version>{{VERSION}}</version>
+      <arch>{{ARCH}}</arch>
+      <reg_code><%= $ltss_es_code %></reg_code>
+      <release_type/>
+    </addon>
+    % }
     </addons>
   </suse_register>
   <add-on>
diff --git a/data/virtualization/autoyast/host_12.xml.ep b/data/virtualization/autoyast/host_12.xml.ep
index 5073b2c388cd..7a35ac34a1d5 100644
--- a/data/virtualization/autoyast/host_12.xml.ep
+++ b/data/virtualization/autoyast/host_12.xml.ep
@@ -53,7 +53,11 @@
     <addons config:type="list">
       % while (my ($key, $addon) = each (%$addons)) {
       <addon>
+        % if ($check_var->('EXTENDED_SECURITY', '1')) {
+        <name>SLES-LTSS-Extended-Security</name>
+        % } else {
         <name><%= $addon->{name} %></name>
+        % }
         <version><%= $addon->{version} %></version>
         <arch><%= $addon->{arch} %></arch>
         <release_type/>
@@ -66,9 +70,12 @@
         % if ($key eq 'rt') {
         <reg_code><%= $get_var->('SCC_REGCODE_RT') %></reg_code>
         % }
-        % if ($key eq 'ltss') {
+        % if ($key eq 'ltss' and $check_var->('EXTENDED_SECURITY', '0')) {
         <reg_code><%= $get_var->('SCC_REGCODE_LTSS') %></reg_code>
         % }
+        % if ($key eq 'ltss' and $check_var->('EXTENDED_SECURITY', '1')) {
+        <reg_code><%= $get_var->('SCC_REGCODE_LTSS_ES') %></reg_code>
+        % }
       </addon>
       % }
     </addons>
diff --git a/lib/main_common.pm b/lib/main_common.pm
index db15d4e2239e..9c02a6aca84d 100644
--- a/lib/main_common.pm
+++ b/lib/main_common.pm
@@ -2402,6 +2402,8 @@ sub set_mu_virt_vars {
     bmwqemu::save_vars();
     diag("BUILD is $BUILD, UPDATE_PACKAGE is set to " . get_var('UPDATE_PACKAGE', ''));
 
+    # Check if repo is LTSS-Extended-Security and sets EXTENDED_SECURITY to 1
+    set_var('EXTENDED_SECURITY', (get_var('INCIDENT_REPO') =~ /LTSS-Extended-Security/) ? 1 : 0);
     # Set PATCH_WITH_ZYPPER
     set_var('PATCH_WITH_ZYPPER', 1) unless (check_var('PATCH_WITH_ZYPPER', 0));
 }
diff --git a/lib/virt_autotest/common.pm b/lib/virt_autotest/common.pm
index 752054986034..25e34c1573be 100644
--- a/lib/virt_autotest/common.pm
+++ b/lib/virt_autotest/common.pm
@@ -243,6 +243,9 @@ if (get_var("REGRESSION", '') =~ /xen/) {
         sles12sp5 => {
             name => 'sles12sp5',
         },
+        sles12sp5ES => {
+            name => 'sles12sp5ES',
+        },
         sles15sp2 => {
             name => 'sles15sp2',
         },
@@ -262,7 +265,7 @@ if (get_var("REGRESSION", '') =~ /xen/) {
             name => 'sles15sp6',
         },
     );
-    %guests = get_var('TERADATA') ? %guests{"sles${guest_version}TD"} : %guests{"sles${guest_version}"};
+    %guests = get_var('TERADATA') ? %guests{"sles${guest_version}TD"} : $check_var->('EXTENDED_SECURITY', '1') ? %guests{"sles${guest_version}ES"} : %guests{"sles${guest_version}"};
 
 } elsif (get_var("REGRESSION", '') =~ /hyperv/) {
     %guests = (
@@ -272,6 +275,9 @@ if (get_var("REGRESSION", '') =~ /xen/) {
         sles12sp5 => {
             vm_name => 'sles-12.5_openQA-virtualization-maintenance',
         },
+        sles12sp5ES => {
+            vm_name => 'sles-12.5_openQA-virtualization-maintenance-ES',
+        },
         sles15sp2 => {
             vm_name => 'sles-15.2_openQA-virtualization-maintenance',
         },
@@ -291,7 +297,7 @@ if (get_var("REGRESSION", '') =~ /xen/) {
             vm_name => 'sles-15.6_openQA-virtualization-maintenance',
         },
     );
-    %guests = get_var('TERADATA') ? %guests{"sles${guest_version}TD"} : %guests{"sles${guest_version}"};
+    %guests = get_var('TERADATA') ? %guests{"sles${guest_version}TD"} : $check_var->('EXTENDED_SECURITY', '1') ? %guests{"sles${guest_version}ES"} : %guests{"sles${guest_version}"};
 }
 
 our %imports = ();    # imports are virtual machines that we don't install but just import. We test those separately.
diff --git a/tests/virtualization/universal/prepare_guests.pm b/tests/virtualization/universal/prepare_guests.pm
index d3194da78f76..96eca251cf92 100644
--- a/tests/virtualization/universal/prepare_guests.pm
+++ b/tests/virtualization/universal/prepare_guests.pm
@@ -29,6 +29,7 @@ sub create_profile {
     my $path = $version >= 15 ? "virtualization/autoyast/guest_15.xml.ep" : "virtualization/autoyast/guest_12.xml.ep";
     my $scc_code = get_required_var("SCC_REGCODE");
     my %ltss_products = @{get_var_array("LTSS_REGCODES_SECRET")};
+    my %ltss_es_products = @{get_var_array("LTSS_ES_REGCODES_SECRET")};
     my $ca_str = "SLE_" . $version =~ s/\./_SP/r;
     my $sut_ip = get_required_var("SUT_IP");
     my $profile = get_test_data($path);
@@ -47,6 +48,7 @@ sub create_profile {
     my $vars = {
         vm_name => $vm_name,
         ltss_code => $ltss_products{$version},
+        ltss_es_code => $ltss_es_products{$version},
         repos => [split(/,/, $incident_repos)],
         check_var => \&check_var,
         get_var => \&get_required_var