def test_with_files_flat_dir_walker(self, list_dir):
"""Tests calling HostBroker._dir_walker() with files in a directory"""
files = [str(uuid.uuid4()), str(uuid.uuid4())]
list_dir.return_value = files
file_list = [x for x in HostBroker._dir_walker(self.root_path, False)]
for i in range(2):
self.assertEqual(os.path.join(self.root_path, files[i]),
file_list[i])
self.assertEqual(len(file_list), 2)
def test_with_files_recursive_dir_walker(self, walk):
"""Tests calling HostBroker._dir_walker() with files throughout tree"""
files = []
dirs = {
self.root_path: str(uuid.uuid4()),
os.path.join(self.root_path, str(uuid.uuid4())): str(uuid.uuid4()),
os.path.join(self.root_path, str(uuid.uuid4())): str(uuid.uuid4())
}
files = [os.path.join(x, dirs[x]) for x in dirs]
walk.return_value = [(x, (), (dirs[x], )) for x in dirs]
file_list = [x for x in HostBroker._dir_walker(self.root_path, True)]
for i in range(3):
self.assertEqual(os.path.join(self.root_path, files[i]),
file_list[i])
self.assertEqual(len(file_list), 3)
def test_no_files_flat_walk(self, list_dir):
"""Tests calling HostBroker._dir_walker() with no files in directory"""
file_list = [x for x in HostBroker._dir_walker(self.root_path, False)]
self.assertEqual(len(file_list), 0)
