Files
python-keystoneclient/keystoneclient/tests/unit/auth/test_default_cli.py
Sean McGinnis ee55c043c2 Use unittest.mock instead of third party mock
Now that we no longer support py27, we can use the standard library
unittest.mock module instead of the third party mock lib.

Change-Id: I7498ea2353cccca7b23d9ef74015a566ac431f90
Signed-off-by: Sean McGinnis <sean.mcginnis@gmail.com>
2020-04-18 11:58:31 -05:00

93 lines
3.2 KiB
Python

# 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 argparse
from unittest import mock
import uuid
from keystoneclient.auth.identity.generic import cli
from keystoneclient import exceptions
from keystoneclient.tests.unit import utils
class DefaultCliTests(utils.TestCase):
def setUp(self):
super(DefaultCliTests, self).setUp()
self.deprecations.expect_deprecations()
def new_plugin(self, argv):
parser = argparse.ArgumentParser()
cli.DefaultCLI.register_argparse_arguments(parser)
opts = parser.parse_args(argv)
return cli.DefaultCLI.load_from_argparse_arguments(opts)
def test_endpoint_override(self):
password = uuid.uuid4().hex
url = uuid.uuid4().hex
p = self.new_plugin(['--os-auth-url', 'url',
'--os-endpoint', url,
'--os-password', password])
self.assertEqual(url, p.get_endpoint(None))
self.assertEqual(password, p._password)
def test_token_only_override(self):
self.assertRaises(exceptions.CommandError,
self.new_plugin,
['--os-token', uuid.uuid4().hex])
def test_token_endpoint_override(self):
token = uuid.uuid4().hex
endpoint = uuid.uuid4().hex
p = self.new_plugin(['--os-endpoint', endpoint,
'--os-token', token])
self.assertEqual(endpoint, p.get_endpoint(None))
self.assertEqual(token, p.get_token(None))
def test_no_auth_url(self):
exc = self.assertRaises(exceptions.CommandError,
self.new_plugin,
['--os-username', uuid.uuid4().hex])
self.assertIn('auth-url', str(exc))
@mock.patch('sys.stdin', autospec=True)
@mock.patch('getpass.getpass')
def test_prompt_password(self, mock_getpass, mock_stdin):
password = uuid.uuid4().hex
mock_stdin.isatty = lambda: True
mock_getpass.return_value = password
p = self.new_plugin(['--os-auth-url', uuid.uuid4().hex,
'--os-username', uuid.uuid4().hex])
self.assertEqual(password, p._password)
@mock.patch('sys.stdin', autospec=True)
@mock.patch('getpass.getpass')
def test_prompt_no_password(self, mock_getpass, mock_stdin):
mock_stdin.isatty = lambda: True
mock_getpass.return_value = ''
exc = self.assertRaises(exceptions.CommandError,
self.new_plugin,
['--os-auth-url', uuid.uuid4().hex,
'--os-username', uuid.uuid4().hex])
self.assertIn('password', str(exc))