def test_it_can_ship_a_class():
cr = Cereal(serialize_naively=True)
class SomeObject():
pass
res = cr.loads(cr.dumps(SomeObject))
assert res.__name__ == 'EmulatedSomeObject'
def test_it_can_serialize_naively():
cr = Cereal(serialize_naively=True)
o = SomeObject(1, 'test', [5.5, True])
dump = cr.dumps(o)
res = cr.loads(dump)
assert o.a == res.a
assert o.wild == res.wild
assert o.attribute == res.attribute
assert o.__class__ != res.__class__
with pytest.raises(LimitedMethodError):
res.some_method()
def test_it_can_emulate_an_iterable():
cr = Cereal(serialize_naively=True)
class SomeObject():
def __iter__(self):
return iter([1, 2, 3, 4])
o = SomeObject()
res = cr.loads(cr.dumps(o))
assert res.__class__.__name__ == 'EmulatedSomeObject'
assert list(res) == [1, 2, 3, 4]
def test_it_doesnt_get_caught_in_recursion():
class A:
pass
class B:
pass
a = A()
b = B()
a.b = b
b.a = a
cr = Cereal(serialize_naively=True)
dump = cr.dumps(a)
res = cr.loads(dump)
assert res.b
def test_it_can_catch_exceptions_and_transport_them():
cr = Cereal(serialize_naively=True)
expected_msg = 'Something happened, yo'
class SomeNicheException(Exception):
pass
class SomeObject():
@property
def will_raise(self):
raise SomeNicheException(expected_msg)
o = SomeObject()
res = cr.loads(cr.dumps(o))
assert res.__class__.__name__ == 'EmulatedSomeObject'
with pytest.raises(Exception, match=expected_msg) as exc:
res.will_raise
assert exc.type.__name__ == 'SomeNicheException'
def test_it_can_ship_a_registered_class():
cr = Cereal(serialize_naively=True)
class SomeObject():
pass
cr.register_class('SomeObject', SomeObject, None, None)
res = cr.loads(cr.dumps(SomeObject))
assert res == SomeObject
def test_it_can_ignore_load_errors():
cr = Cereal(serialize_naively=True, raise_load_errors=False)
o = RaiseObject('1')
def to_builtin(v):
return v.a
def from_builtin(v):
return RaiseObject(v)
cr.register_class('RaiseObject', RaiseObject, to_builtin, from_builtin)
res = cr.loads(cr.dumps(o))
assert res == {INSTANCE_KEY: ['RaiseObject', 1]}
def test_it_can_throw_a_load_error():
cr = Cereal()
o = RaiseObject('1')
def to_builtin(v):
return v.a
def from_builtin(v):
return RaiseObject(v)
cr.register_class('RaiseObject', RaiseObject, to_builtin, from_builtin)
expected = 'Some init exception'
with pytest.raises(Exception, match=expected) as ex:
res = cr.loads(cr.dumps(o))
import datetime
from functools import partial
from itertools import permutations
import pytest
import pytz
from cereal_lazer.cereal import INSTANCE_KEY, Cereal
from cereal_lazer.naive_serializing import LimitedMethodError
cr = Cereal()
to_test = [
# Bools, Int, Float, str
True,
False,
5,
5.5,
'test',
# Dates and Datetimes
datetime.date(2018, 1, 1),
datetime.datetime(2018, 1, 1, tzinfo=pytz.UTC),
datetime.datetime(2018, 1, 1),
# Lists
[1, 2, 3, 4, 5], # int
[datetime.date(2018, 1, 1),
datetime.date(2018, 1, 2)], # dates
[1, 'test', [2., {
'test,1',
}]], # nested types