-
Notifications
You must be signed in to change notification settings - Fork 37
/
test_example_files.py
169 lines (129 loc) · 5.12 KB
/
test_example_files.py
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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
from __future__ import annotations
import dataclasses
import importlib.util
import shlex
from pathlib import Path
import pytest
import ruamel.yaml
from check_jsonschema.parsers.json5 import ENABLED as JSON5_ENABLED
from check_jsonschema.parsers.toml import ENABLED as TOML_ENABLED
yaml = ruamel.yaml.YAML(typ="safe")
HERE = Path(__file__).parent
EXAMPLE_FILES = HERE.parent / "example-files"
EXAMPLE_HOOK_FILES = HERE.parent / "example-files" / "hooks"
EXAMPLE_EXPLICIT_FILES = HERE.parent / "example-files" / "explicit-schema"
def _iter_hook_config():
config_file = HERE.parent.parent / ".pre-commit-hooks.yaml"
with open(config_file) as fp:
for hook in yaml.load(fp):
hookid = hook["id"]
if not hookid.startswith("check-"):
continue
hookid = hookid[len("check-") :]
entry = shlex.split(hook["entry"])
yield (hookid, entry)
HOOK_CONFIG = dict(_iter_hook_config())
def _build_hook_cases(category):
res = {}
for hookid in HOOK_CONFIG:
example_dir = EXAMPLE_HOOK_FILES / category / hookid
if example_dir.exists():
for example in example_dir.iterdir():
if example.name == "_config.yaml":
continue
res[str(example.relative_to(EXAMPLE_HOOK_FILES / category))] = hookid
return res
def _get_explicit_cases(category):
res = []
example_dir = EXAMPLE_EXPLICIT_FILES / category
for example in example_dir.iterdir():
res.append(str(example.relative_to(EXAMPLE_EXPLICIT_FILES / category)))
return res
POSITIVE_HOOK_CASES = _build_hook_cases("positive")
NEGATIVE_HOOK_CASES = _build_hook_cases("negative")
@pytest.mark.parametrize("case_name", POSITIVE_HOOK_CASES.keys())
def test_hook_positive_examples(case_name, run_line):
rcase = ResolvedCase.load_positive(case_name)
hook_id = POSITIVE_HOOK_CASES[case_name]
ret = run_line(HOOK_CONFIG[hook_id] + [rcase.path] + rcase.add_args)
assert ret.exit_code == 0
@pytest.mark.parametrize("case_name", NEGATIVE_HOOK_CASES.keys())
def test_hook_negative_examples(case_name, run_line):
rcase = ResolvedCase.load_negative(case_name)
hook_id = NEGATIVE_HOOK_CASES[case_name]
ret = run_line(HOOK_CONFIG[hook_id] + [rcase.path] + rcase.add_args)
assert ret.exit_code == 1
@pytest.mark.parametrize("case_name", _get_explicit_cases("positive"))
def test_explicit_positive_examples(case_name, run_line):
_check_file_format_skip(case_name)
casedir = EXAMPLE_EXPLICIT_FILES / "positive" / case_name
instance = casedir / "instance.json"
if not instance.exists():
instance = casedir / "instance.yaml"
if not instance.exists():
instance = casedir / "instance.toml"
if not instance.exists():
raise Exception("could not find an instance file for test case")
schema = casedir / "schema.json"
if not schema.exists():
schema = casedir / "schema.yaml"
if not schema.exists():
raise Exception("could not find a schema file for test case")
ret = run_line(
[
"check-jsonschema",
"--schemafile",
str(schema),
str(instance),
]
)
assert ret.exit_code == 0
def _check_file_format_skip(case_name):
if case_name.endswith("json5") and not JSON5_ENABLED:
pytest.skip("cannot check json5 support without json5 enabled")
if case_name.endswith("toml") and not TOML_ENABLED:
pytest.skip("cannot check toml support without toml enabled")
@dataclasses.dataclass
class ResolvedCase:
category: str
path: str
add_args: list[str]
config: dict
def check_skip(self) -> None:
if "requires_packages" in self.config:
for pkg in self.config["requires_packages"]:
if _package_is_installed(pkg):
continue
pytest.skip(f"cannot check because '{pkg}' is not installed")
def __post_init__(self) -> None:
self.check_skip()
@classmethod
def load_positive(cls: type[ResolvedCase], case_name: str) -> ResolvedCase:
return cls._load("positive", case_name)
@classmethod
def load_negative(cls: type[ResolvedCase], case_name: str) -> ResolvedCase:
return cls._load("negative", case_name)
@classmethod
def _load(cls: type[ResolvedCase], category: str, case_name: str) -> ResolvedCase:
_check_file_format_skip(case_name)
path = EXAMPLE_HOOK_FILES / category / case_name
config = cls._load_file_config(path.parent / "_config.yaml", path.name)
return cls(
category=category,
path=str(path),
add_args=config.get("add_args", []),
config=config,
)
@staticmethod
def _load_file_config(config_path, name):
if not config_path.is_file():
return {}
with open(config_path) as fp:
loaded_conf = yaml.load(fp)
files_section = loaded_conf.get("files", {})
return files_section.get(name, {})
def _package_is_installed(pkg: str) -> bool:
spec = importlib.util.find_spec(pkg)
if spec is None:
return False
return True