d4f0c1ccb8
Added hosts_shell, tests_hosts_shell, and common package for cliutils to support handling hosts-list sub command with no arguments. In parallel, the ability to handle subcommand parsing was implemented in a basic form in order to implement and test "craton host-list" with no additional args. Added optional arguments to craton shell and accompanying unit tests for username, password, and url to enable creating a session and instantiate the client properly to enable other unit tests without hardcoding dummy values. Change-Id: I05a42a06e0436e7a0de6b898d4d6e50168e6dd36
104 lines
4.3 KiB
Python
104 lines
4.3 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.
|
|
|
|
"""Tests for `cratonclient.shell.main` module."""
|
|
|
|
|
|
import mock
|
|
import re
|
|
|
|
from testtools import matchers
|
|
|
|
from cratonclient.shell import main
|
|
from cratonclient.tests import base
|
|
|
|
|
|
class TestMainShell(base.ShellTestCase):
|
|
"""Test our craton main shell."""
|
|
|
|
re_options = re.DOTALL | re.MULTILINE
|
|
|
|
@mock.patch('cratonclient.shell.main.CratonShell.main')
|
|
def test_main_returns_successfully(self, cratonShellMainMock):
|
|
"""Verify that main returns as expected."""
|
|
cratonShellMainMock.return_value = 0
|
|
self.assertEqual(main.main(), 0)
|
|
|
|
def test_print_help_no_args(self):
|
|
"""Verify that no arguments prints out help by default."""
|
|
required_help_responses = [
|
|
'.*?^usage: craton',
|
|
'.*?^See "craton help COMMAND" '
|
|
'for help on a specific command.',
|
|
]
|
|
stdout, stderr = self.shell('')
|
|
for r in required_help_responses:
|
|
self.assertThat((stdout + stderr),
|
|
matchers.MatchesRegex(r, self.re_options))
|
|
|
|
def test_print_help_with_args(self):
|
|
"""Verify that help command(s) prints out help text correctly."""
|
|
required_help_responses = [
|
|
'.*?^usage: craton',
|
|
'.*?^See "craton help COMMAND" '
|
|
'for help on a specific command.',
|
|
]
|
|
for help_args in ['-h', '--help']:
|
|
stdout, stderr = self.shell(help_args)
|
|
for r in required_help_responses:
|
|
self.assertThat((stdout + stderr),
|
|
matchers.MatchesRegex(r, self.re_options))
|
|
|
|
@mock.patch('cratonclient.v1.client.Client')
|
|
def test_main_craton_url(self, mock_client):
|
|
"""Verify that craton-url command is used for client connection."""
|
|
self.shell('--craton-url http://localhost:9999/ host-list')
|
|
mock_client.assert_called_with(mock.ANY, 'http://localhost:9999/')
|
|
|
|
@mock.patch('cratonclient.session.Session')
|
|
@mock.patch('cratonclient.v1.client.Client')
|
|
def test_main_craton_project_id(self, mock_client, mock_session):
|
|
"""Verify --craton-project-id command is used for client connection."""
|
|
self.shell('--craton-project-id 99 host-list')
|
|
mock_session.assert_called_with(username=mock.ANY,
|
|
token=mock.ANY,
|
|
project_id=99)
|
|
mock_client.assert_called_with(mock.ANY, mock.ANY)
|
|
|
|
@mock.patch('cratonclient.session.Session')
|
|
@mock.patch('cratonclient.v1.client.Client')
|
|
def test_main_os_username(self, mock_client, mock_session):
|
|
"""Verify --os-username command is used for client connection."""
|
|
self.shell('--os-username test host-list')
|
|
mock_session.assert_called_with(username='test',
|
|
token=mock.ANY,
|
|
project_id=mock.ANY)
|
|
mock_client.assert_called_with(mock.ANY, mock.ANY)
|
|
|
|
@mock.patch('cratonclient.session.Session')
|
|
@mock.patch('cratonclient.v1.client.Client')
|
|
def test_main_os_password(self, mock_client, mock_session):
|
|
"""Verify --os-password command is used for client connection."""
|
|
self.shell('--os-password test host-list')
|
|
mock_session.assert_called_with(username=mock.ANY,
|
|
token='test',
|
|
project_id=mock.ANY)
|
|
|
|
mock_client.assert_called_with(mock.ANY, mock.ANY)
|
|
|
|
@mock.patch('cratonclient.shell.main.CratonShell.main')
|
|
def test_main_catches_exception(self, cratonShellMainMock):
|
|
"""Verify exceptions will be caught and shell will exit properly."""
|
|
cratonShellMainMock.side_effect = Exception(mock.Mock(status=404),
|
|
'some error')
|
|
self.assertRaises(SystemExit, main.main)
|