import pytest

from pydantic.error_wrappers import ValidationError

from sqlmodel import SQLModel

def test_validation_pydantic_v2(clear_sqlmodel):

"""Test validation of implicit and explicit None values.

# For consistency with pydantic, validators are not to be called on

# arguments that are not explicitly provided.

https://github.com/tiangolo/sqlmodel/issues/230

https://github.com/samuelcolvin/pydantic/issues/1223

"""

from pydantic import field_validator

class Hero(SQLModel):

name: str | None = None

secret_name: str | None = None

age: int | None = None

@field_validator("name", "secret_name", "age")

def reject_none(cls, v):

assert v is not None

return v

Hero.model_validate({"age": 25})

with pytest.raises(ValidationError):

Hero.model_validate({"name": None, "age": 25})