Added a primary_key to AuthToken, fixed some unbound variables, and now all unit tests pass

2 files changed, 5 insertions, 2 deletions

diff --git a/nova/api/rackspace/auth.py b/nova/api/rackspace/auth.py
index 1ef90c324..ce5a967eb 100644
--- a/nova/api/rackspace/auth.py
+++ b/nova/api/rackspace/auth.py
@@ -8,6 +8,9 @@ from nova import flags
 from nova import auth
 from nova import manager
 from nova import db
+from nova import utils
+
+FLAGS = flags.FLAGS
 
 class Context(object):
     pass
@@ -15,7 +18,7 @@ class Context(object):
 class BasicApiAuthManager(object):
     """ Implements a somewhat rudimentary version of Rackspace Auth"""
 
-    def __init__(self):
+    def __init__(self, host=None, db_driver=None):
         if not host:
             host = FLAGS.host
         self.host = host                                                                                                                                     
diff --git a/nova/db/sqlalchemy/models.py b/nova/db/sqlalchemy/models.py
index 6e1c0ce16..161c5f1bc 100644
--- a/nova/db/sqlalchemy/models.py
+++ b/nova/db/sqlalchemy/models.py
@@ -401,7 +401,7 @@ class AuthToken(BASE, NovaBase):
     are a string representing the actual token and a user id for mapping 
     to the actual user"""
     __tablename__ = 'auth_tokens'
-    token_hash = Column(String(255))
+    token_hash = Column(String(255), primary_key=True)
     user_id = Column(Integer)
     server_manageent_url = Column(String(255))
     storage_url = Column(String(255))