def test_252_token_manager_claim_fail_missing(self):
m = TokenManager()
consumer = Consumer('consumer-key', 'consumer-secret')
callback = 'oob'
token = m.generateRequestToken(consumer.key, callback)
m.remove(token) # remove it
self.assertRaises(TokenInvalidError, m.claimRequestToken, token, 'u')
self.assertEqual(token.user, None)
def test_103_token_manager_remove(self):
m = TokenManager()
t1 = Token('token-key', 'token-secret')
t2 = Token('token-key2', 'token-secret')
m.add(t1)
m.add(t2)
m.remove(t1.key)
m.remove(t2)
self.assertEqual(len(m._tokens), 1)
def test_500_token_manager_get_dummy(self):
m = TokenManager()
token = m.get(m.DUMMY_KEY)
self.assertEqual(token.secret, m.DUMMY_SECRET)
# Now this is where it might get a bit interesting. For dealing
# with dummy tokens, is it better to generate a new one, or just
# store it and provide one to ensure constant time? Either way,
# if somehow the dummy is removed, the utility shouldn't have to
# die in a fire.
m.remove(token)
token = m.get(m.DUMMY_KEY)
self.assertEqual(token, None)
def test_114_token_manager_addremove_user(self):
m = TokenManager()
t1 = Token('token-key', 'token-secret')
t1.user = '******'
t1.access = True
t2 = Token('token-key2', 'token-secret')
t2.user = '******'
t2.access = True
m.add(t1)
m.add(t2)
self.assertEqual(m.getTokensForUser('user'), [])
self.assertEqual(m.getTokensForUser('t1user'), [t1])
self.assertEqual(m.getTokensForUser('t2user'), [t2])
m.remove(t1.key)
m.remove(t2)
self.assertEqual(m.getTokensForUser('t1user'), [])
self.assertEqual(m.getTokensForUser('t2user'), [])
