craftin/PGPy-3_13_patch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Parse partial length headers correctly.

Browse Source
This commit is contained in:
J08nY
2017-08-04 17:14:17 +02:00
parent e183c68fbf
commit 00d9217f19
2 changed files with 33 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/test_02_packets.py
View File

@@ -40,16 +40,12 @@ def binload(f):
return buf
skip_files = {'tests/testdata/packets/{:s}'.format(pkt) for pkt in ['11.partial.literal']}
pktfiles = sorted(glob.glob('tests/testdata/packets/[0-9]*'))
class TestPacket(object):
@pytest.mark.parametrize('packet', pktfiles, ids=[os.path.basename(f) for f in pktfiles])
def test_load(self, packet):
if packet in skip_files:
pytest.skip("not implemented yet")
b = binload(packet) + _trailer
_b = b[:]
p = Packet(_b)
@@ -59,11 +55,12 @@ class TestPacket(object):
# length is computed correctly
assert p.header.length + len(p.header) == len(p)
assert len(p) == len(b) - len(_trailer)
assert len(p.__bytes__()) == len(b) - len(_trailer)
if packet not in ('tests/testdata/packets/11.partial.literal',):
assert len(p) == len(b) - len(_trailer)
assert len(p.__bytes__()) == len(b) - len(_trailer)
# __bytes__ output is correct
assert p.__bytes__() == b[:-len(_trailer)]
# __bytes__ output is correct
assert p.__bytes__() == b[:-len(_trailer)]
# instantiated class is what we expected
if hasattr(p.header, 'version') and (p.header.tag, p.header.version) in _pclasses: