# coding: utf-8

from datetime import datetime

import pytest

from pandas.io.msgpack import packb, unpackb


class DummyException(Exception):
    pass


class TestExceptions(object):

    def test_raise_on_find_unsupported_value(self):
        msg = "can\'t serialize datetime"
        with pytest.raises(TypeError, match=msg):
            packb(datetime.now())

    def test_raise_from_object_hook(self):
        def hook(_):
            raise DummyException()

        with pytest.raises(DummyException):
            unpackb(packb({}), object_hook=hook)
        with pytest.raises(DummyException):
            unpackb(packb({'fizz': 'buzz'}), object_hook=hook)
        with pytest.raises(DummyException):
            unpackb(packb({'fizz': 'buzz'}), object_pairs_hook=hook)
        with pytest.raises(DummyException):
            unpackb(packb({'fizz': {'buzz': 'spam'}}), object_hook=hook)
        with pytest.raises(DummyException):
            unpackb(packb({'fizz': {'buzz': 'spam'}}), object_pairs_hook=hook)

    def test_invalid_value(self):
        msg = "Unpack failed: error"
        with pytest.raises(ValueError, match=msg):
            unpackb(b"\xd9\x97#DL_")