mirror of
https://github.com/thegeeklab/ansible-later.git
synced 2024-11-18 10:50:40 +00:00
238 lines
6.5 KiB
Python
238 lines
6.5 KiB
Python
"""
|
|
Python representations of the JSON Schema Test Suite tests.
|
|
|
|
"""
|
|
|
|
import json
|
|
import os
|
|
import re
|
|
import subprocess
|
|
import sys
|
|
import unittest
|
|
|
|
from twisted.python.filepath import FilePath
|
|
import attr
|
|
|
|
from jsonschema.compat import PY3
|
|
from jsonschema.validators import validators
|
|
import jsonschema
|
|
|
|
|
|
def _find_suite():
|
|
root = os.environ.get("JSON_SCHEMA_TEST_SUITE")
|
|
if root is not None:
|
|
return FilePath(root)
|
|
|
|
root = FilePath(jsonschema.__file__).parent().sibling("json")
|
|
if not root.isdir(): # pragma: no cover
|
|
raise ValueError(
|
|
(
|
|
"Can't find the JSON-Schema-Test-Suite directory. "
|
|
"Set the 'JSON_SCHEMA_TEST_SUITE' environment "
|
|
"variable or run the tests from alongside a checkout "
|
|
"of the suite."
|
|
),
|
|
)
|
|
return root
|
|
|
|
|
|
@attr.s(hash=True)
|
|
class Suite(object):
|
|
|
|
_root = attr.ib(default=attr.Factory(_find_suite))
|
|
|
|
def _remotes(self):
|
|
jsonschema_suite = self._root.descendant(["bin", "jsonschema_suite"])
|
|
remotes = subprocess.check_output(
|
|
[sys.executable, jsonschema_suite.path, "remotes"],
|
|
)
|
|
return {
|
|
"http://localhost:1234/" + name: schema
|
|
for name, schema in json.loads(remotes.decode("utf-8")).items()
|
|
}
|
|
|
|
def benchmark(self, runner): # pragma: no cover
|
|
for name in validators:
|
|
self.version(name=name).benchmark(runner=runner)
|
|
|
|
def version(self, name):
|
|
return Version(
|
|
name=name,
|
|
path=self._root.descendant(["tests", name]),
|
|
remotes=self._remotes(),
|
|
)
|
|
|
|
|
|
@attr.s(hash=True)
|
|
class Version(object):
|
|
|
|
_path = attr.ib()
|
|
_remotes = attr.ib()
|
|
|
|
name = attr.ib()
|
|
|
|
def benchmark(self, runner): # pragma: no cover
|
|
for suite in self.tests():
|
|
for test in suite:
|
|
runner.bench_func(
|
|
name=test.fully_qualified_name,
|
|
func=test.validate_ignoring_errors,
|
|
)
|
|
|
|
def tests(self):
|
|
return (
|
|
test
|
|
for child in self._path.globChildren("*.json")
|
|
for test in self._tests_in(
|
|
subject=child.basename()[:-5],
|
|
path=child,
|
|
)
|
|
)
|
|
|
|
def format_tests(self):
|
|
path = self._path.descendant(["optional", "format"])
|
|
return (
|
|
test
|
|
for child in path.globChildren("*.json")
|
|
for test in self._tests_in(
|
|
subject=child.basename()[:-5],
|
|
path=child,
|
|
)
|
|
)
|
|
|
|
def tests_of(self, name):
|
|
return self._tests_in(
|
|
subject=name,
|
|
path=self._path.child(name + ".json"),
|
|
)
|
|
|
|
def optional_tests_of(self, name):
|
|
return self._tests_in(
|
|
subject=name,
|
|
path=self._path.descendant(["optional", name + ".json"]),
|
|
)
|
|
|
|
def to_unittest_testcase(self, *suites, **kwargs):
|
|
name = kwargs.pop("name", "Test" + self.name.title())
|
|
methods = {
|
|
test.method_name: test.to_unittest_method(**kwargs)
|
|
for suite in suites
|
|
for tests in suite
|
|
for test in tests
|
|
}
|
|
cls = type(name, (unittest.TestCase,), methods)
|
|
|
|
try:
|
|
cls.__module__ = _someone_save_us_the_module_of_the_caller()
|
|
except Exception: # pragma: no cover
|
|
# We're doing crazy things, so if they go wrong, like a function
|
|
# behaving differently on some other interpreter, just make them
|
|
# not happen.
|
|
pass
|
|
|
|
return cls
|
|
|
|
def _tests_in(self, subject, path):
|
|
for each in json.loads(path.getContent().decode("utf-8")):
|
|
yield (
|
|
_Test(
|
|
version=self,
|
|
subject=subject,
|
|
case_description=each["description"],
|
|
schema=each["schema"],
|
|
remotes=self._remotes,
|
|
**test
|
|
) for test in each["tests"]
|
|
)
|
|
|
|
|
|
@attr.s(hash=True, repr=False)
|
|
class _Test(object):
|
|
|
|
version = attr.ib()
|
|
|
|
subject = attr.ib()
|
|
case_description = attr.ib()
|
|
description = attr.ib()
|
|
|
|
data = attr.ib()
|
|
schema = attr.ib(repr=False)
|
|
|
|
valid = attr.ib()
|
|
|
|
_remotes = attr.ib()
|
|
|
|
def __repr__(self): # pragma: no cover
|
|
return "<Test {}>".format(self.fully_qualified_name)
|
|
|
|
@property
|
|
def fully_qualified_name(self): # pragma: no cover
|
|
return " > ".join(
|
|
[
|
|
self.version.name,
|
|
self.subject,
|
|
self.case_description,
|
|
self.description,
|
|
]
|
|
)
|
|
|
|
@property
|
|
def method_name(self):
|
|
delimiters = r"[\W\- ]+"
|
|
name = "test_%s_%s_%s" % (
|
|
re.sub(delimiters, "_", self.subject),
|
|
re.sub(delimiters, "_", self.case_description),
|
|
re.sub(delimiters, "_", self.description),
|
|
)
|
|
|
|
if not PY3: # pragma: no cover
|
|
name = name.encode("utf-8")
|
|
return name
|
|
|
|
def to_unittest_method(self, skip=lambda test: None, **kwargs):
|
|
if self.valid:
|
|
def fn(this):
|
|
self.validate(**kwargs)
|
|
else:
|
|
def fn(this):
|
|
with this.assertRaises(jsonschema.ValidationError):
|
|
self.validate(**kwargs)
|
|
|
|
fn.__name__ = self.method_name
|
|
reason = skip(self)
|
|
return unittest.skipIf(reason is not None, reason)(fn)
|
|
|
|
def validate(self, Validator=None, **kwargs):
|
|
resolver = jsonschema.RefResolver.from_schema(
|
|
schema=self.schema, store=self._remotes,
|
|
)
|
|
jsonschema.validate(
|
|
instance=self.data,
|
|
schema=self.schema,
|
|
cls=Validator,
|
|
resolver=resolver,
|
|
**kwargs
|
|
)
|
|
|
|
def validate_ignoring_errors(self, **kwargs): # pragma: no cover
|
|
try:
|
|
self.validate(**kwargs)
|
|
except jsonschema.ValidationError:
|
|
pass
|
|
|
|
|
|
def _someone_save_us_the_module_of_the_caller():
|
|
"""
|
|
The FQON of the module 2nd stack frames up from here.
|
|
|
|
This is intended to allow us to dynamicallly return test case classes that
|
|
are indistinguishable from being defined in the module that wants them.
|
|
|
|
Otherwise, trial will mis-print the FQON, and copy pasting it won't re-run
|
|
the class that really is running.
|
|
|
|
Save us all, this is all so so so so so terrible.
|
|
"""
|
|
|
|
return sys._getframe(2).f_globals["__name__"]
|