def test_find_by_jti(self): token1 = TokenRegistry('jti_1', 'type_1', 'username', True, dt.datetime.utcnow()) token2 = TokenRegistry('jti_2', 'type_2', 'username', False, dt.datetime.utcnow() + dt.timedelta(hours=1)) token1.save(False) token2.save() retrieved = TokenRegistry.find_by_jti('jti_1') jti_none = TokenRegistry.find_by_jti('jti_dontexist') assert (retrieved.jti == 'jti_1') assert (jti_none is None)
def delete(self): jti = get_raw_jwt()["jti"] token = TokenRegistry.find_by_jti(jti) if token is not None: try: token.revoked = False token.save() except: db.session.rollback() report_error_and_abort(500, "logout", "Logout failed(2).") return generate_message_response(200, "logout", "Logged out.")