def test_token_removal(self):
"""Assert that the plugin deletes tokens whose expiration date passed
over a week ago.
"""
# Start with a clean database.
db_api.model_query(AccessToken).delete()
self.assertEqual(0, db_api.model_query(AccessToken).count())
# Build 100 tokens, each one day older than the other, with 24 hour
# expiration dates. I subtract 5 seconds here because the time it
# takes to execute the script may, or may not, result in an
# 8-day-old-token to remain valid.
new_access_tokens = []
new_refresh_tokens = []
for i in range(0, 100):
created_at = datetime.now(pytz.utc) - timedelta(days=i)
expires_in = (60 * 60 * 24) - 5 # Minus five seconds, see above.
expires_at = created_at + timedelta(seconds=expires_in)
new_access_tokens.append(
AccessToken(
user_id=1,
created_at=created_at,
expires_in=expires_in,
expires_at=expires_at,
access_token='test_token_%s' % (i,)),
)
new_access_tokens = load_data(new_access_tokens)
for token in new_access_tokens:
new_refresh_tokens.append(
RefreshToken(
user_id=1,
created_at=token.created_at,
expires_in=token.expires_in,
expires_at=token.expires_at,
access_token_id=token.id,
refresh_token='test_refresh_%s' % (token.id,))
)
new_refresh_tokens = load_data(new_refresh_tokens)
# Make sure we have 100 tokens.
self.assertEqual(100, db_api.model_query(AccessToken).count())
self.assertEqual(100, db_api.model_query(RefreshToken).count())
# Run the plugin.
plugin = TokenCleaner(CONF)
plugin.run()
# Make sure we have 8 tokens left (since one plugin starts today).
self.assertEqual(8, db_api.model_query(AccessToken).count())
self.assertEqual(8, db_api.model_query(RefreshToken).count())
def test_enabled(self):
"""Assert that this plugin responds to the flag set in our
oauth configuration block.
"""
CONF.set_override('enable', False, 'plugin_token_cleaner')
plugin = TokenCleaner(CONF)
self.assertFalse(plugin.enabled())
CONF.set_override('enable', True, 'plugin_token_cleaner')
plugin = TokenCleaner(CONF)
self.assertTrue(plugin.enabled())
CONF.clear_override('enable', 'plugin_token_cleaner')
def test_token_removal(self):
"""Assert that the plugin deletes tokens whose expiration date passed
over a week ago.
"""
# Start with a clean database.
db_api.model_query(AccessToken).delete()
self.assertEqual(0, db_api.model_query(AccessToken).count())
# Build 100 tokens, each one day older than the other, with 24 hour
# expiration dates. I subtract 5 seconds here because the time it
# takes to execute the script may, or may not, result in an
# 8-day-old-token to remain valid.
new_access_tokens = []
new_refresh_tokens = []
for i in range(0, 100):
created_at = datetime.now(pytz.utc) - timedelta(days=i)
expires_in = (60 * 60 * 24) - 5 # Minus five seconds, see above.
expires_at = created_at + timedelta(seconds=expires_in)
new_access_tokens.append(
AccessToken(user_id=1,
created_at=created_at,
expires_in=expires_in,
expires_at=expires_at,
access_token='test_token_%s' % (i, )), )
new_access_tokens = load_data(new_access_tokens)
for token in new_access_tokens:
new_refresh_tokens.append(
RefreshToken(user_id=1,
created_at=token.created_at,
expires_in=token.expires_in,
expires_at=token.expires_at,
access_token_id=token.id,
refresh_token='test_refresh_%s' % (token.id, )))
new_refresh_tokens = load_data(new_refresh_tokens)
# Make sure we have 100 tokens.
self.assertEqual(100, db_api.model_query(AccessToken).count())
self.assertEqual(100, db_api.model_query(RefreshToken).count())
# Run the plugin.
plugin = TokenCleaner(CONF)
plugin.run()
# Make sure we have 8 tokens left (since one plugin starts today).
self.assertEqual(8, db_api.model_query(AccessToken).count())
self.assertEqual(8, db_api.model_query(RefreshToken).count())
def test_enabled(self):
"""Assert that this plugin responds to the flag set in our
oauth configuration block.
"""
CONF.set_override('enable', False, 'plugin_token_cleaner')
plugin = TokenCleaner(CONF)
self.assertFalse(plugin.enabled())
CONF.set_override('enable', True, 'plugin_token_cleaner')
plugin = TokenCleaner(CONF)
self.assertTrue(plugin.enabled())
CONF.clear_override('enable', 'plugin_token_cleaner')
def test_trigger(self):
"""Assert that the this plugin runs every minute."""
plugin = TokenCleaner(CONF)
trigger = plugin.trigger()
self.assertEqual(3600, trigger.interval_length)
def test_trigger(self):
"""Assert that the this plugin runs every minute."""
plugin = TokenCleaner(CONF)
trigger = plugin.trigger()
self.assertEqual(3600, trigger.interval_length)
