Sandbox::https://redirect.github.com/michaeloliverx/sqlmodel/blob/main/tests/test

Repository navigation

from typing import Optional

import pytest

from pydantic import validator

from pydantic.error_wrappers import ValidationError

from sqlmodel import SQLModel

def test_validation(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

"""

class Hero(SQLModel):

name: Optional[str] = None

secret_name: Optional[str] = None

age: Optional[int] = None

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

def reject_none(cls, v):

assert v is not None

return v

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

with pytest.raises(ValidationError):

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