From 8381b2fb7bcc9c45cf3e9fff1f4c794eca22634c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Alexis=20M=C3=A9taireau?= Date: Tue, 25 Feb 2025 15:44:08 +0100 Subject: [PATCH] fixup! (WIP) Add tests --- tests/test_signatures.py | 21 +++++++++++++++++++-- 1 file changed, 19 insertions(+), 2 deletions(-) diff --git a/tests/test_signatures.py b/tests/test_signatures.py index a8c6b7a..b568347 100644 --- a/tests/test_signatures.py +++ b/tests/test_signatures.py @@ -298,8 +298,25 @@ def test_is_update_available_when_no_local_image(mocker): assert update_available is True assert digest == RANDOM_DIGEST -def test_verify_signature_wrong_payload_digest(): - pass + +def test_verify_signature(valid_signature): + """Test that verify_signature raises an error when the payload digest doesn't match.""" + verify_signature( + valid_signature, + Signature(valid_signature).manifest_digest, + TEST_PUBKEY_PATH, + ) + + +def test_verify_signature_tempered(tempered_signature): + """Test that verify_signature raises an error when the payload digest doesn't match.""" + # Call verify_signature and expect an error + with pytest.raises(errors.SignatureError): + verify_signature( + tempered_signature, + Signature(tempered_signature).manifest_digest, + TEST_PUBKEY_PATH, + ) def test_verify_signatures_empty_list():