mirror of
https://github.com/arsenetar/dupeguru.git
synced 2025-05-07 17:29:50 +00:00
Andrew Senetar
e16df489bd
- Improves performance significantly - Add xxhash to requirements.txt and sort requirements - Rename md5 based members to digest - Update all tests to use new member names and hashing methods NOTE: May consider supporting multiple hashing algorithms in the future.
110 lines
4.2 KiB
Python
110 lines
4.2 KiB
Python
# Created By: Virgil Dupras
|
|
# Created On: 2009-10-23
|
|
# Copyright 2015 Hardcoded Software (http://www.hardcoded.net)
|
|
#
|
|
# This software is licensed under the "GPLv3" License as described in the "LICENSE" file,
|
|
# which should be included with this package. The terms are also available at
|
|
# http://www.gnu.org/licenses/gpl-3.0.html
|
|
|
|
import xxhash
|
|
from os import urandom
|
|
|
|
from hscommon.path import Path
|
|
from hscommon.testutil import eq_
|
|
from core.tests.directories_test import create_fake_fs
|
|
|
|
from .. import fs
|
|
|
|
|
|
def create_fake_fs_with_random_data(rootpath):
|
|
rootpath = rootpath["fs"]
|
|
rootpath.mkdir()
|
|
rootpath["dir1"].mkdir()
|
|
rootpath["dir2"].mkdir()
|
|
rootpath["dir3"].mkdir()
|
|
fp = rootpath["file1.test"].open("wb")
|
|
data1 = urandom(200 * 1024) # 200KiB
|
|
data2 = urandom(1024 * 1024) # 1MiB
|
|
data3 = urandom(10 * 1024 * 1024) # 10MiB
|
|
fp.write(data1)
|
|
fp.close()
|
|
fp = rootpath["file2.test"].open("wb")
|
|
fp.write(data2)
|
|
fp.close()
|
|
fp = rootpath["file3.test"].open("wb")
|
|
fp.write(data3)
|
|
fp.close()
|
|
fp = rootpath["dir1"]["file1.test"].open("wb")
|
|
fp.write(data1)
|
|
fp.close()
|
|
fp = rootpath["dir2"]["file2.test"].open("wb")
|
|
fp.write(data2)
|
|
fp.close()
|
|
fp = rootpath["dir3"]["file3.test"].open("wb")
|
|
fp.write(data3)
|
|
fp.close()
|
|
return rootpath
|
|
|
|
|
|
def test_size_aggregates_subfiles(tmpdir):
|
|
p = create_fake_fs(Path(str(tmpdir)))
|
|
b = fs.Folder(p)
|
|
eq_(b.size, 12)
|
|
|
|
|
|
def test_digest_aggregate_subfiles_sorted(tmpdir):
|
|
# dir.allfiles can return child in any order. Thus, bundle.digest must aggregate
|
|
# all files' digests it contains, but it must make sure that it does so in the
|
|
# same order everytime.
|
|
p = create_fake_fs_with_random_data(Path(str(tmpdir)))
|
|
b = fs.Folder(p)
|
|
digest1 = fs.File(p["dir1"]["file1.test"]).digest
|
|
digest2 = fs.File(p["dir2"]["file2.test"]).digest
|
|
digest3 = fs.File(p["dir3"]["file3.test"]).digest
|
|
digest4 = fs.File(p["file1.test"]).digest
|
|
digest5 = fs.File(p["file2.test"]).digest
|
|
digest6 = fs.File(p["file3.test"]).digest
|
|
# The expected digest is the hash of digests for folders and the direct digest for files
|
|
folder_digest1 = xxhash.xxh128_digest(digest1)
|
|
folder_digest2 = xxhash.xxh128_digest(digest2)
|
|
folder_digest3 = xxhash.xxh128_digest(digest3)
|
|
digest = xxhash.xxh128_digest(folder_digest1 + folder_digest2 + folder_digest3 + digest4 + digest5 + digest6)
|
|
eq_(b.digest, digest)
|
|
|
|
|
|
def test_partial_digest_aggregate_subfile_sorted(tmpdir):
|
|
p = create_fake_fs_with_random_data(Path(str(tmpdir)))
|
|
b = fs.Folder(p)
|
|
digest1 = fs.File(p["dir1"]["file1.test"]).digest_partial
|
|
digest2 = fs.File(p["dir2"]["file2.test"]).digest_partial
|
|
digest3 = fs.File(p["dir3"]["file3.test"]).digest_partial
|
|
digest4 = fs.File(p["file1.test"]).digest_partial
|
|
digest5 = fs.File(p["file2.test"]).digest_partial
|
|
digest6 = fs.File(p["file3.test"]).digest_partial
|
|
# The expected digest is the hash of digests for folders and the direct digest for files
|
|
folder_digest1 = xxhash.xxh128_digest(digest1)
|
|
folder_digest2 = xxhash.xxh128_digest(digest2)
|
|
folder_digest3 = xxhash.xxh128_digest(digest3)
|
|
digest = xxhash.xxh128_digest(folder_digest1 + folder_digest2 + folder_digest3 + digest4 + digest5 + digest6)
|
|
eq_(b.digest_partial, digest)
|
|
|
|
digest1 = fs.File(p["dir1"]["file1.test"]).digest_samples
|
|
digest2 = fs.File(p["dir2"]["file2.test"]).digest_samples
|
|
digest3 = fs.File(p["dir3"]["file3.test"]).digest_samples
|
|
digest4 = fs.File(p["file1.test"]).digest_samples
|
|
digest5 = fs.File(p["file2.test"]).digest_samples
|
|
digest6 = fs.File(p["file3.test"]).digest_samples
|
|
# The expected digest is the digest of digests for folders and the direct digest for files
|
|
folder_digest1 = xxhash.xxh128_digest(digest1)
|
|
folder_digest2 = xxhash.xxh128_digest(digest2)
|
|
folder_digest3 = xxhash.xxh128_digest(digest3)
|
|
digest = xxhash.xxh128_digest(folder_digest1 + folder_digest2 + folder_digest3 + digest4 + digest5 + digest6)
|
|
eq_(b.digest_samples, digest)
|
|
|
|
|
|
def test_has_file_attrs(tmpdir):
|
|
# a Folder must behave like a file, so it must have mtime attributes
|
|
b = fs.Folder(Path(str(tmpdir)))
|
|
assert b.mtime > 0
|
|
eq_(b.extension, "")
|