From d09a3f881095afc1e061e06e1ba2b08208f01bce Mon Sep 17 00:00:00 2001
From: tmpz23 <28760271+tmpz23@users.noreply.github.com>
Date: Sat, 29 Jan 2022 06:20:36 +0100
Subject: [PATCH] Update afstest.py

added stats tests on folders & AFS
---
 afstest.py | 28 +++++++++++++++++++---------
 1 file changed, 19 insertions(+), 9 deletions(-)

diff --git a/afstest.py b/afstest.py
index a9ad364..10c8bbe 100644
--- a/afstest.py
+++ b/afstest.py
@@ -1,12 +1,11 @@
 #!/usr/bin/env python3
 from afstool import AfsInvalidFileLenError, Afs, FilenameResolver
-from math import floor, ceil
 import os
 from pathlib import Path
 import shutil
 from time import time
 
-__version__ = "0.0.3"
+__version__ = "0.0.4"
 __author__ = "rigodron, algoflash, GGLinnk"
 __license__ = "MIT"
 __status__ = "developpement"
@@ -94,7 +93,6 @@ def compare_folders(folder1: Path, folder2: Path, compare_mtime=False):
     for path1 in folder1_tmp_paths:
         path2 = folder2 / Path(*path1.parts[len1:])
         if compare_mtime:
-            # When using FD Date it floor to second
             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)}")
         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):
     if os.system(f"python afstool.py -u \"{afs_path}\" \"{folder_path}\"") != 0:
         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():
@@ -189,7 +190,7 @@ def repack_unpack2_compare():
     shutil.rmtree(unpack2_path)
 
 
-TEST_COUNT = 6
+TEST_COUNT = 7
 
 start = time()
 print("###############################################################################")
@@ -203,7 +204,6 @@ print("#########################################################################
 print(f"# TEST 1/{TEST_COUNT}")
 print("# Comparing afss_path->unpack->[unpack_path] AFS with [afspacker_unpack_path].")
 print("###############################################################################")
-
 unpack_path.mkdir()
 afss_paths = list(afss_path.glob("*"))
 
@@ -224,6 +224,16 @@ for folder_path in unpack_path.glob("*"):
 
 print("###############################################################################")
 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("###############################################################################")
 repack_path.mkdir()
@@ -237,7 +247,7 @@ compare_folders(afss_path, repack_path)
 shutil.rmtree(repack_path)
 
 print("###############################################################################")
-print(f"# TEST 3/{TEST_COUNT}")
+print(f"# TEST 4/{TEST_COUNT}")
 print("# Comparing [unpack_path]->patch->pack->unpack->[unpack2_path].")
 print("###############################################################################")
 # Patch unpack files whithout changing their len
@@ -249,7 +259,7 @@ for folder_path in unpack_path.glob("*"):
 repack_unpack2_compare()
 
 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("###############################################################################")
 # Patch unpack files changing len to max
@@ -258,7 +268,7 @@ patch_unpackedfiles_in_folder(unpack_path)
 repack_unpack2_compare()
 
 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("###############################################################################")
 # 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)
 
 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("###############################################################################")
 # Patch unpack files with 1 block less