From 22a6e8891ef7b50f0c6a4849ea9f6768c394f7ee Mon Sep 17 00:00:00 2001
From: Dean Troyer <dtroyer@gmail.com>
Date: Fri, 16 Aug 2013 00:03:58 -0500
Subject: [PATCH] Add Identity v2 user tests

* implement Identity command tests for v2 user

Also re-work the user create and set commands for exclusive options
(--enable|--disable) to actually behave properly.  Yay tests!

Change-Id: Ie1ec2569b3d85a9d556ee70f2e8f69fd2a3c03c8
---
 openstackclient/identity/v2_0/user.py         |  68 ++-
 .../tests/identity/v2_0/test_users.py         | 496 ++++++++++++++++++
 2 files changed, 537 insertions(+), 27 deletions(-)
 create mode 100644 openstackclient/tests/identity/v2_0/test_users.py

diff --git a/openstackclient/identity/v2_0/user.py b/openstackclient/identity/v2_0/user.py
index 78f2164637..7174d4ce74 100644
--- a/openstackclient/identity/v2_0/user.py
+++ b/openstackclient/identity/v2_0/user.py
@@ -17,7 +17,6 @@
 
 import logging
 import six
-import sys
 
 from cliff import command
 from cliff import lister
@@ -53,15 +52,14 @@ class CreateUser(show.ShowOne):
         enable_group = parser.add_mutually_exclusive_group()
         enable_group.add_argument(
             '--enable',
-            dest='enabled',
             action='store_true',
-            default=True,
-            help='Enable user')
+            help='Enable user (default)',
+        )
         enable_group.add_argument(
             '--disable',
-            dest='enabled',
-            action='store_false',
-            help='Disable user')
+            action='store_true',
+            help='Disable user',
+        )
         return parser
 
     def take_action(self, parsed_args):
@@ -74,13 +72,19 @@ class CreateUser(show.ShowOne):
             ).id
         else:
             project_id = None
+        enabled = True
+        if parsed_args.disable:
+            enabled = False
         user = identity_client.users.create(
             parsed_args.name,
             parsed_args.password,
             parsed_args.email,
             tenant_id=project_id,
-            enabled=parsed_args.enabled,
+            enabled=enabled,
         )
+        # NOTE(dtroyer): The users.create() method wants 'tenant_id' but
+        #                the returned resource has 'tenantId'.  Sigh.
+        #                We're using project_id now inside OSC so there.
         user._info.update(
             {'project_id': user._info.pop('tenantId')}
         )
@@ -217,40 +221,50 @@ class SetUser(command.Command):
         enable_group = parser.add_mutually_exclusive_group()
         enable_group.add_argument(
             '--enable',
-            dest='enabled',
             action='store_true',
-            default=True,
-            help='Enable user (default)')
+            help='Enable user (default)',
+        )
         enable_group.add_argument(
             '--disable',
-            dest='enabled',
-            action='store_false',
-            help='Disable user')
+            action='store_true',
+            help='Disable user',
+        )
         return parser
 
     def take_action(self, parsed_args):
         self.log.debug('take_action(%s)' % parsed_args)
+
         identity_client = self.app.client_manager.identity
         user = utils.find_resource(identity_client.users, parsed_args.user)
-        kwargs = {}
-        if parsed_args.name:
-            kwargs['name'] = parsed_args.name
-        if parsed_args.email:
-            kwargs['email'] = parsed_args.email
+
+        if parsed_args.password:
+            identity_client.users.update_password(
+                user.id,
+                parsed_args.password,
+            )
+
         if parsed_args.project:
             project = utils.find_resource(
                 identity_client.tenants,
                 parsed_args.project,
             )
-            kwargs['tenantId'] = project.id
-        if 'enabled' in parsed_args:
-            kwargs['enabled'] = parsed_args.enabled
+            identity_client.users.update_tenant(
+                user.id,
+                project.id,
+            )
 
-        if not len(kwargs):
-            sys.stdout.write("User not updated, no arguments present")
-            return
-        identity_client.users.update(user.id, **kwargs)
-        return
+        kwargs = {}
+        if parsed_args.name:
+            kwargs['name'] = parsed_args.name
+        if parsed_args.email:
+            kwargs['email'] = parsed_args.email
+        if parsed_args.enable:
+            kwargs['enabled'] = True
+        if parsed_args.disable:
+            kwargs['enabled'] = False
+
+        if len(kwargs):
+            identity_client.users.update(user.id, **kwargs)
 
 
 class ShowUser(show.ShowOne):
