File: test_deserialization.py

package info (click to toggle)
python-apischema 0.18.3-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 1,636 kB
  • sloc: python: 15,281; makefile: 3; sh: 2
file content (65 lines) | stat: -rw-r--r-- 1,328 bytes parent folder | download | duplicates (2)
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
from dataclasses import InitVar, dataclass, field
from typing import Generic, TypeVar

import pytest

from apischema import settings
from apischema.deserialization import get_deserialization_flattened_aliases
from apischema.metadata import flatten, init_var
from apischema.objects import object_fields


@dataclass
class A:
    a: int
    b: "B" = field(metadata=flatten)
    c: "C[int]" = field(metadata=flatten)
    d: "D" = field(metadata=flatten)
    e: InitVar[int] = field(metadata=init_var(int))
    f: int = field(init=False)


@dataclass
class B:
    g: int


T = TypeVar("T")


@dataclass
class C(Generic[T]):
    h: T


@dataclass
class D(Generic[T]):
    i: T


@dataclass
class Data:
    field: A = field(metadata=flatten)


def test_flattened_aliases():
    aliases = get_deserialization_flattened_aliases(
        Data, object_fields(Data)["field"], settings.deserialization.default_conversion
    )
    assert set(aliases) == {"a", "g", "h", "i", "e"}


@dataclass
class BadData:
    field: int = field(metadata=flatten)


def test_invalid_flattened():
    with pytest.raises(TypeError):
        list(
            get_deserialization_flattened_aliases(
                BadData,
                object_fields(BadData)["field"],
                settings.deserialization.default_conversion,
            )
        )