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
|
"""jc - JSON test utils"""
import json
import os
import sys
from pathlib import Path
import jc
VERBOSE = False
if '-v' in sys.argv or '--verbose' in sys.argv:
VERBOSE = True
def _test_print(data: str) -> None:
if VERBOSE:
print(data)
def _open_file(file_path, ext):
return open(Path(file_path).with_suffix(ext), 'r', encoding='utf-8')
def _get_base_dir(file_path):
return os.path.dirname(os.path.abspath(file_path))
def _get_parser_name_from_path(parser_path):
return Path(parser_path).stem[len('test_'):]
def _get_fixtures(base_dir, parser_name):
fixtures = {x.stem: str(x.with_suffix('')) for x in
(list(Path(base_dir).glob(f"**/{parser_name}--*.*")))}
return fixtures
def run_no_data(self, test_parser_path, expected):
"""Call this function to run a test for no input data for a parser"""
parser_name = _get_parser_name_from_path(test_parser_path)
with self.subTest(f"'no data test' for parser '{parser_name}': "):
self.assertEqual(jc.parse(parser_name, '', quiet=True), expected)
def run_all_fixtures(self, test_parser_path):
"""Call this function to run tests for all fixtures for a parser"""
parser_name = _get_parser_name_from_path(test_parser_path)
base_dir = _get_base_dir(test_parser_path)
fixtures = _get_fixtures(base_dir, parser_name).items()
if not fixtures:
raise ValueError(f"No fixtures found for '{parser_name}' tests!")
_test_print(f"\n Run all fixtures for parser '{parser_name}':")
for file, file_path in fixtures:
_test_print(f" - test '{parser_name}' parser with fixture: '{file}'")
with self.subTest(f"fixture: '{file}'"):
with _open_file(file_path, '.out') as in_file, \
_open_file(file_path, '.json') as json_file:
f_in = in_file.read()
f_json = json.loads(json_file.read())
self.assertEqual(
jc.parse(parser_name, f_in, quiet=True),
f_json,
f"Should be equal for test files: '{file_path}.*'"
)
|
