kopia lustrzana https://github.com/bellingcat/auto-archiver
66 wiersze
2.4 KiB
Python
66 wiersze
2.4 KiB
Python
import pytest
|
|
|
|
from auto_archiver.modules.hash_enricher import HashEnricher
|
|
from auto_archiver.core import Metadata, Media
|
|
from auto_archiver.core.module import ModuleFactory
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"algorithm, filename, expected_hash",
|
|
[
|
|
("SHA-256", "tests/data/testfile_1.txt", "1b4f0e9851971998e732078544c96b36c3d01cedf7caa332359d6f1d83567014"),
|
|
("SHA-256", "tests/data/testfile_2.txt", "60303ae22b998861bce3b28f33eec1be758a213c86c93c076dbe9f558c11c752"),
|
|
(
|
|
"SHA3-512",
|
|
"tests/data/testfile_1.txt",
|
|
"d2d8cc4f369b340130bd2b29b8b54e918b7c260c3279176da9ccaa37c96eb71735fc97568e892dc6220bf4ae0d748edb46bd75622751556393be3f482e6f794e",
|
|
),
|
|
(
|
|
"SHA3-512",
|
|
"tests/data/testfile_2.txt",
|
|
"e35970edaa1e0d8af7d948491b2da0450a49fd9cc1e83c5db4c6f175f9550cf341f642f6be8cfb0bfa476e4258e5088c5ad549087bf02811132ac2fa22b734c6",
|
|
),
|
|
],
|
|
)
|
|
def test_calculate_hash(algorithm, filename, expected_hash, setup_module):
|
|
# test SHA-256
|
|
he = setup_module(HashEnricher, {"algorithm": algorithm, "chunksize": 100})
|
|
assert he.calculate_hash(filename) == expected_hash
|
|
|
|
|
|
def test_default_config_values(setup_module):
|
|
he = setup_module(HashEnricher)
|
|
assert he.algorithm == "SHA-256"
|
|
assert he.chunksize == 16000000
|
|
|
|
|
|
def test_config():
|
|
# test default config
|
|
c = ModuleFactory().get_module_lazy("hash_enricher").configs
|
|
assert c["algorithm"]["default"] == "SHA-256"
|
|
assert c["chunksize"]["default"] == 16000000
|
|
assert c["algorithm"]["choices"] == ["SHA-256", "SHA3-512"]
|
|
assert c["algorithm"]["help"] == "hash algorithm to use"
|
|
assert (
|
|
c["chunksize"]["help"]
|
|
== "number of bytes to use when reading files in chunks (if this value is too large you will run out of RAM), default is 16MB"
|
|
)
|
|
|
|
|
|
def test_hash_media(setup_module):
|
|
he = setup_module(HashEnricher, {"algorithm": "SHA-256", "chunksize": 1})
|
|
|
|
# generate metadata with two test files
|
|
m = Metadata().set_url("https://example.com")
|
|
|
|
# noop - the metadata has no media. Shouldn't fail
|
|
he.enrich(m)
|
|
|
|
m.add_media(Media("tests/data/testfile_1.txt"))
|
|
m.add_media(Media("tests/data/testfile_2.txt"))
|
|
|
|
he.enrich(m)
|
|
|
|
assert m.media[0].get("hash") == "SHA-256:1b4f0e9851971998e732078544c96b36c3d01cedf7caa332359d6f1d83567014"
|
|
assert m.media[1].get("hash") == "SHA-256:60303ae22b998861bce3b28f33eec1be758a213c86c93c076dbe9f558c11c752"
|