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'), [])
def test_111_token_manager_addget_user(self): m = TokenManager() token = Token('token-key', 'token-secret') token.user = '******' token.access = True m.add(token) result = m.getTokensForUser('user') self.assertEqual(result, [token])
def test_112_token_manager_addget_user(self): m = TokenManager() token = Token('token-key', 'token-secret') token.user = '******' m.add(token) result = m.getTokensForUser('user') # only access tokens are tracked. self.assertEqual(result, [])
def test_113_token_manager_doubleadd_user(self): m = TokenManager() token = Token('token-key', 'token-secret') token.user = '******' token.access = True m.add(token) self.assertRaises(ValueError, m.add, token) result = m.getTokensForUser('user') # should not result in double entry. self.assertEqual(result, [token])
def test_115_token_manager_user_inconsistency(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) m._del_user_map(t2) # User must know about the token for the getter to work. self.assertEqual(m.getTokensForUser('t2user'), [])
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])