53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
|
import hashlib
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from lim.errors import LimError
|
||
|
|
from lim.image import verify
|
||
|
|
|
||
|
|
CONTENT = b"fake image content"
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def image(tmp_path):
|
||
|
|
path = tmp_path / "image.img"
|
||
|
|
path.write_bytes(CONTENT)
|
||
|
|
return path
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize("algorithm", ["md5", "sha1", "sha256", "sha512"])
|
||
|
|
def test_matching_checksum_passes(image, algorithm):
|
||
|
|
checksum = hashlib.new(algorithm, CONTENT).hexdigest()
|
||
|
|
verify.verify_checksum(image, checksum)
|
||
|
|
|
||
|
|
|
||
|
|
def test_uppercase_checksum_passes(image):
|
||
|
|
checksum = hashlib.sha256(CONTENT).hexdigest().upper()
|
||
|
|
verify.verify_checksum(image, checksum)
|
||
|
|
|
||
|
|
|
||
|
|
def test_wrong_checksum_raises(image):
|
||
|
|
checksum = hashlib.sha256(b"other content").hexdigest()
|
||
|
|
with pytest.raises(LimError):
|
||
|
|
verify.verify_checksum(image, checksum)
|
||
|
|
|
||
|
|
|
||
|
|
def test_unrecognized_digest_length_raises(image):
|
||
|
|
with pytest.raises(LimError):
|
||
|
|
verify.verify_checksum(image, "abc123")
|
||
|
|
|
||
|
|
|
||
|
|
def test_missing_checksum_is_skipped(image):
|
||
|
|
verify.verify_checksum(image, None) # must not raise
|
||
|
|
|
||
|
|
|
||
|
|
def test_resolve_checksum_takes_first_available(fake_runner):
|
||
|
|
fake_runner.success_by_fragment["image.img.sha1"] = False
|
||
|
|
fake_runner.success_by_fragment["image.img.sha512"] = True
|
||
|
|
fake_runner.outputs["-q -O -"] = "deadbeef image.img"
|
||
|
|
assert verify.resolve_checksum("https://example.org/image.img") == "deadbeef"
|
||
|
|
|
||
|
|
|
||
|
|
def test_resolve_checksum_returns_none_without_sources(fake_runner):
|
||
|
|
assert verify.resolve_checksum("https://example.org/image.img") is None
|