Beispiel #1
0
 def wrapper(*args, **kwargs):
     validator = Validator(schema)
     data = request.get_json()
     validator.validate(data)
     if validator.isvalid == False:
         response_body = { 'message': 'Invalid payload' }
         return make_response(jsonify(response_body), 400)
     g.validated_data = data
     return f(*args, **kwargs)
Beispiel #2
0
class ValidatorTestCase(unittest.TestCase):
    def setUp(self):
        self.data = {}
        self.schema_mock = mock.MagicMock(spec=colander.MappingSchema)
        self.validator = Validator(self.schema_mock)

    def test_validate(self):
        self.validator.validate(self.data)
        self.schema_mock.deserialize.assert_called_once_with(self.data)
        self.assertTrue(self.validator.isvalid)

    def test_validate_invalid(self):
        schema_mock = mock.MagicMock(spec=colander.SchemaNode)
        self.schema_mock.deserialize.side_effect = mock.MagicMock(
            side_effect=colander.Invalid(schema_mock))
        self.validator.validate(self.data)
        self.assertFalse(self.validator.isvalid)