saymrwulf/cryptography
1
0
Fork 0
mirror of https://github.com/saymrwulf/cryptography.git synced 2026-05-14 20:37:55 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Additional type asserts for latest mypy (#10560)

Browse source
This commit is contained in:
Alex Gaynor 2024-03-09 18:24:00 -05:00 committed by GitHub
parent 9c3cec6740
commit dd1d6059b3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 10 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
tests/hazmat/primitives/test_pkcs7.py
View file

@ -316,11 +316,15 @@ class TestPKCS7Builder:
# Parse the message to get the signed data, which is the
# first payload in the message
message = email.parser.BytesParser().parsebytes(sig)
signed_data = message.get_payload()[0].get_payload().encode()
payload = message.get_payload()
assert isinstance(payload, list)
assert isinstance(payload[0], email.message.Message)
signed_data = payload[0].get_payload()
assert isinstance(signed_data, str)
_pkcs7_verify(
serialization.Encoding.SMIME,
sig,
signed_data,
signed_data.encode(),
[cert],
options,
backend,
@ -546,7 +550,10 @@ class TestPKCS7Builder:
# Parse the message to get the signed data, which is the
# first payload in the message
message = email.parser.BytesParser().parsebytes(sig_pem)
signed_data = message.get_payload()[0].as_bytes(
payload = message.get_payload()
assert isinstance(payload, list)
assert isinstance(payload[0], email.message.Message)
signed_data = payload[0].as_bytes(
policy=message.policy.clone(linesep="\r\n")
)
_pkcs7_verify(