test_hypothesis_plugin.py

import typing

import pytest
from hypothesis import given, settings, strategies as st

import pydantic


class EmailsModel(pydantic.BaseModel):
    # Each of these models contains a few related fields; the idea is that
    # if there's a bug we have neither too many fields to dig through nor
    # too many models to read.
    email: pydantic.EmailStr
    name_email: pydantic.NameEmail
    pydantic.PyObject
    color: pydantic.color.Color


class JsonModel(pydantic.BaseModel):
    json_any: pydantic.Json
    json_int: pydantic.Json[int]
    json_float: pydantic.Json[float]
    json_str: pydantic.Json[str]
    json_int_or_str: pydantic.Json[typing.Union[int, str]]
    json_list_of_float: pydantic.Json[typing.List[float]]


class StringsModel(pydantic.BaseModel):
    # TODO: get PaymentCardNumber working
    # card: pydantic.PaymentCardNumber
    uuid1: pydantic.UUID1
    uuid3: pydantic.UUID3
    uuid4: pydantic.UUID4
    uuid5: pydantic.UUID5
    secbytes: pydantic.SecretBytes
    secstr: pydantic.SecretStr


class IPsModel(pydantic.BaseModel):
    address: pydantic.IPvAnyAddress
    interface: pydantic.IPvAnyInterface
    network: pydantic.IPvAnyNetwork


class URLsModel(pydantic.BaseModel):
    anyurl: pydantic.AnyUrl
    anyhttp: pydantic.AnyHttpUrl
    http: pydantic.HttpUrl
    postgres: pydantic.PostgresDsn
    redis: pydantic.RedisDsn


class NumbersModel(pydantic.BaseModel):
    strictbool: pydantic.StrictBool
    strictint: pydantic.StrictInt
    strictfloat: pydantic.StrictFloat
    strictstr: pydantic.StrictStr
    conintt: pydantic.conint(gt=10, lt=100)
    coninte: pydantic.conint(ge=10, le=100)
    conintmul: pydantic.conint(ge=10, le=100, multiple_of=7)
    confloatt: pydantic.confloat(gt=10, lt=100)
    confloate: pydantic.confloat(ge=10, le=100)
    condecimalt: pydantic.condecimal(gt=10, lt=100)
    condecimale: pydantic.condecimal(ge=10, le=100)


class ConBytesModel(pydantic.BaseModel):
    # This and the strings model aim to exercise all the length- and regex-based
    # special cases in our implementation, including Hypothesis' special handling
    # for bytestrings of fixed length.
    b00_false = pydantic.conbytes(min_length=0, max_length=0, strip_whitespace=False)
    b33_false = pydantic.conbytes(min_length=3, max_length=3, strip_whitespace=False)
    b04_false = pydantic.conbytes(min_length=0, max_length=4, strip_whitespace=False)
    b14_false = pydantic.conbytes(min_length=1, max_length=4, strip_whitespace=False)
    b24_false = pydantic.conbytes(min_length=2, max_length=4, strip_whitespace=False)
    b34_false = pydantic.conbytes(min_length=3, max_length=4, strip_whitespace=False)
    b00_true = pydantic.conbytes(min_length=0, max_length=0, strip_whitespace=True)
    b33_true = pydantic.conbytes(min_length=3, max_length=3, strip_whitespace=True)
    b04_true = pydantic.conbytes(min_length=0, max_length=4, strip_whitespace=True)
    b14_true = pydantic.conbytes(min_length=1, max_length=4, strip_whitespace=True)
    b24_true = pydantic.conbytes(min_length=2, max_length=4, strip_whitespace=True)
    b34_true = pydantic.conbytes(min_length=3, max_length=4, strip_whitespace=True)


class ConStringsModel(pydantic.BaseModel):
    s00_false = pydantic.constr(min_length=0, max_length=0, strip_whitespace=False)
    s33_false = pydantic.constr(min_length=3, max_length=3, strip_whitespace=False)
    s04_false = pydantic.constr(min_length=0, max_length=4, strip_whitespace=False)
    s14_false = pydantic.constr(min_length=1, max_length=4, strip_whitespace=False)
    s24_false = pydantic.constr(min_length=2, max_length=4, strip_whitespace=False)
    s34_false = pydantic.constr(min_length=3, max_length=4, strip_whitespace=False)
    s00_true = pydantic.constr(min_length=0, max_length=0, strip_whitespace=True)
    s33_true = pydantic.constr(min_length=3, max_length=3, strip_whitespace=True)
    s04_true = pydantic.constr(min_length=0, max_length=4, strip_whitespace=True)
    s14_true = pydantic.constr(min_length=1, max_length=4, strip_whitespace=True)
    s24_true = pydantic.constr(min_length=2, max_length=4, strip_whitespace=True)
    s34_true = pydantic.constr(min_length=3, max_length=4, strip_whitespace=True)


class CollectionsModel(pydantic.BaseModel):
    conset: pydantic.conset(int, min_items=2, max_items=4)
    conlist: pydantic.conlist(int, min_items=2, max_items=4)


class Foo:
    # Trivial class to test constrained collections element type
    pass


class CollectionsFooModel(pydantic.BaseModel):
    conset: pydantic.conset(Foo, min_items=2, max_items=4)
    conlist: pydantic.conlist(Foo, min_items=2, max_items=4)

    class Config:
        arbitrary_types_allowed = True


@pytest.mark.parametrize(
    """model""",
    [
        EmailsModel,
        # JsonModel,
        StringsModel,
        IPsModel,
        URLsModel,
        # NumbersModel,
        ConBytesModel,
        ConStringsModel,
        # CollectionsModel,
        # CollectionsFooModel,
    ],
)
@settings(max_examples=20)
@given(data=st.data())
def test_can_construct_models_with_all_fields(data, model):
    # We take successful creation of an instance to demonstrate that Hypothesis
    # knows how to provide valid values for each field, so we don't need any
    # additional assertions.
    instance = data.draw(st.from_type(model))
    assert isinstance(instance, model)