Extend VARIANT_ID check to support both workstation and kde editions

Co-authored-by: superm1 <494526+superm1@users.noreply.github.com>
Signed-off-by: Mario Limonciello (AMD) <superm1@kernel.org>

Author: Copilot

src/amd_debug/installer.py

@@ -64,7 +64,7 @@ def install(self):
             for line in release.split("\n"):
                 if line.startswith("VARIANT_ID"):
                     variant = line.split("=")[-1]
-            if variant != "workstation":
+            if variant not in ("workstation", "kde"):
                 return False
             installer = ["dnf", "install", "-y", self.rpm]
         elif dist == "arch" or os.path.exists("/etc/arch-release"):

src/test_installer.py

@@ -125,6 +125,29 @@ def test_install_iasl_fedora(
         )
         self.assertTrue(ret)
 
+    @patch("builtins.print")
+    @patch("amd_debug.installer.get_distro", return_value="fedora")
+    @patch("builtins.open", new_callable=mock_open, read_data="VARIANT_ID=kde\n")
+    @patch("os.execvp", return_value=None)
+    @patch("subprocess.check_call", return_value=0)
+    @patch("subprocess.call", return_value=1)
+    def test_install_iasl_fedora_kde(
+        self,
+        _mock_call,
+        _mock_check_call,
+        _mock_variant,
+        _mock_distro,
+        _fake_sudo,
+        _mock_print,
+    ):
+        """Test install requirements function on Fedora KDE"""
+        self.installer.set_requirements("iasl")
+        ret = self.installer.install_dependencies()
+        _mock_check_call.assert_called_once_with(
+            ["dnf", "install", "-y", "acpica-tools"]
+        )
+        self.assertTrue(ret)
+
     @patch("builtins.print")
     @patch("amd_debug.installer.get_distro", return_value="ubuntu")
     @patch("os.execvp", return_value=None)
@@ -162,6 +185,27 @@ def test_install_ethtool_fedora(
         _mock_check_call.assert_called_once_with(["dnf", "install", "-y", "ethtool"])
         self.assertTrue(ret)
 
+    @patch("builtins.print")
+    @patch("amd_debug.installer.get_distro", return_value="fedora")
+    @patch("builtins.open", new_callable=mock_open, read_data="VARIANT_ID=kde\n")
+    @patch("os.execvp", return_value=None)
+    @patch("subprocess.check_call", return_value=0)
+    @patch("subprocess.call", return_value=1)
+    def test_install_ethtool_fedora_kde(
+        self,
+        _mock_call,
+        _mock_check_call,
+        _mock_variant,
+        _mock_distro,
+        _fake_sudo,
+        _mock_print,
+    ):
+        """Test install requirements function on Fedora KDE"""
+        self.installer.set_requirements("ethtool")
+        ret = self.installer.install_dependencies()
+        _mock_check_call.assert_called_once_with(["dnf", "install", "-y", "ethtool"])
+        self.assertTrue(ret)
+
     @patch("builtins.print")
     @patch("amd_debug.installer.get_distro", return_value="arch")
     @patch("os.execvp", return_value=None)
@@ -235,6 +279,29 @@ def test_install_edid_decode_fedora(
         )
         self.assertTrue(ret)
 
+    @patch("builtins.print")
+    @patch("amd_debug.installer.get_distro", return_value="fedora")
+    @patch("builtins.open", new_callable=mock_open, read_data="VARIANT_ID=kde\n")
+    @patch("os.execvp", return_value=None)
+    @patch("subprocess.check_call", return_value=0)
+    @patch("subprocess.call", return_value=1)
+    def test_install_edid_decode_fedora_kde(
+        self,
+        _mock_call,
+        _mock_check_call,
+        _mock_variant,
+        _mock_distro,
+        _fake_sudo,
+        _mock_print,
+    ):
+        """Test install requirements function for edid-decode on Fedora KDE"""
+        self.installer.set_requirements("edid-decode")
+        ret = self.installer.install_dependencies()
+        _mock_check_call.assert_called_once_with(
+            ["dnf", "install", "-y", "libdisplay-info-tools"]
+        )
+        self.assertTrue(ret)
+
     @patch("builtins.print")
     @patch("amd_debug.installer.get_distro", return_value="arch")
     @patch("os.execvp", return_value=None)