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'), [])