mirror of
https://github.com/msgpack/msgpack-python.git
synced 2025-10-31 17:40:54 +00:00
66 lines
1.8 KiB
Python
66 lines
1.8 KiB
Python
"""Test Unpacker's read_array_header and read_map_header methods"""
|
|
from msgpack import packb, Unpacker
|
|
UnexpectedTypeException = ValueError
|
|
|
|
def test_read_array_header():
|
|
unpacker = Unpacker()
|
|
unpacker.feed(packb(['a', 'b', 'c']))
|
|
assert unpacker.read_array_header() == 3
|
|
assert unpacker.unpack() == b'a'
|
|
assert unpacker.unpack() == b'b'
|
|
assert unpacker.unpack() == b'c'
|
|
try:
|
|
unpacker.unpack()
|
|
assert 0, 'should raise exception'
|
|
except StopIteration:
|
|
assert 1, 'okay'
|
|
|
|
|
|
def test_read_map_header():
|
|
unpacker = Unpacker()
|
|
unpacker.feed(packb({'a': 'A'}))
|
|
assert unpacker.read_map_header() == 1
|
|
assert unpacker.unpack() == B'a'
|
|
assert unpacker.unpack() == B'A'
|
|
try:
|
|
unpacker.unpack()
|
|
assert 0, 'should raise exception'
|
|
except StopIteration:
|
|
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'
|
|
|