예제 #1
0
    def test_crud(self):
        user = UserModel('Test', 'password')

        self.assertIsNone(
            user.find_by_username('Test'),
            "Found a user with name '{}', but expected not to.".format(
                user.username))
        self.assertIsNone(user.find_by_id(1),
                          "Found a user with id: 1, but expected not to.")

        user.save_to_db()

        self.assertIsNotNone(
            user.find_by_username('Test'),
            "Did not find a user with name '{}', but expected to.".format(
                user.username))
        self.assertIsNotNone(
            user.find_by_id(1),
            "Did not find a user with id: 1, but expected to.")

        user.delete_from_db()

        self.assertIsNone(
            user.find_by_username('Test'),
            "Found a user with name '{}', but expected not to.".format(
                user.username))
        self.assertIsNone(user.find_by_id(1),
                          "Found a user with id: 1, but expected not to.")
예제 #2
0
def add_claims_to_jwt(identity):
    user = UserModel.find_by_id(identity)
    return {'is_admin': user.role == UserModel.ROLE_ADMIN}
예제 #3
0
 def get(self, id):
     user = UserModel.find_by_id(id)
     if user:
         return user.json()
     return {'message': 'User not found.'}, 404