e05d39abb5
Switch from using novaclient to SDK for openstack console url show operation. Depends-On: https://review.opendev.org/756286 Change-Id: Ibe247825148788c549c2c1e991aae92338cdf557
248 lines
7.9 KiB
Python
248 lines
7.9 KiB
Python
# Copyright 2016 Huawei, Inc. All rights reserved.
|
|
#
|
|
# 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.
|
|
#
|
|
|
|
from unittest import mock
|
|
|
|
from openstackclient.compute.v2 import console
|
|
from openstackclient.tests.unit.compute.v2 import fakes as compute_fakes
|
|
from openstackclient.tests.unit import utils
|
|
|
|
|
|
class TestConsole(compute_fakes.TestComputev2):
|
|
|
|
def setUp(self):
|
|
super(TestConsole, self).setUp()
|
|
|
|
# SDK mock
|
|
self.app.client_manager.sdk_connection = mock.Mock()
|
|
self.app.client_manager.sdk_connection.compute = mock.Mock()
|
|
self.sdk_client = self.app.client_manager.sdk_connection.compute
|
|
self.sdk_client.find_server = mock.Mock()
|
|
self.sdk_client.get_server_console_output = mock.Mock()
|
|
|
|
|
|
class TestConsoleLog(TestConsole):
|
|
_server = compute_fakes.FakeServer.create_one_server()
|
|
|
|
def setUp(self):
|
|
super(TestConsoleLog, self).setUp()
|
|
|
|
self.sdk_client.find_server.return_value = self._server
|
|
|
|
self.cmd = console.ShowConsoleLog(self.app, None)
|
|
|
|
def test_show_no_args(self):
|
|
arglist = [
|
|
]
|
|
verifylist = [
|
|
]
|
|
self.assertRaises(utils.ParserException,
|
|
self.check_parser,
|
|
self.cmd,
|
|
arglist,
|
|
verifylist)
|
|
|
|
def test_show(self):
|
|
arglist = [
|
|
'fake_server'
|
|
]
|
|
verifylist = [
|
|
('server', 'fake_server')
|
|
]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
|
|
output = {
|
|
'output': '1st line\n2nd line\n'
|
|
}
|
|
self.sdk_client.get_server_console_output.return_value = output
|
|
self.cmd.take_action(parsed_args)
|
|
|
|
self.sdk_client.find_server.assert_called_with(
|
|
name_or_id='fake_server', ignore_missing=False)
|
|
self.sdk_client.get_server_console_output.assert_called_with(
|
|
self._server.id,
|
|
length=None
|
|
)
|
|
stdout = self.app.stdout.content
|
|
self.assertEqual(stdout[0], output['output'])
|
|
|
|
def test_show_lines(self):
|
|
arglist = [
|
|
'fake_server',
|
|
'--lines', '15'
|
|
]
|
|
verifylist = [
|
|
('server', 'fake_server'),
|
|
('lines', 15)
|
|
]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
|
|
output = {
|
|
'output': '1st line\n2nd line'
|
|
}
|
|
self.sdk_client.get_server_console_output.return_value = output
|
|
self.cmd.take_action(parsed_args)
|
|
|
|
self.sdk_client.find_server.assert_called_with(
|
|
name_or_id='fake_server', ignore_missing=False)
|
|
self.sdk_client.get_server_console_output.assert_called_with(
|
|
self._server.id,
|
|
length=15
|
|
)
|
|
|
|
|
|
class TestConsoleUrlShow(TestConsole):
|
|
_server = compute_fakes.FakeServer.create_one_server()
|
|
|
|
def setUp(self):
|
|
super(TestConsoleUrlShow, self).setUp()
|
|
self.sdk_client.find_server.return_value = self._server
|
|
fake_console_data = {'url': 'http://localhost',
|
|
'protocol': 'fake_protocol',
|
|
'type': 'fake_type'}
|
|
self.sdk_client.create_console = mock.Mock(
|
|
return_value=fake_console_data)
|
|
|
|
self.columns = (
|
|
'protocol',
|
|
'type',
|
|
'url',
|
|
)
|
|
self.data = (
|
|
fake_console_data['protocol'],
|
|
fake_console_data['type'],
|
|
fake_console_data['url']
|
|
)
|
|
|
|
self.cmd = console.ShowConsoleURL(self.app, None)
|
|
|
|
def test_console_url_show_by_default(self):
|
|
arglist = [
|
|
'foo_vm',
|
|
]
|
|
verifylist = [
|
|
('url_type', 'novnc'),
|
|
('server', 'foo_vm'),
|
|
]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
columns, data = self.cmd.take_action(parsed_args)
|
|
self.sdk_client.create_console.assert_called_once_with(
|
|
self._server.id,
|
|
console_type='novnc')
|
|
self.assertEqual(self.columns, columns)
|
|
self.assertEqual(self.data, data)
|
|
|
|
def test_console_url_show_with_novnc(self):
|
|
arglist = [
|
|
'--novnc',
|
|
'foo_vm',
|
|
]
|
|
verifylist = [
|
|
('url_type', 'novnc'),
|
|
('server', 'foo_vm'),
|
|
]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
columns, data = self.cmd.take_action(parsed_args)
|
|
self.sdk_client.create_console.assert_called_once_with(
|
|
self._server.id,
|
|
console_type='novnc')
|
|
self.assertEqual(self.columns, columns)
|
|
self.assertEqual(self.data, data)
|
|
|
|
def test_console_url_show_with_xvpvnc(self):
|
|
arglist = [
|
|
'--xvpvnc',
|
|
'foo_vm',
|
|
]
|
|
verifylist = [
|
|
('url_type', 'xvpvnc'),
|
|
('server', 'foo_vm'),
|
|
]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
columns, data = self.cmd.take_action(parsed_args)
|
|
self.sdk_client.create_console.assert_called_once_with(
|
|
self._server.id,
|
|
console_type='xvpvnc')
|
|
self.assertEqual(self.columns, columns)
|
|
self.assertEqual(self.data, data)
|
|
|
|
def test_console_url_show_with_spice(self):
|
|
arglist = [
|
|
'--spice',
|
|
'foo_vm',
|
|
]
|
|
verifylist = [
|
|
('url_type', 'spice-html5'),
|
|
('server', 'foo_vm'),
|
|
]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
columns, data = self.cmd.take_action(parsed_args)
|
|
self.sdk_client.create_console.assert_called_once_with(
|
|
self._server.id,
|
|
console_type='spice-html5')
|
|
self.assertEqual(self.columns, columns)
|
|
self.assertEqual(self.data, data)
|
|
|
|
def test_console_url_show_with_rdp(self):
|
|
arglist = [
|
|
'--rdp',
|
|
'foo_vm',
|
|
]
|
|
verifylist = [
|
|
('url_type', 'rdp-html5'),
|
|
('server', 'foo_vm'),
|
|
]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
columns, data = self.cmd.take_action(parsed_args)
|
|
self.sdk_client.create_console.assert_called_once_with(
|
|
self._server.id,
|
|
console_type='rdp-html5')
|
|
self.assertEqual(self.columns, columns)
|
|
self.assertEqual(self.data, data)
|
|
|
|
def test_console_url_show_with_serial(self):
|
|
arglist = [
|
|
'--serial',
|
|
'foo_vm',
|
|
]
|
|
verifylist = [
|
|
('url_type', 'serial'),
|
|
('server', 'foo_vm'),
|
|
]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
columns, data = self.cmd.take_action(parsed_args)
|
|
self.sdk_client.create_console.assert_called_once_with(
|
|
self._server.id,
|
|
console_type='serial')
|
|
self.assertEqual(self.columns, columns)
|
|
self.assertEqual(self.data, data)
|
|
|
|
def test_console_url_show_with_mks(self):
|
|
arglist = [
|
|
'--mks',
|
|
'foo_vm',
|
|
]
|
|
verifylist = [
|
|
('url_type', 'webmks'),
|
|
('server', 'foo_vm'),
|
|
]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
columns, data = self.cmd.take_action(parsed_args)
|
|
self.sdk_client.create_console.assert_called_once_with(
|
|
self._server.id,
|
|
console_type='webmks')
|
|
self.assertEqual(self.columns, columns)
|
|
self.assertEqual(self.data, data)
|