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
|
import unittest
from dataclasses import field
from typing import Optional
import marshmallow
from marshmallow_dataclass import dataclass
class TestOptionalField(unittest.TestCase):
def test_optional_field(self):
@dataclass
class OptionalValue:
value: Optional[str] = "value"
schema = OptionalValue.Schema()
self.assertEqual(schema.load({"value": None}), OptionalValue(value=None))
self.assertEqual(schema.load({"value": "hello"}), OptionalValue(value="hello"))
self.assertEqual(schema.load({}), OptionalValue())
def test_optional_field_not_none(self):
@dataclass
class OptionalValueNotNone:
value: Optional[str] = field(
default="value", metadata={"allow_none": False}
)
schema = OptionalValueNotNone.Schema()
self.assertEqual(schema.load({}), OptionalValueNotNone())
self.assertEqual(
schema.load({"value": "hello"}), OptionalValueNotNone(value="hello")
)
with self.assertRaises(marshmallow.exceptions.ValidationError) as exc_cm:
schema.load({"value": None})
self.assertEqual(
exc_cm.exception.messages, {"value": ["Field may not be null."]}
)
def test_required_optional_field(self):
@dataclass
class RequiredOptionalValue:
value: Optional[str] = field(default="default", metadata={"required": True})
schema = RequiredOptionalValue.Schema()
self.assertEqual(schema.load({"value": None}), RequiredOptionalValue(None))
self.assertEqual(
schema.load({"value": "hello"}), RequiredOptionalValue(value="hello")
)
with self.assertRaises(marshmallow.exceptions.ValidationError) as exc_cm:
schema.load({})
self.assertEqual(
exc_cm.exception.messages, {"value": ["Missing data for required field."]}
)
|
