Virtual-World-RE/NeoGF
4
0
Fork 0
mirror of https://github.com/Virtual-World-RE/NeoGF.git synced 2024-11-15 12:45:34 +01:00
Code Issues Releases Wiki Activity

Update afstest.py

Browse Source 
added stats tests on folders & AFS
This commit is contained in:
tmpz23 2022-01-29 06:20:36 +01:00 committed by GitHub
parent 835a0a4a2d
commit d09a3f8810
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 19 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
afstest.py
View File

@ -1,12 +1,11 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
from afstool import AfsInvalidFileLenError, Afs, FilenameResolver from afstool import AfsInvalidFileLenError, Afs, FilenameResolver
from math import floor, ceil
import os import os
from pathlib import Path from pathlib import Path
import shutil import shutil
from time import time from time import time
__version__ = "0.0.3" __version__ = "0.0.4"
__author__ = "rigodron, algoflash, GGLinnk" __author__ = "rigodron, algoflash, GGLinnk"
__license__ = "MIT" __license__ = "MIT"
__status__ = "developpement" __status__ = "developpement"
@ -94,7 +93,6 @@ def compare_folders(folder1: Path, folder2: Path, compare_mtime=False):
for path1 in folder1_tmp_paths: for path1 in folder1_tmp_paths:
path2 = folder2 / Path(*path1.parts[len1:]) path2 = folder2 / Path(*path1.parts[len1:])
if compare_mtime: if compare_mtime:
# When using FD Date it floor to second
if round(path1.stat().st_mtime) != round(path2.stat().st_mtime): if round(path1.stat().st_mtime) != round(path2.stat().st_mtime):
raise Exception(f"\"{path1}\" and \"{path2}\" mtime (update time) are different:\n {round(path1.stat().st_mtime)}-{round(path2.stat().st_mtime)}") raise Exception(f"\"{path1}\" and \"{path2}\" mtime (update time) are different:\n {round(path1.stat().st_mtime)}-{round(path2.stat().st_mtime)}")
if not compare_files(path1, path2): if not compare_files(path1, path2):
@ -165,6 +163,9 @@ def afstool_pack(folder_path:Path, afs_path:Path):
def afstool_unpack(afs_path:Path, folder_path:Path): def afstool_unpack(afs_path:Path, folder_path:Path):
if os.system(f"python afstool.py -u \"{afs_path}\" \"{folder_path}\"") != 0: if os.system(f"python afstool.py -u \"{afs_path}\" \"{folder_path}\"") != 0:
raise Exception("Error while unpacking.") raise Exception("Error while unpacking.")
def afstool_stats(path:Path):
if os.system(f"python afstool.py -s \"{path}\" > NUL") != 0:
raise Exception("Error while getting stats.")
def repack_unpack2_compare(): def repack_unpack2_compare():
@ -189,7 +190,7 @@ def repack_unpack2_compare():
shutil.rmtree(unpack2_path) shutil.rmtree(unpack2_path)
TEST_COUNT = 6 TEST_COUNT = 7
start = time() start = time()
print("###############################################################################") print("###############################################################################")
@ -203,7 +204,6 @@ print("#########################################################################
print(f"# TEST 1/{TEST_COUNT}") print(f"# TEST 1/{TEST_COUNT}")
print("# Comparing afss_path->unpack->[unpack_path] AFS with [afspacker_unpack_path].") print("# Comparing afss_path->unpack->[unpack_path] AFS with [afspacker_unpack_path].")
print("###############################################################################") print("###############################################################################")
unpack_path.mkdir() unpack_path.mkdir()
afss_paths = list(afss_path.glob("*")) afss_paths = list(afss_path.glob("*"))
@ -224,6 +224,16 @@ for folder_path in unpack_path.glob("*"):
print("###############################################################################") print("###############################################################################")
print(f"# TEST 2/{TEST_COUNT}") print(f"# TEST 2/{TEST_COUNT}")
print("# Testing --stats command with all AFS and all unpacked AFS.")
print("###############################################################################")
for afs_path in afss_path.glob("*"):
afstool_stats(afs_path)
for folder_path in unpack_path.glob("*"):
afstool_stats(folder_path)
print("###############################################################################")
print(f"# TEST 3/{TEST_COUNT}")
print("# Comparing unpack_path->pack->[repack_path] AFS with [afss_path].") print("# Comparing unpack_path->pack->[repack_path] AFS with [afss_path].")
print("###############################################################################") print("###############################################################################")
repack_path.mkdir() repack_path.mkdir()
@ -237,7 +247,7 @@ compare_folders(afss_path, repack_path)
shutil.rmtree(repack_path) shutil.rmtree(repack_path)
print("###############################################################################") print("###############################################################################")
print(f"# TEST 3/{TEST_COUNT}") print(f"# TEST 4/{TEST_COUNT}")
print("# Comparing [unpack_path]->patch->pack->unpack->[unpack2_path].") print("# Comparing [unpack_path]->patch->pack->unpack->[unpack2_path].")
print("###############################################################################") print("###############################################################################")
# Patch unpack files whithout changing their len # Patch unpack files whithout changing their len
@ -249,7 +259,7 @@ for folder_path in unpack_path.glob("*"):
repack_unpack2_compare() repack_unpack2_compare()
print("###############################################################################") print("###############################################################################")
print(f"# TEST 4/{TEST_COUNT}") print(f"# TEST 5/{TEST_COUNT}")
print("# Comparing [unpack_path]->patch(max_size)->pack->unpack->[unpack2_path].") print("# Comparing [unpack_path]->patch(max_size)->pack->unpack->[unpack2_path].")
print("###############################################################################") print("###############################################################################")
# Patch unpack files changing len to max # Patch unpack files changing len to max
@ -258,7 +268,7 @@ patch_unpackedfiles_in_folder(unpack_path)
repack_unpack2_compare() repack_unpack2_compare()
print("###############################################################################") print("###############################################################################")
print(f"# TEST 5/{TEST_COUNT}") print(f"# TEST 6/{TEST_COUNT}")
print("# Testing exception unpack_path->patch(max_size+1)->[pack]->repack_path.") print("# Testing exception unpack_path->patch(max_size+1)->[pack]->repack_path.")
print("###############################################################################") print("###############################################################################")
# Patch unpack files with 1 byte in a new used block in the first file # Patch unpack files with 1 byte in a new used block in the first file
@ -278,7 +288,7 @@ for folder_path in unpack_path.glob("*"):
shutil.rmtree(repack_path) shutil.rmtree(repack_path)
print("###############################################################################") print("###############################################################################")
print(f"# TEST 6/{TEST_COUNT}") print(f"# TEST 7/{TEST_COUNT}")
print("# Comparing [unpack_path]->patch(blocks - 1)->pack->unpack->[unpack2_path].") print("# Comparing [unpack_path]->patch(blocks - 1)->pack->unpack->[unpack2_path].")
print("###############################################################################") print("###############################################################################")
# Patch unpack files with 1 block less # Patch unpack files with 1 block less