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
|
from __future__ import annotations
import re
from typing import Any
import pytest
from pydantic_core import ValidationError
from .conftest import Err, PyAndJson
@pytest.mark.parametrize(
'strict_to_validator,strict_in_schema,input_value,expected',
[
(False, False, 123, 123),
(False, False, '123', 123),
(None, False, 123, 123),
(None, False, '123', 123),
(True, False, 123, 123),
(True, False, '123', Err('Input should be a valid integer [type=int_type')),
(False, True, 123, 123),
(False, True, '123', 123),
(None, True, 123, 123),
(None, True, '123', Err('Input should be a valid integer [type=int_type')),
(True, True, 123, 123),
(True, True, '123', Err('Input should be a valid integer [type=int_type')),
(False, None, 123, 123),
(False, None, '123', 123),
(None, None, 123, 123),
(None, None, '123', 123),
(True, None, 123, 123),
(True, None, '123', Err('Input should be a valid integer [type=int_type')),
],
)
def test_int_strict_argument(
py_and_json: PyAndJson, strict_to_validator: bool | None, strict_in_schema: bool | None, input_value, expected
):
schema: dict[str, Any] = {'type': 'int'}
if strict_in_schema is not None:
schema['strict'] = strict_in_schema
v = py_and_json(schema)
if isinstance(expected, Err):
assert v.isinstance_test(input_value, strict_to_validator) is False
with pytest.raises(ValidationError, match=re.escape(expected.message)):
v.validate_test(input_value, strict_to_validator)
else:
assert v.isinstance_test(input_value, strict_to_validator) is True
assert v.validate_test(input_value, strict_to_validator) == expected
|