def test_get_all_files_by_extension(): filesystem = FileSystem() folder = helper.populate_folder(5) files = set() files.update(filesystem.get_all_files(folder)) length = len(files) assert length == 5, length files = set() files.update(filesystem.get_all_files(folder, 'jpg')) length = len(files) assert length == 3, length files = set() files.update(filesystem.get_all_files(folder, 'txt')) length = len(files) assert length == 2, length files = set() files.update(filesystem.get_all_files(folder, 'gif')) length = len(files) assert length == 0, length shutil.rmtree(folder)
def test_get_all_files_success(): filesystem = FileSystem() folder = helper.populate_folder(5) files = filesystem.get_all_files(folder) shutil.rmtree(folder) length = len(files) assert length == 5, length
def test_get_all_files_success(): filesystem = FileSystem() folder = helper.populate_folder(5) files = filesystem.get_all_files(folder) shutil.rmtree(folder) length = len(files) assert length == 5, length
def test_get_all_files_with_invalid_file(): filesystem = FileSystem() folder = helper.populate_folder(5, include_invalid=True) files = set() files.update(filesystem.get_all_files(folder)) shutil.rmtree(folder) length = len(files) assert length == 5, length
def test_get_all_files_for_loop(): filesystem = FileSystem() folder = helper.populate_folder(5) files = set() files.update() counter = 0 for file in filesystem.get_all_files(folder): counter += 1 shutil.rmtree(folder) assert counter == 5, counter
def test_get_all_files_by_extension(): filesystem = FileSystem() folder = helper.populate_folder(5) files = filesystem.get_all_files(folder) length = len(files) assert length == 5, length files = filesystem.get_all_files(folder, 'jpg') length = len(files) assert length == 3, length files = filesystem.get_all_files(folder, 'txt') length = len(files) assert length == 2, length files = filesystem.get_all_files(folder, 'gif') length = len(files) assert length == 0, length shutil.rmtree(folder)
def main(argv): filesystem = FileSystem() result = Result() subclasses = get_all_subclasses() paths = argv[1:] for path in paths: path = os.path.expanduser(path) if os.path.isdir(path): for source in filesystem.get_all_files(path, None): status = add_original_name(source, subclasses) result.append((_decode(source), status)) else: status = add_original_name(path, subclasses) result.append((_decode(path), status)) result.write()