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)
