Fix tests for Python 2
This commit is contained in:
parent
19e05151e7
commit
0299fdc50e
|
@ -428,39 +428,38 @@ def test_embedded_files():
|
|||
pdf = PdfReader(fdata=pdf_bytes)
|
||||
embedded = pdf.Root.Names.EmbeddedFiles.Names
|
||||
|
||||
assert zlib.decompress(
|
||||
embedded[1].EF.F.stream.encode('latin-1')) == b'hi there'
|
||||
def zlib_decompress(stream):
|
||||
if not isinstance(embedded[1].EF.F.stream, bytes):
|
||||
stream = stream.encode('latin-1')
|
||||
return zlib.decompress(stream)
|
||||
|
||||
assert zlib_decompress(embedded[1].EF.F.stream) == b'hi there'
|
||||
assert embedded[1].EF.F.Params.CheckSum == (
|
||||
'<{}>'.format(hashlib.md5(b'hi there').hexdigest()))
|
||||
assert embedded[1].F.decode() == ''
|
||||
assert embedded[1].UF.decode() == 'attachment.bin'
|
||||
assert embedded[1].Desc.decode() == 'some file attachment äöü'
|
||||
|
||||
assert zlib.decompress(
|
||||
embedded[3].EF.F.stream.encode('latin-1')) == b'12345678'
|
||||
assert zlib_decompress(embedded[3].EF.F.stream) == b'12345678'
|
||||
assert embedded[3].EF.F.Params.CheckSum == (
|
||||
'<{}>'.format(hashlib.md5(adata).hexdigest()))
|
||||
assert embedded[3].UF.decode() == os.path.basename(absolute_tmp_file)
|
||||
|
||||
assert zlib.decompress(
|
||||
embedded[5].EF.F.stream.encode('latin-1')) == b'abcdefgh'
|
||||
assert zlib_decompress(embedded[5].EF.F.stream) == b'abcdefgh'
|
||||
assert embedded[5].EF.F.Params.CheckSum == (
|
||||
'<{}>'.format(hashlib.md5(rdata).hexdigest()))
|
||||
assert embedded[5].UF.decode() == os.path.basename(relative_tmp_file)
|
||||
|
||||
assert zlib.decompress(
|
||||
embedded[7].EF.F.stream.encode('latin-1')) == b'oob attachment'
|
||||
assert zlib_decompress(embedded[7].EF.F.stream) == b'oob attachment'
|
||||
assert embedded[7].EF.F.Params.CheckSum == (
|
||||
'<{}>'.format(hashlib.md5(b'oob attachment').hexdigest()))
|
||||
assert embedded[7].Desc.decode() == 'Hello'
|
||||
|
||||
assert zlib.decompress(
|
||||
embedded[9].EF.F.stream.encode('latin-1')) == b'raw URL'
|
||||
assert zlib_decompress(embedded[9].EF.F.stream) == b'raw URL'
|
||||
assert embedded[9].EF.F.Params.CheckSum == (
|
||||
'<{}>'.format(hashlib.md5(b'raw URL').hexdigest()))
|
||||
|
||||
assert zlib.decompress(
|
||||
embedded[11].EF.F.stream.encode('latin-1')) == b'file like obj'
|
||||
assert zlib_decompress(embedded[11].EF.F.stream) == b'file like obj'
|
||||
assert embedded[11].EF.F.Params.CheckSum == (
|
||||
'<{}>'.format(hashlib.md5(b'file like obj').hexdigest()))
|
||||
|
||||
|
|
Loading…
Reference in New Issue