openstack
/
keystone
Code Issues Proposed changes

utf-8 encode user keys in memcache (bug 1056373) 

(cherry picked from commit 431e50a785)

Change-Id: I026dd4282742213e69c7aa02e109439b07a73c8e
This commit is contained in:
Dolph Mathews 2012-09-25 19:04:50 +00:00 committed by Joe Heck
parent 176ee9bce7
commit 0e1f05e7a8
2 changed files with 19 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
keystone/token/backends/memcache.py
View File

@ -45,6 +45,9 @@ class Token(token.Driver):
def _prefix_token_id(self, token_id):
return 'token-%s' % token_id.encode('utf-8')
def _prefix_user_id(self, user_id):
return 'usertokens-%s' % user_id.encode('utf-8')
def get_token(self, token_id):
ptk = self._prefix_token_id(token_id)
token = self.client.get(ptk)
@ -66,7 +69,7 @@ class Token(token.Driver):
if 'id' in data['user']:
token_data = token_id
user_id = data['user']['id']
user_key = 'usertokens-%s' % user_id
user_key = self._prefix_user_id(user_id)
if not self.client.append(user_key, ',%s' % token_data):
if not self.client.add(user_key, token_data):
if not self.client.append(user_key, ',%s' % token_data):
@ -83,7 +86,8 @@ class Token(token.Driver):
def list_tokens(self, user_id, tenant_id=None):
tokens = []
user_record = self.client.get('usertokens-%s' % user_id) or ""
user_key = self._prefix_user_id(user_id)
user_record = self.client.get(user_key) or ""
token_list = user_record.split(',')
for token_id in token_list:
ptk = self._prefix_token_id(token_id)

14
tests/test_backend_memcache.py
View File

@ -79,9 +79,21 @@ class MemcacheToken(test.TestCase, test_backend.TokenTests):
fake_client = MemcacheClient()
self.token_api = token_memcache.Token(client=fake_client)
def test_get_unicode(self):
def test_create_unicode_token_id(self):
token_id = unicode(uuid.uuid4().hex)
data = {'id': token_id, 'a': 'b',
'user': {'id': 'testuserid'}}
self.token_api.create_token(token_id, data)
self.token_api.get_token(token_id)
def test_create_unicode_user_id(self):
token_id = uuid.uuid4().hex
user_id = unicode(uuid.uuid4().hex)
data = {'id': token_id, 'a': 'b',
'user': {'id': user_id}}
self.token_api.create_token(token_id, data)
self.token_api.get_token(token_id)
def test_list_tokens_unicode_user_id(self):
user_id = unicode(uuid.uuid4().hex)
self.token_api.list_tokens(user_id)