mirror of
https://github.com/nonebot/nonebot2.git
synced 2024-11-28 03:15:24 +08:00
57 lines
1.3 KiB
Python
57 lines
1.3 KiB
Python
|
from typing import Any
|
||
|
|
from dataclasses import dataclass
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from nonebot.compat import (
|
||
|
|
DEFAULT_CONFIG,
|
||
|
|
Required,
|
||
|
|
FieldInfo,
|
||
|
|
PydanticUndefined,
|
||
|
|
custom_validation,
|
||
|
|
type_validate_python,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_default_config():
|
||
|
|
assert DEFAULT_CONFIG.get("extra") == "allow"
|
||
|
|
assert DEFAULT_CONFIG.get("arbitrary_types_allowed") is True
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_field_info():
|
||
|
|
# required should be convert to PydanticUndefined
|
||
|
|
assert FieldInfo(Required).default is PydanticUndefined
|
||
|
|
|
||
|
|
# field info should allow extra attributes
|
||
|
|
assert FieldInfo(test="test").extra["test"] == "test"
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_custom_validation():
|
||
|
|
called = []
|
||
|
|
|
||
|
|
@custom_validation
|
||
|
|
@dataclass
|
||
|
|
class TestModel:
|
||
|
|
test: int
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def __get_validators__(cls):
|
||
|
|
yield cls._validate_1
|
||
|
|
yield cls._validate_2
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def _validate_1(cls, v: Any) -> Any:
|
||
|
|
called.append(1)
|
||
|
|
return v
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def _validate_2(cls, v: Any) -> Any:
|
||
|
|
called.append(2)
|
||
|
|
return cls(test=v["test"])
|
||
|
|
|
||
|
|
assert type_validate_python(TestModel, {"test": 1}) == TestModel(test=1)
|
||
|
|
assert called == [1, 2]
|