mirror of
https://github.com/msgpack/msgpack-python.git
synced 2025-10-19 20:03:16 +00:00
33e0e86f4e
* use isort * fallback: use BytesIO instead of StringIO. We had dropped Python 2 already.
72 lines
1.8 KiB
Python
72 lines
1.8 KiB
Python
"""Test Unpacker's read_array_header and read_map_header methods"""
|
|
|
|
from msgpack import OutOfData, Unpacker, packb
|
|
|
|
UnexpectedTypeException = ValueError
|
|
|
|
|
|
def test_read_array_header():
|
|
unpacker = Unpacker()
|
|
unpacker.feed(packb(["a", "b", "c"]))
|
|
assert unpacker.read_array_header() == 3
|
|
assert unpacker.unpack() == "a"
|
|
assert unpacker.unpack() == "b"
|
|
assert unpacker.unpack() == "c"
|
|
try:
|
|
unpacker.unpack()
|
|
assert 0, "should raise exception"
|
|
except OutOfData:
|
|
assert 1, "okay"
|
|
|
|
|
|
def test_read_map_header():
|
|
unpacker = Unpacker()
|
|
unpacker.feed(packb({"a": "A"}))
|
|
assert unpacker.read_map_header() == 1
|
|
assert unpacker.unpack() == "a"
|
|
assert unpacker.unpack() == "A"
|
|
try:
|
|
unpacker.unpack()
|
|
assert 0, "should raise exception"
|
|
except OutOfData:
|
|
assert 1, "okay"
|
|
|
|
|
|
def test_incorrect_type_array():
|
|
unpacker = Unpacker()
|
|
unpacker.feed(packb(1))
|
|
try:
|
|
unpacker.read_array_header()
|
|
assert 0, "should raise exception"
|
|
except UnexpectedTypeException:
|
|
assert 1, "okay"
|
|
|
|
|
|
def test_incorrect_type_map():
|
|
unpacker = Unpacker()
|
|
unpacker.feed(packb(1))
|
|
try:
|
|
unpacker.read_map_header()
|
|
assert 0, "should raise exception"
|
|
except UnexpectedTypeException:
|
|
assert 1, "okay"
|
|
|
|
|
|
def test_correct_type_nested_array():
|
|
unpacker = Unpacker()
|
|
unpacker.feed(packb({"a": ["b", "c", "d"]}))
|
|
try:
|
|
unpacker.read_array_header()
|
|
assert 0, "should raise exception"
|
|
except UnexpectedTypeException:
|
|
assert 1, "okay"
|
|
|
|
|
|
def test_incorrect_type_nested_map():
|
|
unpacker = Unpacker()
|
|
unpacker.feed(packb([{"a": "b"}]))
|
|
try:
|
|
unpacker.read_map_header()
|
|
assert 0, "should raise exception"
|
|
except UnexpectedTypeException:
|
|
assert 1, "okay"
|