def test_101_token_manager_addget(self): m = TokenManager() token = Token('token-key', 'token-secret') m.add(token) result = m.get('token-key') self.assertEqual(result, token) result = m.get(token) self.assertEqual(result, token)
def test_200_token_manager_generate_request_token(self): m = TokenManager() consumer = Consumer('consumer-key', 'consumer-secret') callback = 'oob' token = m.generateRequestToken(consumer.key, callback) self.assertEqual(m.get(token.key), token) self.assertEqual(m.get(token.key).consumer_key, consumer.key) self.assertEqual(m.get(token.key).access, False) self.assertEqual(m.getRequestToken(token.key).key, token.key)
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_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])
def test_100_token_manager_empty(self): m = TokenManager() self.assertEqual(m.get('token-key'), None)