✅ add tests for validator

nonebot/compat.py

@@ -387,15 +387,15 @@ def field_validator(
         field: str,
         /,
         *fields: str,
-        mode: Literal["after"],
+        mode: Literal["after"] = ...,
         check_fields: Optional[bool] = None,
     ): ...
 
     def field_validator(
         field: str,
         /,
         *fields: str,
-        mode: Literal["before", "after"],
+        mode: Literal["before", "after"] = "after",
         check_fields: Optional[bool] = None,
     ):
         if mode == "before":

tests/test_compat.py

@@ -7,11 +7,13 @@
 from nonebot.compat import (
     DEFAULT_CONFIG,
     FieldInfo,
+    field_validator,
     PydanticUndefined,
     Required,
     TypeAdapter,
     custom_validation,
     model_dump,
+    model_validator,
     type_validate_json,
     type_validate_python,
 )
@@ -30,6 +32,32 @@ def test_field_info():
     assert FieldInfo(test="test").extra["test"] == "test"
 
 
+def test_field_validator():
+    class TestModel(BaseModel):
+        foo: int
+        bar: str
+
+        @field_validator("foo")
+        @classmethod
+        def test_validator(cls, v: Any) -> Any:
+            if v > 0:
+                return v
+            raise ValueError("test must be greater than 0")
+
+        @field_validator("bar", mode="before")
+        @classmethod
+        def test_validator_before(cls, v: Any) -> Any:
+            if not isinstance(v, str):
+                v = str(v)
+            return v
+
+    assert TestModel(foo=1, bar="test").foo == 1
+    assert TestModel(foo=1, bar=123).bar == "123"
+
+    with pytest.raises(ValidationError):
+        TestModel(foo=0, bar="test")
+
+
 def test_type_adapter():
     t = TypeAdapter(Annotated[int, FieldInfo(ge=1)])
 
@@ -53,6 +81,30 @@ class TestModel(BaseModel):
     assert model_dump(TestModel(test1=1, test2=2), exclude={"test1"}) == {"test2": 2}
 
 
+def test_model_validator():
+    class TestModel(BaseModel):
+        foo: int
+        bar: str
+
+        @model_validator(mode="before")
+        @classmethod
+        def test_validator_before(cls, data: Any) -> Any:
+            if isinstance(data, dict):
+                if "foo" not in data:
+                    data["foo"] = 1
+            return data
+
+        @model_validator(mode="after")
+        def test_validator_after(self):
+            if self.bar == "test":
+                raise ValueError("bar should not be test")
+
+    assert type_validate_python(TestModel, {"bar": "aaa"}).foo == 1
+
+    with pytest.raises(ValidationError):
+        type_validate_python(TestModel, {"foo": 1, "bar": "test"})
+
+
 def test_custom_validation():
     called = []