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)