1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
from types import MappingProxyType
from unittest import TestCase
from itemadapter.utils import get_field_meta_from_class
from tests import (
AttrsItem,
DataClassItem,
PydanticModel,
PydanticV1Model,
ScrapyItem,
ScrapySubclassedItem,
)
class DataclassTestCase(TestCase):
def test_false(self):
from itemadapter.adapter import DataclassAdapter
self.assertFalse(DataclassAdapter.is_item(int))
self.assertFalse(DataclassAdapter.is_item(sum))
self.assertFalse(DataclassAdapter.is_item(1234))
self.assertFalse(DataclassAdapter.is_item(object()))
self.assertFalse(DataclassAdapter.is_item("a string"))
self.assertFalse(DataclassAdapter.is_item(b"some bytes"))
self.assertFalse(DataclassAdapter.is_item({"a": "dict"}))
self.assertFalse(DataclassAdapter.is_item(["a", "list"]))
self.assertFalse(DataclassAdapter.is_item(("a", "tuple")))
self.assertFalse(DataclassAdapter.is_item({"a", "set"}))
self.assertFalse(DataclassAdapter.is_item(DataClassItem))
try:
import attrs # noqa: F401 # pylint: disable=unused-import
except ImportError:
pass
else:
self.assertFalse(DataclassAdapter.is_item(AttrsItem()))
if PydanticModel is not None:
self.assertFalse(DataclassAdapter.is_item(PydanticModel()))
if PydanticV1Model is not None:
self.assertFalse(DataclassAdapter.is_item(PydanticV1Model()))
try:
import scrapy # noqa: F401 # pylint: disable=unused-import
except ImportError:
pass
else:
self.assertFalse(DataclassAdapter.is_item(ScrapyItem()))
self.assertFalse(DataclassAdapter.is_item(ScrapySubclassedItem()))
def test_true(self):
from itemadapter.adapter import DataclassAdapter
self.assertTrue(DataclassAdapter.is_item(DataClassItem()))
self.assertTrue(DataclassAdapter.is_item(DataClassItem(name="asdf", value=1234)))
# field metadata
self.assertEqual(
get_field_meta_from_class(DataClassItem, "name"),
MappingProxyType({"serializer": str}),
)
self.assertEqual(
get_field_meta_from_class(DataClassItem, "value"),
MappingProxyType({"serializer": int}),
)
with self.assertRaises(KeyError, msg="DataClassItem does not support field: non_existent"):
get_field_meta_from_class(DataClassItem, "non_existent")
|
