def test_encode_auth_token(self):
user = User(userId=Counter.getNextSequence('userId'),
userName='******',
email='*****@*****.**',
dateRegistered=datetime.datetime.utcnow())
user.userPassword = '******'
user.save()
auth_token = user.encode_auth_token(user.userId)
self.assertTrue(isinstance(auth_token, bytes))
def save_new_user(data):
new_user = User(userId=Counter.getNextSequence('userId'),
userName=data['userName'],
email=data['email'],
dateRegistered=datetime.datetime.utcnow())
new_user.userPassword = data['userPassword']
try:
new_user.save()
except DuplicateKeyError:
response_object = {
'status': 'fail',
'message': 'User already exists. Please Log in.'
}
return response_object, 409 # 409 status : Conflict
else:
return generate_token(new_user)
