msgpack-python/test/test_except.py at master · GuyTuval/msgpack-python · GitHub

History

63 lines (45 loc) · 1.68 KB

Raw

#!/usr/bin/env python

# coding: utf-8

from pytest import raises

from msgpack import packb, unpackb, Unpacker, FormatError, StackError, OutOfData

import datetime

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_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 = {u"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)

msgpack-python/test/test_except.py at master · GuyTuval/msgpack-python · GitHub

Expand file tree

Breadcrumbs

test_except.py

Latest commit

History

Breadcrumbs

test_except.py

File metadata and controls