Fix test_default_roles to work with mongo-orchestration.

ajdavis · ajdavis · commit 620c1bd49184 · 2014-11-24T14:20:47.000-05:00
diff --git a/test/test_database.py b/test/test_database.py
@@ -463,14 +463,18 @@ def test_default_roles(self):
         # "self.client" is logged in as root.
         auth_admin = self.client.admin
         try:
-            info = auth_admin.command('usersInfo', db_user)['users'][0]
+            auth_admin.add_user('test_default_roles', 'pass')
+            info = auth_admin.command(
+                'usersInfo', 'test_default_roles')['users'][0]
+
             self.assertEqual("root", info['roles'][0]['role'])
 
             # Read only "admin" user
             auth_admin.add_user('ro-admin', 'pass', read_only=True)
             info = auth_admin.command('usersInfo', 'ro-admin')['users'][0]
             self.assertEqual("readAnyDatabase", info['roles'][0]['role'])
         finally:
+            auth_admin.remove_user('test_default_roles')
             auth_admin.remove_user('ro-admin')
 
         # "Non-admin" user
