handle tests

Author: hramezani

src/validators/dataclass.rs

@@ -454,7 +454,7 @@ impl BuildValidator for DataclassValidator {
             )?)?,
             name,
             frozen: schema.get_as(intern!(py, "frozen"))?.unwrap_or(false),
-            slots: schema.get_as(intern!(py, "slots"))?.unwrap_or(false),
+            slots: matches!(class.hasattr(intern!(class.py(), "__slots__")), Ok(true)),
         }
         .into())
     }

tests/validators/test_dataclasses.py

@@ -1212,7 +1212,7 @@ class Model:
     m: Model
 
     m = val.validate_python({'x': 123})
-    assert m == Model(x=1)
+    assert m == Model(x=123)
 
     with pytest.raises(ValidationError):
         val.validate_python({'x': 'abc'})
@@ -1222,3 +1222,71 @@ class Model:
 
     with pytest.raises(ValidationError):
         val.validate_assignment(m, 'x', 'abc')
+
+
+def test_dataclass_slots_field_before_validator():
+    kwargs = {'slots': True}
+
+    @dataclasses.dataclass(**kwargs)
+    class Foo:
+        a: int
+        b: str
+
+        @classmethod
+        def validate_b(cls, v: bytes, info: core_schema.FieldValidationInfo) -> bytes:
+            assert v == b'hello'
+            assert info.field_name == 'b'
+            assert info.data == {'a': 1}
+            return b'hello world!'
+
+    schema = core_schema.dataclass_schema(
+        Foo,
+        core_schema.dataclass_args_schema(
+            'Foo',
+            [
+                core_schema.dataclass_field(name='a', schema=core_schema.int_schema()),
+                core_schema.dataclass_field(
+                    name='b',
+                    schema=core_schema.field_before_validator_function(Foo.validate_b, core_schema.str_schema()),
+                ),
+            ],
+        ),
+    )
+
+    v = SchemaValidator(schema)
+    foo = v.validate_python({'a': 1, 'b': b'hello'})
+    assert dataclasses.asdict(foo) == {'a': 1, 'b': 'hello world!'}
+
+
+def test_dataclass_slots_field_after_validator():
+    kwargs = {'slots': True}
+
+    @dataclasses.dataclass(**kwargs)
+    class Foo:
+        a: int
+        b: str
+
+        @classmethod
+        def validate_b(cls, v: str, info: core_schema.FieldValidationInfo) -> str:
+            assert v == 'hello'
+            assert info.field_name == 'b'
+            assert info.data == {'a': 1}
+            return 'hello world!'
+
+    schema = core_schema.dataclass_schema(
+        Foo,
+        core_schema.dataclass_args_schema(
+            'Foo',
+            [
+                core_schema.dataclass_field(name='a', schema=core_schema.int_schema()),
+                core_schema.dataclass_field(
+                    name='b',
+                    schema=core_schema.field_after_validator_function(Foo.validate_b, core_schema.str_schema()),
+                ),
+            ],
+        ),
+    )
+
+    v = SchemaValidator(schema)
+    foo = v.validate_python({'a': 1, 'b': b'hello'})
+    assert dataclasses.asdict(foo) == {'a': 1, 'b': 'hello world!'}