def get(self, _id): log = LogModel.find_by_id(int(_id)) is_admin = auth.is_admin() if log: if not is_admin: return {'message': 'You are not privileged to continue!'}, 400 return log.json() return {'message': 'Log not found.'}, 404
def delete(self, _id): log = LogModel.find_by_id(int(_id)) is_admin = auth.is_admin() if log: if not is_admin: return {'message': 'You are not privileged to continue!'}, 400 log.delete_from_db() return {'message': 'Item deleted.'} return {'message': 'Log not found.'}, 404
def test_model_delete_to_db(self): log = LogModel.find_by_id(1).delete_from_db() self.assertEqual(None, LogModel.find_by_id(1))
def test_model_json(self): log = LogModel.find_by_id(1) self.assertEqual(1, log.json()['id'])