def test_01_local_get_bdf_list(self):
tools = LinuxToolsHelper()
# when we pass None for a filter, we should get every device
all_devs = tools.lspci_get_bdf_list(filter=None)
# when we specify "Non-Volatile" we should get a list of nvme devices
ssd_devs = tools.lspci_get_bdf_list(filter="Non-Volatile")
self.assertTrue(len(all_devs) > len(ssd_devs))
# when we don't set any filter, the default is to search for "Non-" which
# will also give us all SSDs (they should return the same list
default_devs = tools.lspci_get_bdf_list()
self.assertEqual(len(ssd_devs), len(default_devs))
print("test_01_local_get_bdf_list: local ssd devices")
self.print_bdf_list(ssd_devs)
def test_05_remote_get_bdf_list(self):
local_tools = LinuxToolsHelper()
local_devs = local_tools.lspci_get_bdf_list(filter=None)
# make sure we do NOT fail local login
valid_login = {
'server_ip': 'localhost',
'user_name': self.local_user,
'user_pwd': self.local_pwd
}
remote_tools = LinuxToolsHelper(valid_login)
# client must be an SSHClient object when connection is successful
self.assertTrue(isinstance(remote_tools.client, SSHClient))
# remote must be True when credentials are supplied
self.assertTrue(remote_tools.remote)
# grab the same list of devices as a local run, and compare
remote_devs = remote_tools.lspci_get_bdf_list(filter=None)
for pci_id in local_devs.keys():
self.assertTrue(pci_id in remote_devs.keys())
def test_10_negative_test_returns_none(self):
local_tools = LinuxToolsHelper()
self.assertIsNone(
local_tools.udevadm_get_path_by_bdf('i:cant:drive.55'))
self.assertIsNone(
local_tools.udevadm_get_path_by_name('/dev/nosuchdev'))
empty_dict = local_tools.lspci_get_bdf_list(
filter="no-such-dev_filter")
self.assertIsNone(empty_dict)