Edit

[EDIT] FILE: test_types.py

"""
Tests on the new type interface. The actual correctness of the type checking
is handled in test_jsonschema_test_suite; these tests check that TypeChecker
functions correctly and can facilitate extensions to type checking
"""
from collections import namedtuple
from unittest import TestCase

from jsonschema import ValidationError, _validators
from jsonschema._types import TypeChecker
from jsonschema.exceptions import UndefinedTypeCheck
from jsonschema.validators import Draft4Validator, extend

def equals_2(checker, instance):
    return instance == 2

def is_namedtuple(instance):
    return isinstance(instance, tuple) and getattr(instance, "_fields", None)

def is_object_or_named_tuple(checker, instance):
    if Draft4Validator.TYPE_CHECKER.is_type(instance, "object"):
        return True
    return is_namedtuple(instance)

def coerce_named_tuple(fn):
    def coerced(validator, value, instance, schema):
        if is_namedtuple(instance):
            instance = instance._asdict()
        return fn(validator, value, instance, schema)
    return coerced

required = coerce_named_tuple(_validators.required)
properties = coerce_named_tuple(_validators.properties)

class TestTypeChecker(TestCase):
    def test_is_type(self):
        checker = TypeChecker({"two": equals_2})
        self.assertEqual(
            (
                checker.is_type(instance=2, type="two"),
                checker.is_type(instance="bar", type="two"),
            ),
            (True, False),
        )

def test_is_unknown_type(self):
        with self.assertRaises(UndefinedTypeCheck) as context:
            TypeChecker().is_type(4, "foobar")
        self.assertIn("foobar", str(context.exception))

def test_checks_can_be_added_at_init(self):
        checker = TypeChecker({"two": equals_2})
        self.assertEqual(checker, TypeChecker().redefine("two", equals_2))

def test_redefine_existing_type(self):
        self.assertEqual(
            TypeChecker().redefine("two", object()).redefine("two", equals_2),
            TypeChecker().redefine("two", equals_2),
        )

def test_remove(self):
        self.assertEqual(
            TypeChecker({"two": equals_2}).remove("two"),
            TypeChecker(),
        )

def test_remove_unknown_type(self):
        with self.assertRaises(UndefinedTypeCheck) as context:
            TypeChecker().remove("foobar")
        self.assertIn("foobar", str(context.exception))

def test_redefine_many(self):
        self.assertEqual(
            TypeChecker().redefine_many({"foo": int, "bar": str}),
            TypeChecker().redefine("foo", int).redefine("bar", str),
        )

def test_remove_multiple(self):
        self.assertEqual(
            TypeChecker({"foo": int, "bar": str}).remove("foo", "bar"),
            TypeChecker(),
        )

def test_type_check_can_raise_key_error(self):
        """
        Make sure no one writes:

try:
                self._type_checkers[type](...)
            except KeyError:

ignoring the fact that the function itself can raise that.
        """

error = KeyError("Stuff")

def raises_keyerror(checker, instance):
            raise error

with self.assertRaises(KeyError) as context:
            TypeChecker({"foo": raises_keyerror}).is_type(4, "foo")

self.assertIs(context.exception, error)

class TestCustomTypes(TestCase):
    def test_simple_type_can_be_extended(self):
        def int_or_str_int(checker, instance):
            if not isinstance(instance, (int, str)):
                return False
            try:
                int(instance)
            except ValueError:
                return False
            return True

CustomValidator = extend(
            Draft4Validator,
            type_checker=Draft4Validator.TYPE_CHECKER.redefine(
                "integer", int_or_str_int,
            ),
        )
        validator = CustomValidator({"type": "integer"})

validator.validate(4)
        validator.validate("4")

with self.assertRaises(ValidationError):
            validator.validate(4.4)

def test_object_can_be_extended(self):
        schema = {"type": "object"}

Point = namedtuple("Point", ["x", "y"])

type_checker = Draft4Validator.TYPE_CHECKER.redefine(
            u"object", is_object_or_named_tuple,
        )

CustomValidator = extend(Draft4Validator, type_checker=type_checker)
        validator = CustomValidator(schema)

validator.validate(Point(x=4, y=5))

def test_object_extensions_require_custom_validators(self):
        schema = {"type": "object", "required": ["x"]}

type_checker = Draft4Validator.TYPE_CHECKER.redefine(
            u"object", is_object_or_named_tuple,
        )

CustomValidator = extend(Draft4Validator, type_checker=type_checker)
        validator = CustomValidator(schema)

Point = namedtuple("Point", ["x", "y"])
        # Cannot handle required
        with self.assertRaises(ValidationError):
            validator.validate(Point(x=4, y=5))

def test_object_extensions_can_handle_custom_validators(self):
        schema = {
            "type": "object",
            "required": ["x"],
            "properties": {"x": {"type": "integer"}},
        }

type_checker = Draft4Validator.TYPE_CHECKER.redefine(
            u"object", is_object_or_named_tuple,
        )

CustomValidator = extend(
            Draft4Validator,
            type_checker=type_checker,
            validators={"required": required, "properties": properties},
        )

validator = CustomValidator(schema)

Point = namedtuple("Point", ["x", "y"])
        # Can now process required and properties
        validator.validate(Point(x=4, y=5))

with self.assertRaises(ValidationError):
            validator.validate(Point(x="not an integer", y=5))

CANCEL