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.")