mirror of
https://github.com/kellyjonbrazil/jc.git
synced 2025-06-17 00:07:37 +02:00
67 lines
2.1 KiB
Python
67 lines
2.1 KiB
Python
"""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}.*'"
|
|
)
|