amd-debug-tools 0.2.2__py3-none-any.whl → 0.2.12__py3-none-any.whl

test_display.py

@@ -40,7 +40,7 @@ class TestDisplay(unittest.TestCase):
         display = Display()
 
         # Verify the EDID paths are correctly set
-        expected_edid = {"card0": "/sys/devices/card0/edid"}
+        expected_edid = ["/sys/devices/card0/edid"]
         self.assertEqual(display.get_edid(), expected_edid)
         mock_context.assert_called_once()
 
@@ -54,7 +54,7 @@ class TestDisplay(unittest.TestCase):
         display = Display()
 
         # Verify the EDID dictionary is empty
-        self.assertEqual(display.get_edid(), {})
+        self.assertEqual(display.get_edid(), [])
 
     @patch("amd_debug.display.Context")
     def test_device_without_card(self, mock_context):
@@ -70,7 +70,7 @@ class TestDisplay(unittest.TestCase):
         display = Display()
 
         # Verify the EDID dictionary is empty
-        self.assertEqual(display.get_edid(), {})
+        self.assertEqual(display.get_edid(), [])
 
     @patch("amd_debug.display.Context")
     @patch("amd_debug.display.read_file")
@@ -94,7 +94,7 @@ class TestDisplay(unittest.TestCase):
         display = Display()
 
         # Verify the EDID dictionary is empty
-        self.assertEqual(display.get_edid(), {})
+        self.assertEqual(display.get_edid(), [])
 
     @patch("amd_debug.display.Context")
     @patch("amd_debug.display.read_file")
@@ -116,7 +116,7 @@ class TestDisplay(unittest.TestCase):
         display = Display()
 
         # Verify the EDID dictionary is empty
-        self.assertEqual(display.get_edid(), {})
+        self.assertEqual(display.get_edid(), [])
 
     @patch("amd_debug.display.Context")
     @patch("amd_debug.display.read_file")
@@ -140,4 +140,4 @@ class TestDisplay(unittest.TestCase):
         display = Display()
 
         # Verify the EDID dictionary is empty
-        self.assertEqual(display.get_edid(), {})
+        self.assertEqual(display.get_edid(), [])

test_installer.py

@@ -125,6 +125,29 @@ class TestInstaller(unittest.TestCase):
         )
         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 @@ class TestInstaller(unittest.TestCase):
         _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)
@@ -231,7 +275,30 @@ class TestInstaller(unittest.TestCase):
         self.installer.set_requirements("edid-decode")
         ret = self.installer.install_dependencies()
         _mock_check_call.assert_called_once_with(
-            ["dnf", "install", "-y", "libdisplay-info"]
+            ["dnf", "install", "-y", "libdisplay-info-tools"]
+        )
+        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)
 

test_kernel.py

@@ -26,7 +26,7 @@ class TestKernelLog(unittest.TestCase):
         kernel_cmdline = "quiet splash"
         expected_output = ""
         with patch(
-            "builtins.open", new_callable=mock_open, read_data=kernel_cmdline
+            "amd_debug.common.open", new_callable=mock_open, read_data=kernel_cmdline
         ) as _mock_file:
             result = get_kernel_command_line()
             self.assertEqual(result, expected_output)
@@ -35,7 +35,7 @@ class TestKernelLog(unittest.TestCase):
         kernel_cmdline = ""
         expected_output = ""
         with patch(
-            "builtins.open", new_callable=mock_open, read_data=kernel_cmdline
+            "amd_debug.common.open", new_callable=mock_open, read_data=kernel_cmdline
         ) as _mock_file:
             result = get_kernel_command_line()
             self.assertEqual(result, expected_output)
@@ -44,7 +44,7 @@ class TestKernelLog(unittest.TestCase):
         kernel_cmdline = "quiet splash --debug=1"
         expected_output = "--debug=1"
         with patch(
-            "builtins.open", new_callable=mock_open, read_data=kernel_cmdline
+            "amd_debug.common.open", new_callable=mock_open, read_data=kernel_cmdline
         ) as _mock_file:
             result = get_kernel_command_line()
             self.assertEqual(result, expected_output)
@@ -53,7 +53,7 @@ class TestKernelLog(unittest.TestCase):
         kernel_cmdline = "quiet splash initrd=foo modprobe.blacklist=foo"
         expected_output = "modprobe.blacklist=foo"
         with patch(
-            "builtins.open", new_callable=mock_open, read_data=kernel_cmdline
+            "amd_debug.common.open", new_callable=mock_open, read_data=kernel_cmdline
         ) as _mock_file:
             result = get_kernel_command_line()
             self.assertEqual(result, expected_output)
@@ -85,7 +85,7 @@ class TestKernelLog(unittest.TestCase):
 
         # test a real post code line
         line = 'ex_trace_args: "  POST CODE: %X  ACPI TIMER: %X  TIME: %d.%d ms\\n", b0003f33, 83528798, 0, 77, 0, 0'
-        expected_output = "POST CODE: B0003F33  ACPI TIMER: 83528798  TIME: 0.77 ms"
+        expected_output = "POST CODE: B0003F33  ACPI TIMER: 83528798  TIME: 0.119 ms"
         result = sscanf_bios_args(line)
         self.assertEqual(result, expected_output)
 
@@ -114,7 +114,8 @@ class TestDmesgLogger(unittest.TestCase):
     """Test Dmesg logger functions"""
 
     @classmethod
-    def setUpClass(cls):
+    @patch("subprocess.run")
+    def setUpClass(cls, _mock_run=None):
         logging.basicConfig(filename="/dev/null", level=logging.DEBUG)
 
     def test_dmesg_logger_initialization(self):

test_launcher.py

@@ -26,10 +26,11 @@ class TestLauncher(unittest.TestCase):
         """Test launching as unknown exe"""
 
         with patch("builtins.print") as mock_print:
-            amd_debug.launch_tool("unknown_exe.py")
+            result = amd_debug.launch_tool("unknown_exe.py")
             mock_print.assert_called_once_with(
                 "\033[91mUnknown exe: unknown_exe.py\033[0m"
             )
+            self.assertIsNotNone(result)
 
     def test_launcher_amd_s2idle(self):
         """Test launching amd_s2idle"""
@@ -51,3 +52,10 @@ class TestLauncher(unittest.TestCase):
         with patch("amd_debug.pstate.main") as mock_main:
             amd_debug.launch_tool("amd_pstate.py")
             mock_main.assert_called_once()
+
+    def test_launcher_amd_ttm(self):
+        """Test launching amd_ttm"""
+
+        with patch("amd_debug.ttm.main") as mock_main:
+            amd_debug.launch_tool("amd_ttm.py")
+            mock_main.assert_called_once()