diff options
| author | Alex Gaynor <alex.gaynor@gmail.com> | 2024-03-09 18:24:00 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-03-09 17:24:00 -0600 |
| commit | dd1d6059b385e0b7d47ee71ce89eebb7f6de1099 (patch) | |
| tree | 4299b6bd6c53cd06abd5cea0b4401a9eb29bba48 /tests | |
| parent | 9c3cec67406ac74ab05638b5170e1fc086683091 (diff) | |
| download | pyca-cryptography-dd1d6059b385e0b7d47ee71ce89eebb7f6de1099.zip pyca-cryptography-dd1d6059b385e0b7d47ee71ce89eebb7f6de1099.tar.gz pyca-cryptography-dd1d6059b385e0b7d47ee71ce89eebb7f6de1099.tar.bz2 | |
Additional type asserts for latest mypy (#10560)
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/hazmat/primitives/test_pkcs7.py | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/tests/hazmat/primitives/test_pkcs7.py b/tests/hazmat/primitives/test_pkcs7.py index 9a9eab3..36abfae 100644 --- a/tests/hazmat/primitives/test_pkcs7.py +++ b/tests/hazmat/primitives/test_pkcs7.py @@ -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( |