def test_serialize_using_serialized_property_setter(self): def serialized_property_setter(container: Dict, value: Dict): container["augmented_key_%s" % value] = value mappings = [JsonPropertyMapping(None, "b", json_property_setter=serialized_property_setter)] serializer = ComplexModelSerializer(mappings) self.assertDictEqual(serializer.serialize(self.complex_model), {"augmented_key_5": 5})
def test_serialize_using_serializer_cls(self): def serialized_property_setter_for_d(container: Dict, value: Dict): container["serialized_d"] = value mappings = [PropertyMapping("d", serialized_property_setter=serialized_property_setter_for_d, serializer_cls=SimpleModelSerializer)] serializer = ComplexModelSerializer(mappings) complex_model = self.complex_model expected = {"serialized_d": [{ "serialized_a": i, "serialized_b": complex_model.b + i } for i in range(len(complex_model.d))]} self.assertDictEqual(serializer.serialize(complex_model), expected)