"""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'