mirror of
https://github.com/msgpack/msgpack-python.git
synced 2026-06-19 06:02:09 +00:00
cd8137093a
This PR makes it so that `self._strict_map_key` is enforced even when using `_object_pairs_hook` which didn't use to be the case.
105 lines
2.9 KiB
Python
105 lines
2.9 KiB
Python
#!/usr/bin/env python
|
|
|
|
import datetime
|
|
|
|
from pytest import raises
|
|
|
|
from msgpack import ExtType, FormatError, OutOfData, StackError, Unpacker, packb, unpackb
|
|
|
|
|
|
class DummyException(Exception):
|
|
pass
|
|
|
|
|
|
def test_raise_on_find_unsupported_value():
|
|
with raises(TypeError):
|
|
packb(datetime.datetime.now())
|
|
|
|
|
|
def test_raise_from_object_hook():
|
|
def hook(obj):
|
|
raise DummyException
|
|
|
|
raises(DummyException, unpackb, packb({}), object_hook=hook)
|
|
raises(DummyException, unpackb, packb({"fizz": "buzz"}), object_hook=hook)
|
|
raises(DummyException, unpackb, packb({"fizz": "buzz"}), object_pairs_hook=hook)
|
|
raises(DummyException, unpackb, packb({"fizz": {"buzz": "spam"}}), object_hook=hook)
|
|
raises(
|
|
DummyException,
|
|
unpackb,
|
|
packb({"fizz": {"buzz": "spam"}}),
|
|
object_pairs_hook=hook,
|
|
)
|
|
|
|
|
|
def test_raise_from_list_hook():
|
|
def hook(lst: list) -> list:
|
|
raise DummyException
|
|
|
|
with raises(DummyException):
|
|
unpackb(packb([1, 2, 3]), list_hook=hook)
|
|
|
|
with raises(DummyException):
|
|
unpacker = Unpacker(list_hook=hook)
|
|
unpacker.feed(packb([1, 2, 3]))
|
|
unpacker.unpack()
|
|
|
|
|
|
def test_raise_from_ext_hook():
|
|
def hook(code: int, data: bytes) -> ExtType:
|
|
raise DummyException
|
|
|
|
packed = packb(ExtType(42, b"hello"))
|
|
|
|
with raises(DummyException):
|
|
unpackb(packed, ext_hook=hook)
|
|
|
|
with raises(DummyException):
|
|
unpacker = Unpacker(ext_hook=hook)
|
|
unpacker.feed(packed)
|
|
unpacker.unpack()
|
|
|
|
|
|
def test_invalidvalue():
|
|
incomplete = b"\xd9\x97#DL_" # raw8 - length=0x97
|
|
with raises(ValueError):
|
|
unpackb(incomplete)
|
|
|
|
with raises(OutOfData):
|
|
unpacker = Unpacker()
|
|
unpacker.feed(incomplete)
|
|
unpacker.unpack()
|
|
|
|
with raises(FormatError):
|
|
unpackb(b"\xc1") # (undefined tag)
|
|
|
|
with raises(FormatError):
|
|
unpackb(b"\x91\xc1") # fixarray(len=1) [ (undefined tag) ]
|
|
|
|
with raises(StackError):
|
|
unpackb(b"\x91" * 3000) # nested fixarray(len=1)
|
|
|
|
|
|
def test_strict_map_key():
|
|
valid = {"unicode": 1, b"bytes": 2}
|
|
packed = packb(valid, use_bin_type=True)
|
|
assert valid == unpackb(packed, raw=False, strict_map_key=True)
|
|
|
|
invalid = {42: 1}
|
|
packed = packb(invalid, use_bin_type=True)
|
|
with raises(ValueError):
|
|
unpackb(packed, raw=False, strict_map_key=True)
|
|
|
|
|
|
def test_strict_map_key_with_object_pairs_hook():
|
|
# strict_map_key should be enforced even when object_pairs_hook is set
|
|
invalid = {42: "value"}
|
|
packed = packb(invalid, use_bin_type=True)
|
|
with raises(ValueError):
|
|
unpackb(packed, raw=False, strict_map_key=True, object_pairs_hook=list)
|
|
|
|
# valid keys (str/bytes) should still work with object_pairs_hook
|
|
valid = {"key": "value"}
|
|
packed = packb(valid, use_bin_type=True)
|
|
result = unpackb(packed, raw=False, strict_map_key=True, object_pairs_hook=list)
|
|
assert result == [("key", "value")]
|