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
|
"""A dataclass for a test case suitable for the CTS JSON schema."""
from dataclasses import dataclass
from dataclasses import field
from typing import Any
from typing import Dict
from typing import List
from typing import Mapping
from typing import Optional
from typing import Sequence
from typing import Union
from jsonpath import NodeList
@dataclass
class Case:
name: str
selector: str
document: Union[Mapping[str, Any], Sequence[Any], None] = None
result: Any = None
results: Optional[List[Any]] = None
result_paths: Optional[List[str]] = None
results_paths: Optional[List[List[str]]] = None
invalid_selector: Optional[bool] = None
tags: List[str] = field(default_factory=list)
def as_dict(self) -> Dict[str, Any]:
rv: Dict[str, Any] = {
"name": self.name,
"selector": self.selector,
}
if self.document is not None:
rv["document"] = self.document
if self.result is not None:
rv["result"] = self.result
rv["result_paths"] = self.result_paths
else:
rv["results"] = self.results
rv["results_paths"] = self.results_paths
else:
assert self.invalid_selector
rv["invalid_selector"] = True
rv["tags"] = self.tags
return rv
def assert_nodes(self, nodes: NodeList) -> None:
"""Assert that `nodes` matches this test case."""
if self.results is not None:
assert self.results_paths is not None
assert nodes.values() in self.results
assert nodes.paths() in self.results_paths
else:
assert self.result_paths is not None
assert nodes.values() == self.result
assert nodes.paths() == self.result_paths
|
