mirror of
https://github.com/datahub-project/datahub.git
synced 2025-07-07 09:11:47 +00:00
25 lines
647 B
Python
25 lines
647 B
Python
|
import pytest
|
||
|
|
from pydantic import ValidationError
|
||
|
|
|
||
|
|
from datahub.configuration.common import ConfigModel
|
||
|
|
from datahub.configuration.validate_field_rename import pydantic_renamed_field
|
||
|
|
|
||
|
|
|
||
|
|
def test_field_rename():
|
||
|
|
class TestModel(ConfigModel):
|
||
|
|
b: str
|
||
|
|
|
||
|
|
_validate_deprecated = pydantic_renamed_field("a", "b")
|
||
|
|
|
||
|
|
v = TestModel.parse_obj({"b": "original"})
|
||
|
|
assert v.b == "original"
|
||
|
|
|
||
|
|
v = TestModel.parse_obj({"a": "renamed"})
|
||
|
|
assert v.b == "renamed"
|
||
|
|
|
||
|
|
with pytest.raises(ValidationError):
|
||
|
|
TestModel.parse_obj({"a": "foo", "b": "bar"})
|
||
|
|
|
||
|
|
with pytest.raises(ValidationError):
|
||
|
|
TestModel.parse_obj({})
|