def test_clean_auth_tokens_expired(self): now = datetime.datetime.now() self.store.add( "testkey", json.dumps({ "token": { "exp": totimestamp(now - datetime.timedelta(days=8)) }, "revoked": False, }).encode("utf-8"), ) self.store.add( "testkey2", json.dumps({ "token": { "exp": totimestamp(now + datetime.timedelta(days=8)) }, "revoked": False, }).encode("utf-8"), ) self.assertEqual(len(self.store.keys()), 2) commands.clean_auth_tokens() self.assertEqual(len(self.store.keys()), 1) self.assertEqual(self.store.keys()[0], "testkey2")
def test_clean_auth_tokens_revoked(self): now = datetime.datetime.now() self.store.add( "testkey", json.dumps({ "token": { "exp": totimestamp(now + datetime.timedelta(days=8)) }, "revoked": False }).encode("utf-8")) self.store.add( "testkey2", json.dumps({ "token": { "exp": totimestamp(now + datetime.timedelta(days=8)) }, "revoked": True }).encode("utf-8")) self.assertEquals(len(self.store.keys()), 2) for key in self.store.keys(): print(self.store.get(key)) commands.clean_auth_tokens() self.assertEquals(len(self.store.keys()), 1) self.assertEquals(self.store.keys()[0], "testkey")
def clean_auth_tokens(): "Remove revoked and expired tokens." commands.clean_auth_tokens()
def clear_all_auth_tokens(): "Remove all authentication tokens." commands.clean_auth_tokens()