def test_250_token_manager_claim(self):
m = TokenManager()
consumer = Consumer('consumer-key', 'consumer-secret')
callback = 'oob'
token = m.generateRequestToken(consumer.key, callback)
m.claimRequestToken(token, 'user')
self.assertEqual(token.user, 'user')
self.assertTrue(token.expiry > int(time.time()))
def test_300_token_manager_generate_access_token(self):
m = TokenManager()
consumer = Consumer('consumer-key', 'consumer-secret')
callback = 'oob'
server_token = m.generateRequestToken(consumer.key, callback)
verifier = server_token.verifier
# Also simulate user claiming the token
m.claimRequestToken(server_token.key, 'user')
# now simulate passing only the key and secret to consumer
request_token = Token(server_token.key, server_token.secret)
token = m.generateAccessToken(consumer.key, request_token.key)
self.assertEqual(m.get(token.key), token)
self.assertEqual(m.get(token.key).consumer_key, consumer.key)
self.assertEqual(m.get(token.key).access, True)
self.assertEqual(m.get(token.key).user, 'user')
# Also, assert that this token key is available in the method
# that returns these keys by user id.
self.assertEqual(m.getTokensForUser('user'), [token])
