mirror of
https://github.com/msgpack/msgpack-python.git
synced 2025-10-19 20:03:16 +00:00
c8d0751fe3
The following steps have been taken: 1. Black was updated to latest version. The code has been formatted with the new version. 2. The pyupgrade utility is installed. This helped to remove all the code that was needed to support Python < 3.7. Fix #541. Co-authored-by: Inada Naoki <songofacandy@gmail.com>
165 lines
3.7 KiB
Python
165 lines
3.7 KiB
Python
#!/usr/bin/env python
|
|
import pytest
|
|
|
|
from msgpack import (
|
|
packb,
|
|
unpackb,
|
|
Packer,
|
|
Unpacker,
|
|
ExtType,
|
|
PackOverflowError,
|
|
PackValueError,
|
|
UnpackValueError,
|
|
)
|
|
|
|
|
|
def test_integer():
|
|
x = -(2**63)
|
|
assert unpackb(packb(x)) == x
|
|
with pytest.raises(PackOverflowError):
|
|
packb(x - 1)
|
|
|
|
x = 2**64 - 1
|
|
assert unpackb(packb(x)) == x
|
|
with pytest.raises(PackOverflowError):
|
|
packb(x + 1)
|
|
|
|
|
|
def test_array_header():
|
|
packer = Packer()
|
|
packer.pack_array_header(2**32 - 1)
|
|
with pytest.raises(PackValueError):
|
|
packer.pack_array_header(2**32)
|
|
|
|
|
|
def test_map_header():
|
|
packer = Packer()
|
|
packer.pack_map_header(2**32 - 1)
|
|
with pytest.raises(PackValueError):
|
|
packer.pack_array_header(2**32)
|
|
|
|
|
|
def test_max_str_len():
|
|
d = "x" * 3
|
|
packed = packb(d)
|
|
|
|
unpacker = Unpacker(max_str_len=3, raw=False)
|
|
unpacker.feed(packed)
|
|
assert unpacker.unpack() == d
|
|
|
|
unpacker = Unpacker(max_str_len=2, raw=False)
|
|
with pytest.raises(UnpackValueError):
|
|
unpacker.feed(packed)
|
|
unpacker.unpack()
|
|
|
|
|
|
def test_max_bin_len():
|
|
d = b"x" * 3
|
|
packed = packb(d, use_bin_type=True)
|
|
|
|
unpacker = Unpacker(max_bin_len=3)
|
|
unpacker.feed(packed)
|
|
assert unpacker.unpack() == d
|
|
|
|
unpacker = Unpacker(max_bin_len=2)
|
|
with pytest.raises(UnpackValueError):
|
|
unpacker.feed(packed)
|
|
unpacker.unpack()
|
|
|
|
|
|
def test_max_array_len():
|
|
d = [1, 2, 3]
|
|
packed = packb(d)
|
|
|
|
unpacker = Unpacker(max_array_len=3)
|
|
unpacker.feed(packed)
|
|
assert unpacker.unpack() == d
|
|
|
|
unpacker = Unpacker(max_array_len=2)
|
|
with pytest.raises(UnpackValueError):
|
|
unpacker.feed(packed)
|
|
unpacker.unpack()
|
|
|
|
|
|
def test_max_map_len():
|
|
d = {1: 2, 3: 4, 5: 6}
|
|
packed = packb(d)
|
|
|
|
unpacker = Unpacker(max_map_len=3, strict_map_key=False)
|
|
unpacker.feed(packed)
|
|
assert unpacker.unpack() == d
|
|
|
|
unpacker = Unpacker(max_map_len=2, strict_map_key=False)
|
|
with pytest.raises(UnpackValueError):
|
|
unpacker.feed(packed)
|
|
unpacker.unpack()
|
|
|
|
|
|
def test_max_ext_len():
|
|
d = ExtType(42, b"abc")
|
|
packed = packb(d)
|
|
|
|
unpacker = Unpacker(max_ext_len=3)
|
|
unpacker.feed(packed)
|
|
assert unpacker.unpack() == d
|
|
|
|
unpacker = Unpacker(max_ext_len=2)
|
|
with pytest.raises(UnpackValueError):
|
|
unpacker.feed(packed)
|
|
unpacker.unpack()
|
|
|
|
|
|
# PyPy fails following tests because of constant folding?
|
|
# https://bugs.pypy.org/issue1721
|
|
# @pytest.mark.skipif(True, reason="Requires very large memory.")
|
|
# def test_binary():
|
|
# x = b'x' * (2**32 - 1)
|
|
# assert unpackb(packb(x)) == x
|
|
# del x
|
|
# x = b'x' * (2**32)
|
|
# with pytest.raises(ValueError):
|
|
# packb(x)
|
|
#
|
|
#
|
|
# @pytest.mark.skipif(True, reason="Requires very large memory.")
|
|
# def test_string():
|
|
# x = 'x' * (2**32 - 1)
|
|
# assert unpackb(packb(x)) == x
|
|
# x += 'y'
|
|
# with pytest.raises(ValueError):
|
|
# packb(x)
|
|
#
|
|
#
|
|
# @pytest.mark.skipif(True, reason="Requires very large memory.")
|
|
# def test_array():
|
|
# x = [0] * (2**32 - 1)
|
|
# assert unpackb(packb(x)) == x
|
|
# x.append(0)
|
|
# with pytest.raises(ValueError):
|
|
# packb(x)
|
|
|
|
|
|
# auto max len
|
|
|
|
|
|
def test_auto_max_array_len():
|
|
packed = b"\xde\x00\x06zz"
|
|
with pytest.raises(UnpackValueError):
|
|
unpackb(packed, raw=False)
|
|
|
|
unpacker = Unpacker(max_buffer_size=5, raw=False)
|
|
unpacker.feed(packed)
|
|
with pytest.raises(UnpackValueError):
|
|
unpacker.unpack()
|
|
|
|
|
|
def test_auto_max_map_len():
|
|
# len(packed) == 6 -> max_map_len == 3
|
|
packed = b"\xde\x00\x04zzz"
|
|
with pytest.raises(UnpackValueError):
|
|
unpackb(packed, raw=False)
|
|
|
|
unpacker = Unpacker(max_buffer_size=6, raw=False)
|
|
unpacker.feed(packed)
|
|
with pytest.raises(UnpackValueError):
|
|
unpacker.unpack()
|