From 20025e476035a2b53201ad267c760bc02c7dcd70 Mon Sep 17 00:00:00 2001
From: Lubos Kocman <Lubos.Kocman@suse.com>
Date: Thu, 5 Sep 2024 17:26:12 +0200
Subject: [PATCH] KDE enablement for agama.py

---
 tests/installation/agama.pm | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/tests/installation/agama.pm b/tests/installation/agama.pm
index 88591aa25cc0..dab52d9cb54f 100644
--- a/tests/installation/agama.pm
+++ b/tests/installation/agama.pm
@@ -119,13 +119,19 @@ sub run {
     assert_and_click('agama-show-tabs');
 
     # default is just a minimal server style install
-    if (check_var('DESKTOP', 'gnome')) {
+    if (get_var('DESKTOP')) {
         assert_and_click('agama-software-tab');
 
-        wait_still_screen 5;
+        wait_still_screen(5);
         assert_and_click('agama-change-software-selection');
-        wait_still_screen 5;
-        assert_and_click('agama-software-selection-gnome-desktop-wayland');
+        wait_still_screen(5);
+
+        if (check_var('DESKTOP', 'gnome')) {
+            assert_and_click('agama-software-selection-gnome-desktop-wayland');
+        } elsif (check_var('DESKTOP', 'kde')) {
+            assert_and_click('agama-software-selection-kde-desktop-wayland');
+        }
+
         assert_and_click('agama-software-selection-close');
     }