def test_deserialization_new_object(self):
gs = GoldenSchema(
Alchemist,
nested_map=self.nested_map['alchemists']['nested_map'],
new_obj=True)
serialized = {
'formulae': [{
'author_id': 1,
'id': 1,
'title': 'transmutation'
}],
'id': 1,
'name': 'Albertus Magnus'
}
deserialized = gs.load(serialized)
assert isinstance(deserialized, Alchemist)
assert deserialized.id is None
assert len(deserialized.formulae) == 1
formula = deserialized.formulae[0]
assert isinstance(formula, Formula)
assert formula.id is None
def test_serialization(self):
gs = GoldenSchema(WizardCollege, nested_map=self.nested_map)
serialized = gs.dump(self.school)
expected = {
"id":
1,
"name":
"Bogwarts",
"alchemists": [{
"formulae": [{
"title": "transmutation",
"author_id": 1,
"id": 1
}],
"school_id":
1,
"id":
1,
"name":
"Albertus Magnus"
}]
}
assert serialized == expected
def test_serialization_nest_golden_schema_with_extra_field(self):
class GoldenFormulaWithIngredients(GoldenSchema):
ingredients = fields.List(fields.String())
self.nested_map['alchemists']['nested_map']['formulae']['class'] = \
GoldenFormulaWithIngredients(Formula)
gs = GoldenSchema(WizardCollege, nested_map=self.nested_map)
self.school.alchemists[0].formulae[0].ingredients = ['magic', 'lead']
serialized = gs.dump(self.school)
expected = {
"id":
1,
"name":
"Bogwarts",
"alchemists": [{
"formulae": [{
"title": "transmutation",
"author_id": 1,
"id": 1,
"ingredients": ['magic', 'lead']
}],
"school_id":
1,
"id":
1,
"name":
"Albertus Magnus"
}]
}
assert serialized == expected
def test_serialize_camel_to_snake(self):
gs = GoldenSchema(CamelFormula,
nested_map=self.nested_map,
camel_to_snake=True)
camel_formula = CamelFormula(id=1,
title='transmutation',
camelAttribute='value')
serialized = gs.dump(camel_formula)
expected = {
'camel_attribute': 'value',
'id': 1,
'title': 'transmutation'
}
assert serialized == expected
def test_deserialization(self):
gs = GoldenSchema(WizardCollege, nested_map=self.nested_map)
serialized = {
"id":
1,
"name":
"Bogwarts",
"alchemists": [{
"formulae": [{
"title": "transmutation",
"author_id": 1,
"id": 1
}],
"school_id":
1,
"id":
1,
"name":
"Albertus Magnus"
}]
}
college = gs.load(serialized)
assert isinstance(college, WizardCollege)
assert college.id == 1
assert college.name == 'Bogwarts'
assert len(college.alchemists) == 1
alchemist = college.alchemists[0]
assert isinstance(alchemist, Alchemist)
assert alchemist.id == 1
assert alchemist.name == 'Albertus Magnus'
assert alchemist.school_id == 1
assert len(alchemist.formulae) == 1
formula = alchemist.formulae[0]
assert isinstance(formula, Formula)
assert formula.id == 1
assert formula.title == 'transmutation'
def test_serialize_snake_to_camel(self):
gs = GoldenSchema(
Alchemist,
nested_map=self.nested_map['alchemists']['nested_map'],
snake_to_camel=True)
serialized = gs.dump(self.alchemist)
expected = {
'formulae': [{
'authorId': 1,
'id': 1,
'title': 'transmutation'
}],
'id': 1,
'name': 'Albertus Magnus',
'schoolId': 1
}
assert serialized == expected
def test_error_when_setting_both_casing_types(self):
with pytest.raises(ValueError) as excinfo:
GoldenSchema(Alchemist, camel_to_snake=True, snake_to_camel=True)
assert ('Only one of snake_to_camel or camel_to_snake can be True'
in str(excinfo.value))