mirror of
https://github.com/simon987/sist2.git
synced 2025-04-16 08:56:45 +00:00
86 lines
2.4 KiB
Python
86 lines
2.4 KiB
Python
import unittest
|
|
import subprocess
|
|
import shutil
|
|
import json
|
|
import os
|
|
|
|
TEST_FILES = "third-party/libscan/libscan-test-files/test_files"
|
|
|
|
|
|
def copy_files(files):
|
|
base = os.path.basename(files)
|
|
new_path = os.path.join("/tmp/sist2_test/", base)
|
|
|
|
shutil.rmtree(new_path, ignore_errors=True)
|
|
shutil.copytree(files, new_path)
|
|
return new_path
|
|
|
|
|
|
def sist2(*args):
|
|
print("./sist2_debug " + " ".join(args))
|
|
|
|
args = list(args)
|
|
args.append("--fast-epub")
|
|
# args.append("--very-verbose")
|
|
|
|
try:
|
|
return subprocess.check_output(
|
|
args=["./sist2_debug", *args],
|
|
)
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
|
|
def sist2_index(files, *args):
|
|
path = copy_files(files)
|
|
|
|
shutil.rmtree("test_i", ignore_errors=True)
|
|
sist2("scan", path, "-o", "test_i", *args)
|
|
return iter(sist2_index_to_dict("test_i"))
|
|
|
|
|
|
def sist2_incremental_index(files, func=None, incremental_index=False, *args):
|
|
path = copy_files(files)
|
|
|
|
if func:
|
|
func(path)
|
|
|
|
shutil.rmtree("test_i_inc", ignore_errors=True)
|
|
sist2("scan", path, "-o", "test_i_inc", "--incremental", "test_i", *args)
|
|
return iter(sist2_index_to_dict("test_i_inc", incremental_index))
|
|
|
|
|
|
def sist2_index_to_dict(index, incremental_index=False):
|
|
args = ["--incremental-index"] if incremental_index else []
|
|
|
|
res = sist2("index", "--print", "--very-verbose", *args, index)
|
|
|
|
for line in res.splitlines():
|
|
if line:
|
|
yield json.loads(line)
|
|
|
|
|
|
class ScanTest(unittest.TestCase):
|
|
|
|
def test_incremental1(self):
|
|
def remove_files(path):
|
|
os.remove(os.path.join(path, "msdoc/test1.doc"))
|
|
os.remove(os.path.join(path, "msdoc/test2.doc"))
|
|
|
|
def add_files(path):
|
|
with open(os.path.join(path, "newfile1"), "w"):
|
|
pass
|
|
with open(os.path.join(path, "newfile2"), "w"):
|
|
pass
|
|
with open(os.path.join(path, "newfile3"), "w"):
|
|
pass
|
|
|
|
file_count = sum(1 for _ in sist2_index(TEST_FILES))
|
|
self.assertEqual(sum(1 for _ in sist2_incremental_index(TEST_FILES, remove_files)), file_count - 2)
|
|
self.assertEqual(sum(1 for _ in sist2_incremental_index(TEST_FILES, add_files, incremental_index=True)), 3)
|
|
self.assertEqual(sum(1 for _ in sist2_incremental_index(TEST_FILES, add_files)), file_count + 3)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|