diff --git a/pzztest.py b/pzztest.py index fef01ca..975ebd3 100644 --- a/pzztest.py +++ b/pzztest.py @@ -5,7 +5,7 @@ import shutil from time import time -__version__ = "0.0.5" +__version__ = "0.0.6" __author__ = "rigodron, algoflash, GGLinnk" __license__ = "MIT" __status__ = "developpement" @@ -25,6 +25,12 @@ batchcompress_path = Path("batch_compress") batchdecompress_path = Path("batch_decompress") +def test_storage(): + total, used, free = shutil.disk_usage("/") + if free - 10**10 < 6 * sum(path.stat().st_size for path in afsdump_path.glob('*.pzz') if path.is_file()): + raise Exception("Error - Not enought free space on the disk to run tests.") + + # compare two files def compare_files(file1_path:Path, file2_path:Path): CLUSTER_LEN = 131072 @@ -47,10 +53,10 @@ def compare_folders(folder1: Path, folder2: Path): folder1_file_count = len(folder1_paths) print(f"compare \"{folder1}\" - \"{folder2}\" ({folder1_file_count} files)") if folder1_file_count == 0: - raise Exception(f"ERROR - EMPTY FOLDER: {folder1}") + raise Exception(f"Error - Empty folder: {folder1}") for file_path in folder1_paths: if not compare_files(file_path, (folder2 / file_path.name)): - raise Exception(f"ERROR - INVALID FILE: {folder2 / file_path.name}") + raise Exception(f"Error - Invalid file: {folder2 / file_path.name}") ################################################## @@ -82,6 +88,8 @@ def pzztool_bpzz(folder_path:Path, out_pzzfolder_path:Path): raise Exception("Error while batch pzz.") +TEST_COUNT = 5 + start = time() print("###############################################################################") print("# Checking tests folder -> tests take 3 hour 35 minutes") @@ -90,8 +98,10 @@ print("######################################################################### if unpack_path.is_dir() or repack_path.is_dir() or compress_path.is_dir() or pzzfolder_path.is_dir() or batchdecompress_path.is_dir() or batchcompress_path.is_dir(): raise Exception(f"Error - Please remove:\n-{unpack_path}\n-{repack_path}\n-{compress_path}\n-{pzzfolder_path}\n-{batchdecompress_path}\n-{batchcompress_path}") +test_storage() + print("###############################################################################") -print("# TEST 1/5") +print(f"# TEST 1/{TEST_COUNT}") print("# Comparing [original pzz]->unpacked->repacked->[repack_path]") print("###############################################################################") pzzfolder_path.mkdir() @@ -109,7 +119,7 @@ compare_folders(pzzfolder_path, repack_path) shutil.rmtree(repack_path) print("###############################################################################") -print("# TEST 2/5") +print(f"# TEST 2/{TEST_COUNT}") print("# Comparing unpack_path/[*.pzzp]->decompress->compress_path/[*.pzzp] PZZ part") print("###############################################################################") compress_path.mkdir() @@ -129,10 +139,10 @@ for pzzp_path in unpack_paths: file_path = compress_path / pzzp_path.parent.name / pzzp_path.name print(f"compare \"{pzzp_path}\" - \"{file_path}\"") if not compare_files(pzzp_path, file_path): - raise Exception(f"INVALID FILE: {file_path}") + raise Exception(f"Invalid file: {file_path}") print("###############################################################################") -print("# TEST 3/5") +print(f"# TEST 3/{TEST_COUNT}") print("# Comparing [compress/*]->batch-decompress->batch-compress->[batchcompress_path]") print("###############################################################################") batchdecompress_path.mkdir() @@ -159,7 +169,7 @@ shutil.rmtree(compress_path) shutil.rmtree(batchcompress_path) print("###############################################################################") -print("# TEST 4/5") +print(f"# TEST 4/{TEST_COUNT}") print("# Comparing [pzzfolder_path]->batch-unpzz->batch-pzz->[repacked_path]") print("###############################################################################") pzztool_bunpzz(pzzfolder_path, unpack_path) @@ -171,13 +181,13 @@ shutil.rmtree(unpack_path) shutil.rmtree(repack_path) print("###############################################################################") -print("# TEST 5/5") +print(f"# TEST 5/{TEST_COUNT}") print("# Comparing [pzzfolder_path]->batch-unpack->(decompress or compress)->\nbatch-pzz->[repacked_path]") print("###############################################################################") -# if pzz : U -> decomp / already tested because unpzz let it decompressed by default -# if pzz : U -> comp / has to be tested -# if pzz : C -> decomp / already tested because unpzz decompress by default -# if pzz : C -> comp / has to be tested +# if pzz: U -> decomp / already tested because unpzz let it decompressed by default +# if pzz: U -> comp / has to be tested +# if pzz: C -> decomp / already tested because unpzz decompress by default +# if pzz: C -> comp / has to be tested unpack_path.mkdir() repack_path.mkdir() @@ -205,5 +215,5 @@ shutil.rmtree(repack_path) end = time() print("###############################################################################") -print(f"# All tests are OK - elapsed time : {end - start}") +print(f"# All tests are OK - elapsed time: {end - start}") print("###############################################################################")