1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
from collections import namedtuple
from marshmallow import fields, Schema
from marshmallow_polyfield.polyfield import PolyField
import pytest
from tests.shapes import (
Rectangle,
Triangle,
shape_schema_serialization_disambiguation,
shape_schema_deserialization_disambiguation,
fuzzy_pos_schema_selector,
fuzzy_pos_schema_selector_by_type,
)
from tests.polyclasses import ShapePolyField, with_all
def with_both_shapes(func):
return with_all(
PolyField(
serialization_schema_selector=shape_schema_serialization_disambiguation,
deserialization_schema_selector=shape_schema_deserialization_disambiguation
),
ShapePolyField()
)(func)
def test_serializing_named_tuple():
Point = namedtuple('Point', ['x', 'y'])
field = fields.Field()
p = Point(x=4, y=2)
assert field.serialize('x', p) == 4
def test_serializing_named_tuple_with_meta():
Point = namedtuple('Point', ['x', 'y'])
p = Point(x=4, y=2)
class PointSerializer(Schema):
class Meta:
fields = ('x', 'y')
serialized = PointSerializer().dump(p)
assert serialized['x'] == 4
assert serialized['y'] == 2
@with_both_shapes
def test_serializing_polyfield_rectangle(field):
rect = Rectangle("blue", 4, 10)
Sticker = namedtuple('Sticker', ['shape', 'image'])
marshmallow_sticker = Sticker(rect, "marshmallow.png")
rect_dict = field.serialize('shape', marshmallow_sticker)
assert rect_dict == {"length": 4, "width": 10, "color": "blue"}
@with_both_shapes
def test_serializing_polyfield_None(field):
Sticker = namedtuple('Sticker', ['shape', 'image'])
marshmallow_sticker = Sticker(None, "marshmallow.png")
rect_dict = field.serialize('shape', marshmallow_sticker)
assert rect_dict is None
@with_all(
PolyField(
serialization_schema_selector=shape_schema_serialization_disambiguation,
deserialization_schema_selector=shape_schema_deserialization_disambiguation,
many=True
),
ShapePolyField(many=True)
)
def test_serializing_polyfield_many(field):
rect = Rectangle("blue", 4, 10)
tri = Triangle("red", 1, 100)
StickerCollection = namedtuple('StickerCollection', ['shapes', 'image'])
marshmallow_sticker_collection = StickerCollection([rect, tri], "marshmallow.png")
shapes = field.serialize('shapes', marshmallow_sticker_collection)
expected_shapes = [
{"length": 4, "width": 10, "color": "blue"},
{"base": 1, "height": 100, "color": "red"}
]
assert shapes == expected_shapes
@with_both_shapes
def test_invalid_polyfield(field):
Sticker = namedtuple('Sticker', ['shape', 'image'])
with pytest.raises(TypeError):
field.serialize('shape', Sticker(3, 3))
@with_both_shapes
def test_serializing_polyfield_by_parent_type(field):
rect = Rectangle("blue", 4, 10)
Sticker = namedtuple('Sticker', ['shape', 'image', 'type'])
marshmallow_sticker = Sticker(rect, "marshmallow.png", 'rectangle')
rect_dict = field.serialize('shape', marshmallow_sticker)
assert rect_dict == {"length": 4, "width": 10, "color": "blue"}
def test_serializing_polyfield_dict():
class FuzzyPosSchema(Schema):
data = PolyField(
deserialization_schema_selector=fuzzy_pos_schema_selector,
serialization_schema_selector=fuzzy_pos_schema_selector,
many=True
)
positions = {"one": {'x': 1, 'y': 1}, "two": {'x': 100, 'y': 50}}
email = 'dummy@example.com'
expected_data = {'data': [email, positions]}
data = FuzzyPosSchema().dump({'data': [email, positions]})
assert data == expected_data
class FuzzyPosSchema(Schema):
type = fields.Str(required=True)
data = PolyField(
deserialization_schema_selector=fuzzy_pos_schema_selector_by_type,
serialization_schema_selector=fuzzy_pos_schema_selector_by_type,
)
expected_data = {'type': 'str', 'data': 'fixed'}
data = FuzzyPosSchema().dump({'type': 'str', 'data': 'fixed'})
assert data == expected_data
positions = {"one": {'x': 1, 'y': 1}, "two": {'x': 100, 'y': 50}}
expected_data = {'type': 'dict', 'data': positions}
data = FuzzyPosSchema().dump({'type': 'dict', 'data': positions})
assert data == expected_data
|
