You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
63 lines
1.6 KiB
Python
63 lines
1.6 KiB
Python
import glob
|
|
import io
|
|
import os
|
|
|
|
import pytest
|
|
|
|
from .. import decode, protocol
|
|
from ... import util
|
|
|
|
|
|
def test_subpackets():
|
|
s = io.BytesIO(b'\x00\x05\x02\xAB\xCD\x01\xEF')
|
|
assert decode.parse_subpackets(util.Reader(s)) == [b'\xAB\xCD', b'\xEF']
|
|
|
|
|
|
def test_subpackets_prefix():
|
|
for n in [0, 1, 2, 4, 5, 10, 191, 192, 193,
|
|
255, 256, 257, 8383, 8384, 65530]:
|
|
item = b'?' * n # create dummy subpacket
|
|
prefixed = protocol.subpackets(item)
|
|
result = decode.parse_subpackets(util.Reader(io.BytesIO(prefixed)))
|
|
assert [item] == result
|
|
|
|
|
|
def test_mpi():
|
|
s = io.BytesIO(b'\x00\x09\x01\x23')
|
|
assert decode.parse_mpi(util.Reader(s)) == 0x123
|
|
|
|
s = io.BytesIO(b'\x00\x09\x01\x23\x00\x03\x05')
|
|
assert decode.parse_mpis(util.Reader(s), n=2) == [0x123, 5]
|
|
|
|
|
|
cwd = os.path.join(os.path.dirname(__file__))
|
|
input_files = glob.glob(os.path.join(cwd, '*.gpg'))
|
|
|
|
|
|
@pytest.fixture(params=input_files)
|
|
def public_key_path(request):
|
|
return request.param
|
|
|
|
|
|
def test_gpg_files(public_key_path): # pylint: disable=redefined-outer-name
|
|
with open(public_key_path, 'rb') as f:
|
|
assert list(decode.parse_packets(f))
|
|
|
|
|
|
def test_has_custom_subpacket():
|
|
sig = {'unhashed_subpackets': []}
|
|
assert not decode.has_custom_subpacket(sig)
|
|
|
|
custom_markers = [
|
|
protocol.CUSTOM_SUBPACKET,
|
|
protocol.subpacket(10, protocol.CUSTOM_KEY_LABEL),
|
|
]
|
|
for marker in custom_markers:
|
|
sig = {'unhashed_subpackets': [marker]}
|
|
assert decode.has_custom_subpacket(sig)
|
|
|
|
|
|
def test_load_by_keygrip_missing():
|
|
with pytest.raises(KeyError):
|
|
decode.load_by_keygrip(pubkey_bytes=b'', keygrip=b'')
|