def setUp(self):
self.mock_os_listdir_patcher = patch('os.listdir')
self.mock_os_listdir = self.mock_os_listdir_patcher.start()
self.mock_os_listdir.side_effect = mock_listdir
self.mock_path_isdir_patcher = patch('os.path.isdir')
self.mock_path_isdir = self.mock_path_isdir_patcher.start()
self.mock_path_isdir.side_effect = mock_isdir
self.mock_path_join_patcher = patch('os.path.join')
self.mock_path_join = self.mock_path_join_patcher.start()
self.mock_path_join.side_effect = mock_join
self.test_model = USBDeviceManager()
class TestUSBDeviceManager(unittest.TestCase):
def setUp(self):
self.mock_os_listdir_patcher = patch('os.listdir')
self.mock_os_listdir = self.mock_os_listdir_patcher.start()
self.mock_os_listdir.side_effect = mock_listdir
self.mock_path_isdir_patcher = patch('os.path.isdir')
self.mock_path_isdir = self.mock_path_isdir_patcher.start()
self.mock_path_isdir.side_effect = mock_isdir
self.mock_path_join_patcher = patch('os.path.join')
self.mock_path_join = self.mock_path_join_patcher.start()
self.mock_path_join.side_effect = mock_join
self.test_model = USBDeviceManager()
def tearDown(self):
self.mock_os_listdir_patcher.stop()
self.mock_path_isdir_patcher.stop()
self.mock_path_join_patcher.stop()
def test_get_devices_should_check_for_existence(self):
self.mock_path_isdir.side_effect = lambda *args: False
results = self.test_model.get_devices_to_index(['/Volumes'], [])
self.assertEqual(results, [])
def test_get_devices_to_index_should_full_paths(self):
results = self.test_model.get_devices_to_index(['/Volumes'], [])
self.assertEqual(results, [('/Volumes/dev1', 'dev1')])
def test_get_devices_to_index_should_work_with_multiple_mount_points(self):
results = self.test_model.get_devices_to_index(['/Volumes', '/media'], [])
self.assertEqual(results, [('/Volumes/dev1', 'dev1'), ('/media/dev2', 'dev2')])
def test_get_devices_to_index_should_respect_devices_to_ignore(self):
results = self.test_model.get_devices_to_index(['/Volumes', '/media'], ['dev2'])
self.assertEqual(results, [('/Volumes/dev1', 'dev1')])
