diff --git a/mfd_package_manager/linux.py b/mfd_package_manager/linux.py index a1f0609..18c6823 100644 --- a/mfd_package_manager/linux.py +++ b/mfd_package_manager/linux.py @@ -158,7 +158,7 @@ def get_driver_info(self, driver_name: str) -> "DriverInfo": name_regex = r"^\s*name:\s+(?P.*)" version_match = re.search(version_regex, result.stdout, flags=re.MULTILINE) name_match = re.search(name_regex, result.stdout, flags=re.MULTILINE) - name = name_match.group("name") if name_match is not None else "N/A" + name = name_match.group("name") if name_match is not None else driver_name version = version_match.group("version") if version_match is not None else "N/A" return DriverInfo(driver_version=version, driver_name=name) diff --git a/tests/unit/test_mfd_package_manager/test_linux.py b/tests/unit/test_mfd_package_manager/test_linux.py index 6b32391..486fd3a 100644 --- a/tests/unit/test_mfd_package_manager/test_linux.py +++ b/tests/unit/test_mfd_package_manager/test_linux.py @@ -163,7 +163,7 @@ def test_get_driver_info_no_info(self, manager): manager._connection.execute_command.return_value = ConnectionCompletedProcess( args="", stdout=output, return_code=0 ) - assert manager.get_driver_info("i40e") == DriverInfo(driver_name="N/A", driver_version="N/A") + assert manager.get_driver_info("i40e") == DriverInfo(driver_name="i40e", driver_version="N/A") def test_insert_module(self, manager): manager.insert_module("i40e", "-a")