diff --git a/openstackclient/tests/identity/v2_0/test_users.py b/openstackclient/tests/identity/v2_0/test_users.py
new file mode 100644
index 0000000000..49a5262b56
--- /dev/null
+++ b/openstackclient/tests/identity/v2_0/test_users.py
@@ -0,0 +1,496 @@
+#   Copyright 2013 Nebula Inc.
+#
+#   Licensed under the Apache License, Version 2.0 (the "License"); you may
+#   not use this file except in compliance with the License. You may obtain
+#   a copy of the License at
+#
+#        http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#   WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#   License for the specific language governing permissions and limitations
+#   under the License.
+#
+
+import copy
+
+from openstackclient.identity.v2_0 import user
+from openstackclient.tests import fakes
+from openstackclient.tests.identity import fakes as identity_fakes
+from openstackclient.tests import utils
+
+
+IDENTITY_API_VERSION = "2.0"
+
+user_id = 'aaaaaaaa-aaaa-aaaa-aaaa-aaaaaaaaaaaa'
+user_name = 'paul'
+user_description = 'Sir Paul'
+user_email = 'paul@applecorps.com'
+
+project_id = '8-9-64'
+project_name = 'beatles'
+project_description = 'Fab Four'
+
+USER = {
+    'id': user_id,
+    'name': user_name,
+    'tenantId': project_id,
+    'email': user_email,
+    'enabled': True,
+}
+
+PROJECT = {
+    'id': project_id,
+    'name': project_name,
+    'description': project_description,
+    'enabled': True,
+}
+
+PROJECT_2 = {
+    'id': project_id + '-2222',
+    'name': project_name + ' reprise',
+    'description': project_description + 'plus four more',
+    'enabled': True,
+}
+
+
+class TestUser(utils.TestCommand):
+
+    def setUp(self):
+        super(TestUser, self).setUp()
+        self.app.client_manager.identity = \
+            identity_fakes.FakeIdentityv2Client()
+
+        # Get a shortcut to the TenantManager Mock
+        self.projects_mock = self.app.client_manager.identity.tenants
+        # Get a shortcut to the UserManager Mock
+        self.users_mock = self.app.client_manager.identity.users
+
+
+class TestUserCreate(TestUser):
+
+    def setUp(self):
+        super(TestUserCreate, self).setUp()
+
+        self.projects_mock.get.return_value = fakes.FakeResource(
+            None,
+            copy.deepcopy(PROJECT),
+            loaded=True,
+        )
+        self.users_mock.create.return_value = fakes.FakeResource(
+            None,
+            copy.deepcopy(USER),
+            loaded=True,
+        )
+
+        # Get the command object to test
+        self.cmd = user.CreateUser(self.app, None)
+
+    def test_user_create_no_options(self):
+        arglist = [user_name]
+        verifylist = [
+            ('name', user_name),
+        ]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        columns, data = self.cmd.take_action(parsed_args)
+
+        # Set expected values
+        kwargs = {
+            'enabled': True,
+            'tenant_id': None,
+        }
+        # users.create(name, password, email, tenant_id=None, enabled=True)
+        self.users_mock.create.assert_called_with(
+            user_name,
+            None,
+            None,
+            **kwargs
+        )
+
+        collist = ('email', 'enabled', 'id', 'name', 'project_id')
+        self.assertEqual(columns, collist)
+        datalist = (user_email, True, user_id, user_name, project_id)
+        self.assertEqual(data, datalist)
+
+    def test_user_create_password(self):
+        arglist = ['--password', 'secret', user_name]
+        verifylist = [('password', 'secret')]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        columns, data = self.cmd.take_action(parsed_args)
+
+        # Set expected values
+        kwargs = {
+            'enabled': True,
+            'tenant_id': None,
+        }
+        # users.create(name, password, email, tenant_id=None, enabled=True)
+        self.users_mock.create.assert_called_with(
+            user_name,
+            'secret',
+            None,
+            **kwargs
+        )
+
+        collist = ('email', 'enabled', 'id', 'name', 'project_id')
+        self.assertEqual(columns, collist)
+        datalist = (user_email, True, user_id, user_name, project_id)
+        self.assertEqual(data, datalist)
+
+    def test_user_create_email(self):
+        arglist = ['--email', 'barney@example.com', user_name]
+        verifylist = [('email', 'barney@example.com')]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        columns, data = self.cmd.take_action(parsed_args)
+
+        # Set expected values
+        kwargs = {
+            'enabled': True,
+            'tenant_id': None,
+        }
+        # users.create(name, password, email, tenant_id=None, enabled=True)
+        self.users_mock.create.assert_called_with(
+            user_name,
+            None,
+            'barney@example.com',
+            **kwargs
+        )
+
+        collist = ('email', 'enabled', 'id', 'name', 'project_id')
+        self.assertEqual(columns, collist)
+        datalist = (user_email, True, user_id, user_name, project_id)
+        self.assertEqual(data, datalist)
+
+    def test_user_create_project(self):
+        # Return the new project
+        self.projects_mock.get.return_value = fakes.FakeResource(
+            None,
+            copy.deepcopy(PROJECT_2),
+            loaded=True,
+        )
+        # Set up to return an updated user
+        USER_2 = copy.deepcopy(USER)
+        USER_2['tenantId'] = PROJECT_2['id']
+        self.users_mock.create.return_value = fakes.FakeResource(
+            None,
+            USER_2,
+            loaded=True,
+        )
+
+        arglist = ['--project', PROJECT_2['name'], user_name]
+        verifylist = [('project', PROJECT_2['name'])]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        columns, data = self.cmd.take_action(parsed_args)
+
+        # Set expected values
+        kwargs = {
+            'enabled': True,
+            'tenant_id': PROJECT_2['id'],
+        }
+        # users.create(name, password, email, tenant_id=None, enabled=True)
+        self.users_mock.create.assert_called_with(
+            user_name,
+            None,
+            None,
+            **kwargs
+        )
+
+        collist = ('email', 'enabled', 'id', 'name', 'project_id')
+        self.assertEqual(columns, collist)
+        datalist = (user_email, True, user_id, user_name, PROJECT_2['id'])
+        self.assertEqual(data, datalist)
+
+    def test_user_create_enable(self):
+        arglist = ['--enable', project_name]
+        verifylist = [('enable', True), ('disable', False)]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        columns, data = self.cmd.take_action(parsed_args)
+
+        # Set expected values
+        kwargs = {
+            'enabled': True,
+            'tenant_id': None,
+        }
+        # users.create(name, password, email, tenant_id=None, enabled=True)
+        self.users_mock.create.assert_called_with(
+            project_name,
+            None,
+            None,
+            **kwargs
+        )
+
+        collist = ('email', 'enabled', 'id', 'name', 'project_id')
+        self.assertEqual(columns, collist)
+        datalist = (user_email, True, user_id, user_name, project_id)
+        self.assertEqual(data, datalist)
+
+    def test_user_create_disable(self):
+        arglist = ['--disable', project_name]
+        verifylist = [('enable', False), ('disable', True)]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        columns, data = self.cmd.take_action(parsed_args)
+
+        # Set expected values
+        kwargs = {
+            'enabled': False,
+            'tenant_id': None,
+        }
+        # users.create(name, password, email, tenant_id=None, enabled=True)
+        self.users_mock.create.assert_called_with(
+            project_name,
+            None,
+            None,
+            **kwargs
+        )
+
+        collist = ('email', 'enabled', 'id', 'name', 'project_id')
+        self.assertEqual(columns, collist)
+        datalist = (user_email, True, user_id, user_name, project_id)
+        self.assertEqual(data, datalist)
+
+
+class TestUserDelete(TestUser):
+
+    def setUp(self):
+        super(TestUserDelete, self).setUp()
+
+        # This is the return value for utils.find_resource()
+        self.users_mock.get.return_value = fakes.FakeResource(
+            None,
+            copy.deepcopy(USER),
+            loaded=True,
+        )
+        self.users_mock.delete.return_value = None
+
+        # Get the command object to test
+        self.cmd = user.DeleteUser(self.app, None)
+
+    def test_user_delete_no_options(self):
+        arglist = [user_id]
+        verifylist = [
+            ('user', user_id),
+        ]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        self.cmd.take_action(parsed_args)
+
+        self.users_mock.delete.assert_called_with(user_id)
+
+
+class TestUserList(TestUser):
+
+    def setUp(self):
+        super(TestUserList, self).setUp()
+
+        self.users_mock.list.return_value = [
+            fakes.FakeResource(
+                None,
+                copy.deepcopy(USER),
+                loaded=True,
+            ),
+        ]
+
+        # Get the command object to test
+        self.cmd = user.ListUser(self.app, None)
+
+    def test_user_list_no_options(self):
+        arglist = []
+        verifylist = []
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        columns, data = self.cmd.take_action(parsed_args)
+
+        self.users_mock.list.assert_called_with()
+
+        collist = ('ID', 'Name')
+        self.assertEqual(columns, collist)
+        datalist = ((user_id, user_name), )
+        self.assertEqual(tuple(data), datalist)
+
+    def test_user_list_project(self):
+        arglist = ['--project', project_id]
+        verifylist = [('project', project_id)]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        columns, data = self.cmd.take_action(parsed_args)
+
+        self.users_mock.list.assert_called_with()
+
+        collist = ('ID', 'Name')
+        self.assertEqual(columns, collist)
+        datalist = ((user_id, user_name), )
+        self.assertEqual(tuple(data), datalist)
+
+    def test_user_list_long(self):
+        arglist = ['--long']
+        verifylist = [('long', True)]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        columns, data = self.cmd.take_action(parsed_args)
+
+        self.users_mock.list.assert_called_with()
+
+        collist = ('ID', 'Name', 'Project', 'Email', 'Enabled')
+        self.assertEqual(columns, collist)
+        datalist = ((user_id, user_name, project_id, user_email, True), )
+        self.assertEqual(tuple(data), datalist)
+
+
+class TestUserSet(TestUser):
+
+    def setUp(self):
+        super(TestUserSet, self).setUp()
+
+        self.projects_mock.get.return_value = fakes.FakeResource(
+            None,
+            copy.deepcopy(PROJECT),
+            loaded=True,
+        )
+        self.users_mock.get.return_value = fakes.FakeResource(
+            None,
+            copy.deepcopy(USER),
+            loaded=True,
+        )
+
+        # Get the command object to test
+        self.cmd = user.SetUser(self.app, None)
+
+    def test_user_set_no_options(self):
+        arglist = [user_name]
+        verifylist = [
+            ('user', user_name),
+            ('enable', False),
+            ('disable', False),
+            ('project', None),
+        ]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        self.cmd.take_action(parsed_args)
+
+        self.assertFalse(self.users_mock.update.called)
+
+    def test_user_set_name(self):
+        arglist = ['--name', 'qwerty', user_name]
+        verifylist = [('name', 'qwerty')]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        self.cmd.take_action(parsed_args)
+
+        # Set expected values
+        kwargs = {
+            'name': 'qwerty',
+        }
+        self.users_mock.update.assert_called_with(user_id, **kwargs)
+
+    def test_user_set_password(self):
+        arglist = ['--password', 'secret', user_name]
+        verifylist = [('password', 'secret')]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        self.cmd.take_action(parsed_args)
+
+        self.users_mock.update_password.assert_called_with(user_id, 'secret')
+
+    def test_user_set_email(self):
+        arglist = ['--email', 'barney@example.com', user_name]
+        verifylist = [('email', 'barney@example.com')]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        self.cmd.take_action(parsed_args)
+
+        # Set expected values
+        kwargs = {
+            'email': 'barney@example.com',
+        }
+        self.users_mock.update.assert_called_with(user_id, **kwargs)
+
+    def test_user_set_project(self):
+        arglist = ['--project', project_id, user_name]
+        verifylist = [('project', project_id)]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        self.cmd.take_action(parsed_args)
+
+        self.users_mock.update_tenant.assert_called_with(
+            user_id,
+            project_id,
+        )
+
+    def test_user_set_enable(self):
+        arglist = ['--enable', user_name]
+        verifylist = [('enable', True), ('disable', False)]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        self.cmd.take_action(parsed_args)
+
+        # Set expected values
+        kwargs = {
+            'enabled': True,
+        }
+        self.users_mock.update.assert_called_with(user_id, **kwargs)
+
+    def test_user_set_disable(self):
+        arglist = ['--disable', user_name]
+        verifylist = [('enable', False), ('disable', True)]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        self.cmd.take_action(parsed_args)
+
+        # Set expected values
+        kwargs = {
+            'enabled': False,
+        }
+        self.users_mock.update.assert_called_with(user_id, **kwargs)
+
+
+class TestUserShow(TestUser):
+
+    def setUp(self):
+        super(TestUserShow, self).setUp()
+
+        self.users_mock.get.return_value = fakes.FakeResource(
+            None,
+            copy.deepcopy(USER),
+            loaded=True,
+        )
+
+        # Get the command object to test
+        self.cmd = user.ShowUser(self.app, None)
+
+    def test_user_show(self):
+        arglist = [user_id]
+        verifylist = [('user', user_id)]
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        # DisplayCommandBase.take_action() returns two tuples
+        columns, data = self.cmd.take_action(parsed_args)
+
+        self.users_mock.get.assert_called_with(user_id)
+
+        collist = ('email', 'enabled', 'id', 'name', 'project_id')
+        self.assertEqual(columns, collist)
+        datalist = (user_email, True, user_id, user_name, project_id)
+        self.assertEqual(data, datalist)