• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• GenSON-master
  • LICENSE
  • test
    • test_builder.py
    • test_gen_multi.py
    • test_gen_single.py
    • test_bin.py
    • test_add_single.py
    • test_add_multi.py
    • test_custom.py
    • test_misuse.py
    • __init__.py
    • test_seed_schema.py
    • base.py
  • README.rst
  • setup.py
  • AUTHORS.rst
  • genson
    • cli.py
    • __init__.py
    • schema
      • strategies
        • scalar.py
        • object.py
        • array.py
        • __init__.py
        • base.py
      • node.py
      • __init__.py
      • builder.py
  • setup.cfg
  • HISTORY.rst
  • bin
    • test.py
    • genson.py
  • .gitignore
  • MANIFEST.in

import sys
import unittest
import jsonschema
from pkg_resources import Requirement
from genson import SchemaNode, SchemaBuilder

PYTHON_VERSION = sys.version[:sys.version.find(' ')]


class BaseTestCase(unittest.TestCase):

    def setUp(self):
        self.builder = self.CLASS()
        self._objects = []
        self._schemas = []

    def set_schema_options(self, **options):
        self.builder = SchemaNode(**options)

    def add_object(self, obj):
        self.builder.add_object(obj)
        self._objects.append(obj)

    def add_schema(self, schema):
        self.builder.add_schema(schema)
        self._schemas.append(schema)

    def assertObjectValidates(self, obj):
        jsonschema.Draft7Validator(self.builder.to_schema()).validate(obj)

    def assertObjectDoesNotValidate(self, obj):
        with self.assertRaises(jsonschema.exceptions.ValidationError):
            jsonschema.Draft7Validator(self.builder.to_schema()).validate(obj)

    def assertResult(self, expected, enforceUserContract=True):
        self.assertEqual(
            expected, self.builder.to_schema(),
            'Generated schema (below) does not match expected (above)')
        if enforceUserContract:
            self.assertUserContract()

    def assertUserContract(self):
        self._assertSchemaIsValid()
        self._assertComponentObjectsValidate()

    def _assertSchemaIsValid(self):
        jsonschema.Draft7Validator.check_schema(self.builder.to_schema())

    def _assertComponentObjectsValidate(self):
        compiled_schema = self.builder.to_schema()
        for obj in self._objects:
            jsonschema.Draft7Validator(compiled_schema).validate(obj)


class SchemaNodeTestCase(BaseTestCase):
    CLASS = SchemaNode


class SchemaBuilderTestCase(BaseTestCase):
    CLASS = SchemaBuilder


# backwards compatibility

def only_for_python_version(version_specifier):
    req = Requirement.parse('python%s' % version_specifier)

    def handler(func):
        if PYTHON_VERSION in req:
            return func
        else:
            return unittest.skip('Python version %s does not match: %s'
                                 % (PYTHON_VERSION, req))(func)
    return handler