AISTACK_BASIC test

lib/main_micro_alp.pm

@@ -16,11 +16,14 @@ use main_ltp_loader 'load_kernel_tests';
 use main_containers qw(load_container_tests is_container_test load_container_engine_test);
 use main_publiccloud qw(load_publiccloud_download_repos);
 use main_security qw(load_security_tests is_security_test);
-use testapi qw(check_var get_required_var get_var set_var);
+use testapi qw(check_var get_required_var get_var set_var record_info);
+use testapi;
 use version_utils;
 use utils;
 use Utils::Architectures;
 use Utils::Backends;
+use Data::Dumper;
+
 
 sub is_image {
     return get_required_var('FLAVOR') =~ /image|default|kvm|base/i;
@@ -299,6 +302,7 @@ sub load_slem_on_pc_tests {
     } elsif (get_var('PUBLIC_CLOUD_UPLOAD_IMG')) {
         loadtest("boot/boot_to_desktop");
         loadtest("publiccloud/upload_image");
+        # AISTACK test verification
     } else {
         # SLEM basic test
         loadtest("boot/boot_to_desktop");
@@ -311,6 +315,8 @@ sub load_slem_on_pc_tests {
         }
         if (get_var('PUBLIC_CLOUD_LTP', 0)) {
             loadtest("publiccloud/run_ltp", run_args => $args);
+        } elsif (get_var('PUBLIC_CLOUD_AISTACK')) {
+            loadtest('publiccloud/aistack_basic', run_args => $ai_args);
         } elsif (is_container_test) {
             loadtest("publiccloud/ssh_interactive_start", run_args => $args);
             loadtest("publiccloud/instance_overview", run_args => $args);
@@ -321,10 +327,8 @@ sub load_slem_on_pc_tests {
                 $run_args->{runtime} = $_;
                 load_container_engine_test($run_args);
             }
-
             loadtest("publiccloud/ssh_interactive_end", run_args => $args);
-        }
-        else {
+        } else {
             loadtest "publiccloud/check_services", run_args => $args;
             loadtest("publiccloud/slem_basic", run_args => $args);
         }

tests/publiccloud/aistack_basic.pm

@@ -0,0 +1,128 @@
+# SUSE's openQA tests
+#
+# Copyright 2022 SUSE LLC
+# SPDX-License-Identifier: FSFAP
+
+# Basic aistack test
+
+# Summary: This test performs the following actions
+#  - Create a VM in EC2 using SLE-Micro-6-0-BYOS.x86_64-1.0.0-EC2-Build1.36.raw.xz
+#  - Install the required dependencies to install the aistack helm chart and containers
+#  - Test access to OpenWebUI and run integration tests with Ollama and MilvusDB
+# Maintainer: Yogalakshmi Arunachalam <[email protected]>
+#
+
+use Mojo::Base 'publiccloud::basetest';
+use testapi;
+use serial_terminal 'select_serial_terminal';
+use publiccloud::utils qw(is_byos registercloudguest register_openstack);
+use publiccloud::ssh_interactive 'select_host_console';
+use strict;
+use warnings;
+use utils;
+use publiccloud::utils;
+use File::Basename;
+use version_utils;
+use Data::Dumper;
+
+sub ins_dep_pkg {
+    my ($instance, $package) = @_;
+
+    record_info("Install $package",
+        "Install package $package using transactional server and reboot");
+    trup_install($package);
+
+    $instance->ssh_assert_script_run("sudo systemctl start $package",
+        timeout => 100);
+    $instance->ssh_assert_script_run("sudo systemctl enable $package",
+        timeout => 100);
+    $instance->ssh_assert_script_run("sudo systemctl status $package",
+        timeout => 100);
+    record_info("Installed $package");
+}
+
+sub create_aistack {
+    my ($instance, $rke2_url, $kubectl_url, $helm_url) = @_;
+
+    # Install dependencies
+    record_info('Refresh and update');
+    $instance->ssh_assert_script_run('sudo zypper ref; sudo zypper -n up',
+        timeout => 1000);
+    sleep 90;    # Wait for zypper to be available
+
+    ins_dep_pkg($instance, "curl");
+    ins_dep_pkg($instance, "docker");
+
+    # Install RKE2
+    $instance->ssh_assert_script_run("curl -sfL $rke2_url | sh",
+        timeout => 1000);
+    $instance->ssh_assert_script_run('sudo systemctl enable rke2-server',
+        timeout => 100);
+    $instance->ssh_assert_script_run('sudo systemctl start rke2-server',
+        timeout => 100);
+    $instance->ssh_assert_script_run('sudo systemctl status rke2-server',
+        timeout => 100);
+    record_info('Installed RKE2');
+
+    # Install kubectl
+    $instance->ssh_assert_script_run("curl -LO $kubectl_url",
+        timeout => 1000);
+    $instance->ssh_assert_script_run('chmod +x ./kubectl', timeout => 100);
+    $instance->ssh_assert_script_run('sudo mv ./kubectl /usr/local/bin/kubectl',
+        timeout => 100);
+    $instance->ssh_assert_script_run('kubectl version --client',
+        timeout => 100);
+    record_info('kubectl setup complete');
+
+    # Install Helm
+    $instance->ssh_assert_script_run("curl $helm_url | bash",
+        timeout => 1000);
+    record_info('Helm installation complete');
+}
+
+sub run {
+    my ($self, $args) = @_;
+
+    # Required tools
+    my $ins_rke2 = get_var('RKE2_URL');
+    my $ins_kubectl = get_var('KUBECTL_URL');
+    my $ins_helm = get_var('HELM_URL');
+
+    # Create AWS instance
+    my $provider;
+    my $instance;
+    # credentials from openwebui
+    # my $admin_username = get_var('OPENWEBUI_ADMIN');
+    # my $admin_password = get_var('OPENWEBUI_PASSWD');
+    # my $openwebui_hostname = get_var('OPENWEBUI_HOSTNAME');
+
+    select_host_console();
+
+    record_info("DEBUG", "Args passed: " . Dumper($args));
+
+    unless ($args->{my_provider} && $args->{my_instance}) {
+        $args->{my_provider} = $self->provider_factory();
+        $args->{my_instance} = $args->{my_provider}->create_instance();
+    }
+
+    $instance = $args->{my_instance};
+    $provider = $args->{my_provider};
+
+    registercloudguest($instance) if is_byos();
+
+    # Create AI stack
+    create_aistack($instance, $ins_rke2, $ins_kubectl, $ins_helm);
+
+    # Validate OpenWebUI
+    #validate_openwebui($admin_username, $admin_password, $openwebui_hostname);
+    # Verify Ollama api endpoint
+    #verify_api_endpoint();
+    # Milvus DB Test
+    #milvus_db_test();
+    # Milvus DB Test
+    #milvus_db_test();
+    # OpenWebUI Integration test
+    #test_openwebui_interaction();
+}
+
+1;