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)