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()
