pydantic/tests/test_assert_in_validators.py

"""
PYTEST_DONT_REWRITE
"""
import pytest

from pydantic import BaseModel, ValidationError, validator


def test_assert_raises_validation_error():
    class Model(BaseModel):
        a: str

        @validator('a')
        def check_a(cls, v, **kwargs):
            assert v == 'a', 'invalid a'
            return v

    assert Model(a='a').a == 'a'

    with pytest.raises(ValidationError) as exc_info:
        Model(a='snap')

    expected_errors = [
        {
            'type': 'assertion_error',
            'loc': ('a',),
            'msg': 'Assertion failed, invalid a',
            'input': 'snap',
            'ctx': {'error': 'invalid a'},
        }
    ]
    actual_errors = exc_info.value.errors()
    if expected_errors != actual_errors:
        pytest.fail(f'Actual errors: {actual_errors}\nExpected errors: {expected_errors}')