From a0c43a05126d9cca83b58b788fbc8e24f004fb66 Mon Sep 17 00:00:00 2001
From: "sirish.bitra" <sirish.bitra@gmail.com>
Date: Thu, 12 May 2011 16:31:26 +0530
Subject: optimization of test cases and handling multi token

---
 keystone/db/sqlalchemy/api.py | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'keystone/db')

diff --git a/keystone/db/sqlalchemy/api.py b/keystone/db/sqlalchemy/api.py
index f1817a24..9fd9a734 100644
--- a/keystone/db/sqlalchemy/api.py
+++ b/keystone/db/sqlalchemy/api.py
@@ -471,6 +471,13 @@ def token_for_user(user_id, session=None):
     return result
 
 
+def token_for_user_tenant(user_id, tenant_id, session=None):
+    if not session:
+        session = get_session()
+    result = session.query(models.Token).filter_by(
+        user_id=user_id, tenant_id=tenant_id).order_by("expires desc").first()
+    return result
+
 def user_tenant_create(values):
     user_tenant_ref = models.UserTenantAssociation()
     user_tenant_ref.update(values)
-- 
cgit