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)
def __init__(self, account): self.account = account Validator.__init__(self) self.dao = UserDao() self.rules = {'account': { 'account_valid' : [self.account_valid, 'email', 'The account does not exist.'] }, 'password':{ 'password_valid' : [self.password_valid, 'password', 'Wrong password'] } }
def __init__(self): Validator.__init__(self) self.rules = {'username': { 'required' : [None, True], 'maxlength':[None, 32]}, 'password':{'required' : [None, True], 'minlength':[None, 8], 'maxlength':[None, 32] }, 'email': { 'required' : [None, True], 'maxlength':[None, 64], 'unique' : [self.unique, 'email', 'This email exists, please try another.'] } }
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)
def __init__(self, bp, rule="", methods=None, validator=None, pre_processes=None): self.rule = rule self.methods = methods or ['GET'] self.validator = Validator(validator) if validator else None self.pre_processes = pre_processes or [] self.bp = bp
def setUp(self): self.data = {} self.schema_mock = mock.MagicMock(spec=colander.MappingSchema) self.validator = Validator(self.schema_mock)