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
|
import sys
import unittest
from typing import Optional
import marshmallow
import marshmallow.fields
from marshmallow_dataclass import dataclass
if sys.version_info >= (3, 9):
from typing import Annotated
else:
from typing_extensions import Annotated
class TestAnnotatedField(unittest.TestCase):
def test_annotated_field(self):
@dataclass
class AnnotatedValue:
value: Annotated[str, marshmallow.fields.Email]
default_string: Annotated[
Optional[str], marshmallow.fields.String(load_default="Default String")
] = None
schema = AnnotatedValue.Schema()
self.assertEqual(
schema.load({"value": "test@test.com"}),
AnnotatedValue(value="test@test.com", default_string="Default String"),
)
self.assertEqual(
schema.load({"value": "test@test.com", "default_string": "override"}),
AnnotatedValue(value="test@test.com", default_string="override"),
)
with self.assertRaises(marshmallow.exceptions.ValidationError):
schema.load({"value": "notavalidemail"})
|
