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
|
from __future__ import annotations
from datamodel_code_generator.model.dataclass import DataModelField
from datamodel_code_generator.types import DataType
def test_data_model_field_process_const() -> None:
"""Test process_const method functionality."""
field = DataModelField(name="test_field", data_type=DataType(type="str"), required=True, extras={"const": "v1"})
field.process_const()
assert field.const is True
assert field.nullable is False
assert field.data_type.literals == ["v1"]
assert field.default == "v1"
def test_data_model_field_process_const_no_const() -> None:
"""Test process_const when no const is in extras."""
field = DataModelField(name="test_field", data_type=DataType(type="str"), required=True, extras={})
original_nullable = field.nullable
original_default = field.default
original_const = field.const
field.process_const()
assert field.const == original_const
assert field.nullable == original_nullable
assert field.default == original_default
|
