550 lines
22 KiB
Python
550 lines
22 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.
|
|
#
|
|
|
|
import copy
|
|
import mock
|
|
import uuid
|
|
|
|
from osc_lib import exceptions
|
|
from osc_lib import utils
|
|
|
|
from moganclient.osc.v1 import server
|
|
from moganclient.tests.unit import base as test_base
|
|
from moganclient.tests.unit import fakes
|
|
from moganclient.v1 import server as server_mgr
|
|
|
|
|
|
class TestServer(test_base.TestBaremetalComputeV1):
|
|
fake_server = fakes.FakeServer.create_one_server()
|
|
|
|
columns = (
|
|
'availability_zone',
|
|
'created_at',
|
|
'description',
|
|
'extra',
|
|
'image_uuid',
|
|
'instance_type_uuid',
|
|
'links',
|
|
'name',
|
|
'network_info',
|
|
'updated_at',
|
|
'uuid')
|
|
|
|
data = (
|
|
fake_server.availability_zone,
|
|
fake_server.created_at,
|
|
fake_server.description,
|
|
fake_server.image_uuid,
|
|
fake_server.instance_type_uuid,
|
|
fake_server.links,
|
|
fake_server.name,
|
|
fake_server.network_info,
|
|
fake_server.updated_at,
|
|
fake_server.uuid)
|
|
|
|
|
|
@mock.patch.object(utils, 'find_resource')
|
|
@mock.patch.object(server_mgr.ServerManager, '_create')
|
|
class TestServerCreate(TestServer):
|
|
def setUp(self):
|
|
super(TestServerCreate, self).setUp()
|
|
self.cmd = server.CreateServer(self.app, None)
|
|
self.app.client_manager.image = mock.Mock()
|
|
|
|
def _test_create_fake_server(self, mock_create, mock_find,
|
|
name, flavor_id, image_id, networks,
|
|
description=None,
|
|
availability_zone=None, extra=None):
|
|
arglist = [
|
|
name,
|
|
'--flavor', flavor_id,
|
|
'--image', image_id]
|
|
verifylist = [
|
|
('name', name),
|
|
('flavor', flavor_id),
|
|
('image', image_id)]
|
|
called_networks = copy.deepcopy(networks)
|
|
for nic in called_networks:
|
|
if 'port-type' in nic:
|
|
nic['port_type'] = nic['port-type']
|
|
del nic['port-type']
|
|
if 'net-id' in nic:
|
|
nic['net_id'] = nic['net-id']
|
|
del nic['net-id']
|
|
called_data = {'name': name,
|
|
'image_uuid': image_id,
|
|
'instance_type_uuid': flavor_id,
|
|
'networks': called_networks}
|
|
for network in networks:
|
|
network_id = network.get('net-id')
|
|
port_type = network.get('port-type')
|
|
if port_type:
|
|
arglist.extend(
|
|
['--nic',
|
|
'net-id=' + network_id + ',port-type=' + port_type])
|
|
verifylist.append(
|
|
('nic', [{'net-id': network_id, 'port-type': port_type}]))
|
|
else:
|
|
arglist.extend(['--nic', 'net-id=' + network_id])
|
|
verifylist.append(('nic', [{'net-id': network_id}]))
|
|
if description:
|
|
arglist.extend(['--description', description])
|
|
verifylist.append(('description', description))
|
|
called_data['description'] = description
|
|
if availability_zone:
|
|
arglist.extend(['--availability-zone', availability_zone])
|
|
verifylist.append(('availability_zone', availability_zone))
|
|
called_data['availability_zone'] = availability_zone
|
|
if extra:
|
|
arglist.extend(['--extra', extra])
|
|
verifylist.append(('extra', extra))
|
|
called_data['extra'] = extra
|
|
|
|
flavor_obj = mock.Mock()
|
|
flavor_obj.uuid = flavor_id
|
|
image_obj = mock.Mock()
|
|
image_obj.id = image_id
|
|
mock_find.side_effect = [flavor_obj, image_obj]
|
|
fk_server = fakes.FakeServer.create_one_server(called_data)
|
|
mock_create.return_value = fk_server
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
columns, data = self.cmd.take_action(parsed_args)
|
|
|
|
mock_create.assert_called_once_with(
|
|
'/instances',
|
|
data=called_data)
|
|
self.assertEqual(self.columns, columns)
|
|
expected_data = (
|
|
fk_server.availability_zone,
|
|
fk_server.created_at,
|
|
fk_server.description,
|
|
fk_server.extra,
|
|
fk_server.image_uuid,
|
|
fk_server.instance_type_uuid,
|
|
fk_server.links,
|
|
fk_server.name,
|
|
fk_server.network_info,
|
|
fk_server.updated_at,
|
|
fk_server.uuid)
|
|
self.assertEqual(expected_data, data)
|
|
|
|
def test_server_create(self, mock_create, mock_find):
|
|
name = 'server1'
|
|
flavor_id = 'flavor-id-' + uuid.uuid4().hex
|
|
image_id = 'image-id-' + uuid.uuid4().hex
|
|
networks = [{'net-id': 'net-id-' + uuid.uuid4().hex}]
|
|
self._test_create_fake_server(mock_create, mock_find,
|
|
name, flavor_id, image_id, networks)
|
|
|
|
def test_server_create_with_description(self, mock_create, mock_find):
|
|
name = 'server1'
|
|
flavor_id = 'flavor-id-' + uuid.uuid4().hex
|
|
image_id = 'image-id-' + uuid.uuid4().hex
|
|
networks = [{'net-id': 'net-id-' + uuid.uuid4().hex}]
|
|
description = 'fake_description'
|
|
self._test_create_fake_server(mock_create, mock_find,
|
|
name, flavor_id, image_id,
|
|
networks, description)
|
|
|
|
def test_server_create_with_az(self, mock_create, mock_find):
|
|
name = 'server1'
|
|
flavor_id = 'flavor-id-' + uuid.uuid4().hex
|
|
image_id = 'image-id-' + uuid.uuid4().hex
|
|
networks = [{'net-id': 'net-id-' + uuid.uuid4().hex}]
|
|
fake_az = 'fake_availability_zone'
|
|
self._test_create_fake_server(mock_create, mock_find,
|
|
name, flavor_id, image_id,
|
|
networks, availability_zone=fake_az)
|
|
|
|
def test_server_create_with_port_type(self, mock_create, mock_find):
|
|
name = 'server1'
|
|
flavor_id = 'flavor-id-' + uuid.uuid4().hex
|
|
image_id = 'image-id-' + uuid.uuid4().hex
|
|
networks = [{'net-id': 'net-id-' + uuid.uuid4().hex,
|
|
'port-type': 'normal'}]
|
|
self._test_create_fake_server(mock_create, mock_find,
|
|
name, flavor_id, image_id,
|
|
networks)
|
|
|
|
def test_server_create_with_extra(self, mock_create, mock_find):
|
|
name = 'server1'
|
|
flavor_id = 'flavor-id-' + uuid.uuid4().hex
|
|
image_id = 'image-id-' + uuid.uuid4().hex
|
|
networks = [{'net-id': 'net-id-' + uuid.uuid4().hex}]
|
|
extra_info = 'key1=test'
|
|
self._test_create_fake_server(mock_create, mock_find,
|
|
name, flavor_id, image_id,
|
|
networks, extra=extra_info)
|
|
|
|
|
|
@mock.patch.object(utils, 'find_resource')
|
|
@mock.patch.object(server_mgr.ServerManager, '_update')
|
|
class TestServerUpdate(test_base.TestBaremetalComputeV1):
|
|
def setUp(self):
|
|
super(TestServerUpdate, self).setUp()
|
|
self.cmd = server.UpdateServer(self.app, None)
|
|
self.fake_server = fakes.FakeServer.create_one_server()
|
|
|
|
def test_server_update_description(self, mock_update, mock_find):
|
|
mock_find.return_value = self.fake_server
|
|
arglist = [
|
|
'--description', 'test_description',
|
|
self.fake_server.uuid]
|
|
verifylist = [
|
|
('server', self.fake_server.uuid),
|
|
('description', 'test_description')]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
self.cmd.take_action(parsed_args)
|
|
mock_update.assert_called_with(
|
|
'/instances/%s' % self.fake_server.uuid,
|
|
data=[{'path': '/description',
|
|
'value': 'test_description',
|
|
'op': 'replace'}])
|
|
|
|
def test_server_update_add_extra(self, mock_update, mock_find):
|
|
mock_find.return_value = self.fake_server
|
|
arglist = [
|
|
'--add-extra', 'extra_key:extra_value',
|
|
self.fake_server.uuid]
|
|
verifylist = [
|
|
('server', self.fake_server.uuid),
|
|
('add_extra', [('extra_key', 'extra_value')])]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
self.cmd.take_action(parsed_args)
|
|
mock_update.assert_called_with(
|
|
'/instances/%s' % self.fake_server.uuid,
|
|
data=[{'path': '/extra/extra_key',
|
|
'value': 'extra_value',
|
|
'op': 'add'}])
|
|
|
|
def test_server_update_add_replace_remove_multi_extra(
|
|
self, mock_update, mock_find):
|
|
mock_find.return_value = self.fake_server
|
|
arglist = [
|
|
'--add-extra', 'add_key1:add_value1',
|
|
'--add-extra', 'add_key2:add_value2',
|
|
'--replace-extra', 'replace_key1:replace_value1',
|
|
'--replace-extra', 'replace_key2:replace_value2',
|
|
'--remove-extra', 'remove_key1',
|
|
'--remove-extra', 'remove_key2',
|
|
self.fake_server.uuid]
|
|
verifylist = [
|
|
('server', self.fake_server.uuid),
|
|
('add_extra', [('add_key1', 'add_value1'),
|
|
('add_key2', 'add_value2')]),
|
|
('replace_extra', [('replace_key1', 'replace_value1'),
|
|
('replace_key2', 'replace_value2')]),
|
|
('remove_extra', ['remove_key1', 'remove_key2'])]
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
self.cmd.take_action(parsed_args)
|
|
mock_update.assert_called_with(
|
|
'/instances/%s' % self.fake_server.uuid,
|
|
data=[{'path': '/extra/add_key1',
|
|
'value': 'add_value1',
|
|
'op': 'add'},
|
|
{'path': '/extra/add_key2',
|
|
'value': 'add_value2',
|
|
'op': 'add'},
|
|
{'path': '/extra/replace_key1',
|
|
'value': 'replace_value1',
|
|
'op': 'replace'},
|
|
{'path': '/extra/replace_key2',
|
|
'value': 'replace_value2',
|
|
'op': 'replace'},
|
|
{'path': '/extra/remove_key1',
|
|
'op': 'remove'},
|
|
{'path': '/extra/remove_key2',
|
|
'op': 'remove'}])
|
|
|
|
|
|
@mock.patch.object(server_mgr.ServerManager, '_list')
|
|
class TestServerList(test_base.TestBaremetalComputeV1):
|
|
def setUp(self):
|
|
super(TestServerList, self).setUp()
|
|
self.cmd = server.ListServer(self.app, None)
|
|
fake_return_net = {
|
|
"12cffc4a-b845-409e-b589-7c84be4b10d9": {
|
|
"fixed_ips": [
|
|
{
|
|
"ip_address": "172.24.4.4",
|
|
"subnet_id": "a9d47430-f90b-4513-af5f-6315af54de7d"
|
|
},
|
|
{
|
|
"ip_address": "2001:db8::a",
|
|
"subnet_id": "5e7b3e2d-f36f-4e30-874c-16c2d126fe53"
|
|
}
|
|
],
|
|
"mac_address": "52:54:00:6c:c4:17",
|
|
"network": "ade2b658-929b-439f-9528-c47057960942"
|
|
}
|
|
}
|
|
self.fake_servers = fakes.FakeServer.create_servers(
|
|
attrs={'status': 'active'}, count=3)
|
|
for s in self.fake_servers:
|
|
setattr(s, 'network_info', fake_return_net)
|
|
self.list_columns = (
|
|
"UUID",
|
|
"Name",
|
|
"Status",
|
|
)
|
|
|
|
self.list_columns_long = (
|
|
"UUID",
|
|
"Name",
|
|
"Flavor",
|
|
"Status",
|
|
"Image",
|
|
"Description",
|
|
"Availability Zone",
|
|
"Networks"
|
|
)
|
|
|
|
self.list_data = tuple((
|
|
self.fake_servers[i].uuid,
|
|
self.fake_servers[i].name,
|
|
self.fake_servers[i].status,
|
|
) for i in range(3))
|
|
|
|
self.list_data_long = tuple((
|
|
self.fake_servers[i].uuid,
|
|
self.fake_servers[i].name,
|
|
self.fake_servers[i].instance_type_uuid,
|
|
self.fake_servers[i].status,
|
|
self.fake_servers[i].image_uuid,
|
|
self.fake_servers[i].description,
|
|
self.fake_servers[i].availability_zone,
|
|
'172.24.4.4, 2001:db8::a'
|
|
) for i in range(3))
|
|
|
|
def test_server_list(self, mock_list):
|
|
arglist = []
|
|
verifylist = []
|
|
mock_list.return_value = self.fake_servers
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
columns, data = self.cmd.take_action(parsed_args)
|
|
mock_list.assert_called_once_with('/instances',
|
|
response_key='instances')
|
|
self.assertEqual(self.list_columns, columns)
|
|
self.assertEqual(self.list_data, tuple(data))
|
|
|
|
def test_server_list_long(self, mock_list):
|
|
arglist = [
|
|
'--long',
|
|
]
|
|
verifylist = [
|
|
('long', True),
|
|
]
|
|
mock_list.return_value = self.fake_servers
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
columns, data = self.cmd.take_action(parsed_args)
|
|
mock_list.assert_called_once_with('/instances/detail',
|
|
response_key='instances')
|
|
self.assertEqual(self.list_columns_long, columns)
|
|
self.assertEqual(self.list_data_long, tuple(data))
|
|
|
|
def test_server_list_with_all_projects(self, mock_list):
|
|
arglist = [
|
|
'--all-projects',
|
|
]
|
|
verifylist = [
|
|
('all_projects', True),
|
|
]
|
|
mock_list.return_value = self.fake_servers
|
|
parsed_args = self.check_parser(self.cmd, arglist, verifylist)
|
|
columns, data = self.cmd.take_action(parsed_args)
|
|
mock_list.assert_called_once_with('/instances?all_tenants=True',
|
|
response_key='instances')
|
|
self.assertEqual(self.list_columns, columns)
|
|
self.assertEqual(self.list_data, tuple(data))
|
|
|
|
|
|
@mock.patch.object(utils, 'find_resource')
|
|
@mock.patch.object(server_mgr.ServerManager, '_delete')
|
|
class TestServerDelete(test_base.TestBaremetalComputeV1):
|
|
def setUp(self):
|
|
super(TestServerDelete, self).setUp()
|
|
self.cmd = server.DeleteServer(self.app, None)
|
|
|
|
def test_server_delete_one(self, mock_delete, mock_find):
|
|
fake_server = fakes.FakeServer.create_one_server()
|
|
mock_find.return_value = fake_server
|
|
args = ['server1']
|
|
verify_args = [('server', ['server1'])]
|
|
parsed_args = self.check_parser(self.cmd, args, verify_args)
|
|
self.cmd.take_action(parsed_args)
|
|
mock_delete.assert_called_with('/instances/%s' % fake_server.uuid)
|
|
|
|
def test_server_delete_more_than_one(self, mock_delete, mock_find):
|
|
fake_servers = fakes.FakeServer.create_servers(count=3)
|
|
mock_find.side_effect = fake_servers
|
|
args = [s.name for s in fake_servers]
|
|
verify_args = [('server', [s.name for s in fake_servers])]
|
|
parsed_args = self.check_parser(self.cmd, args, verify_args)
|
|
self.cmd.take_action(parsed_args)
|
|
expected = [mock.call('/instances/%s' % s.uuid) for s in fake_servers]
|
|
self.assertEqual(expected, mock_delete.call_args_list)
|
|
|
|
def test_server_delete_more_than_one_partly_failed(
|
|
self, mock_delete, mock_find):
|
|
fake_servers = fakes.FakeServer.create_servers(count=3)
|
|
mock_find.side_effect = fake_servers
|
|
args = [s.name for s in fake_servers]
|
|
verify_args = [('server', [s.name for s in fake_servers])]
|
|
parsed_args = self.check_parser(self.cmd, args, verify_args)
|
|
mock_delete.side_effect = [mock.Mock(), Exception(), mock.Mock()]
|
|
exc = self.assertRaises(exceptions.CommandError,
|
|
self.cmd.take_action, parsed_args)
|
|
self.assertEqual('1 of 3 baremetal servers failed to delete.',
|
|
str(exc))
|
|
|
|
|
|
@mock.patch.object(server_mgr.ServerManager, '_get')
|
|
class TestServerShow(test_base.TestBaremetalComputeV1):
|
|
def setUp(self):
|
|
super(TestServerShow, self).setUp()
|
|
self.cmd = server.ShowServer(self.app, None)
|
|
self.fake_server = fakes.FakeServer.create_one_server()
|
|
|
|
def test_server_show_with_uuid_specified(self, mock_get):
|
|
args = [self.fake_server.uuid]
|
|
verify_args = [('server', self.fake_server.uuid)]
|
|
parsed_args = self.check_parser(self.cmd, args, verify_args)
|
|
self.cmd.take_action(parsed_args)
|
|
mock_get.assert_called_once_with(
|
|
'/instances/%s' % self.fake_server.uuid)
|
|
|
|
@mock.patch.object(server_mgr.ServerManager, 'list')
|
|
def test_server_show_with_name_specified(self, mock_list, mock_get):
|
|
args = [self.fake_server.name]
|
|
verify_args = [('server', self.fake_server.name)]
|
|
parsed_args = self.check_parser(self.cmd, args, verify_args)
|
|
mock_get.side_effect = [exceptions.NotFound(404),
|
|
exceptions.NotFound(404),
|
|
self.fake_server]
|
|
mock_list.return_value = [self.fake_server]
|
|
self.cmd.take_action(parsed_args)
|
|
expected = [mock.call('/instances/%s' % self.fake_server.name),
|
|
mock.call('/instances/%s' % self.fake_server.name),
|
|
mock.call('/instances/%s' % self.fake_server.uuid)]
|
|
self.assertEqual(expected, mock_get.call_args_list)
|
|
|
|
|
|
class TestServerPowerActionBase(test_base.TestBaremetalComputeV1):
|
|
def setUp(self):
|
|
super(TestServerPowerActionBase, self).setUp()
|
|
self.action = None
|
|
self.action_name = None
|
|
|
|
def _test_server_power_action_one(self, mock_update_all, mock_find):
|
|
fake_server = fakes.FakeServer.create_one_server()
|
|
mock_find.return_value = fake_server
|
|
args = [fake_server.uuid]
|
|
verify_args = [('server', [fake_server.uuid])]
|
|
parsed_args = self.check_parser(self.cmd, args, verify_args)
|
|
self.cmd.take_action(parsed_args)
|
|
mock_update_all.assert_called_with(
|
|
'/instances/%s/states/power' % fake_server.uuid,
|
|
data={'target': self.action})
|
|
|
|
def _test_server_power_action_multiple(self, mock_update_all,
|
|
mock_find):
|
|
fake_servers = fakes.FakeServer.create_servers(count=3)
|
|
mock_find.side_effect = fake_servers
|
|
args = [s.name for s in fake_servers]
|
|
verify_args = [('server', [s.name for s in fake_servers])]
|
|
parsed_args = self.check_parser(self.cmd, args, verify_args)
|
|
self.cmd.take_action(parsed_args)
|
|
expected = [mock.call(
|
|
'/instances/%s/states/power' % s.uuid,
|
|
data={'target': self.action}) for s in fake_servers]
|
|
self.assertEqual(expected, mock_update_all.call_args_list)
|
|
|
|
def _test_server_delete_more_than_one_partly_failed(
|
|
self, mock_update_all, mock_find):
|
|
fake_servers = fakes.FakeServer.create_servers(count=3)
|
|
mock_find.side_effect = fake_servers
|
|
args = [s.name for s in fake_servers]
|
|
verify_args = [('server', [s.name for s in fake_servers])]
|
|
parsed_args = self.check_parser(self.cmd, args, verify_args)
|
|
mock_update_all.side_effect = [mock.Mock(), Exception(), mock.Mock()]
|
|
exc = self.assertRaises(exceptions.CommandError,
|
|
self.cmd.take_action, parsed_args)
|
|
self.assertEqual(
|
|
'1 of 3 baremetal servers failed to %s.' % self.action_name,
|
|
str(exc))
|
|
|
|
|
|
@mock.patch.object(utils, 'find_resource')
|
|
@mock.patch.object(server_mgr.ServerManager, '_update_all')
|
|
class TestServerStart(TestServerPowerActionBase):
|
|
def setUp(self):
|
|
super(TestServerStart, self).setUp()
|
|
self.cmd = server.StartServer(self.app, None)
|
|
self.action = 'on'
|
|
self.action_name = 'start'
|
|
|
|
def test_server_start_one(self, mock_update_all, mock_find):
|
|
self._test_server_power_action_one(mock_update_all, mock_find)
|
|
|
|
def test_server_start_multiple(self, mock_update_all, mock_find):
|
|
self._test_server_power_action_multiple(mock_update_all, mock_find)
|
|
|
|
def test_server_start_multiple_partly_failed(self,
|
|
mock_update_all, mock_find):
|
|
self._test_server_delete_more_than_one_partly_failed(
|
|
mock_update_all, mock_find)
|
|
|
|
|
|
@mock.patch.object(utils, 'find_resource')
|
|
@mock.patch.object(server_mgr.ServerManager, '_update_all')
|
|
class TestServerStop(TestServerPowerActionBase):
|
|
def setUp(self):
|
|
super(TestServerStop, self).setUp()
|
|
self.cmd = server.StopServer(self.app, None)
|
|
self.action = 'off'
|
|
self.action_name = 'stop'
|
|
|
|
def test_server_stop_one(self, mock_update_all, mock_find):
|
|
self._test_server_power_action_one(mock_update_all, mock_find)
|
|
|
|
def test_server_stop_multiple(self, mock_update_all, mock_find):
|
|
self._test_server_power_action_multiple(mock_update_all, mock_find)
|
|
|
|
def test_server_stop_multiple_partly_failed(self,
|
|
mock_update_all, mock_find):
|
|
self._test_server_delete_more_than_one_partly_failed(
|
|
mock_update_all, mock_find)
|
|
|
|
|
|
@mock.patch.object(utils, 'find_resource')
|
|
@mock.patch.object(server_mgr.ServerManager, '_update_all')
|
|
class TestServerReboot(TestServerPowerActionBase):
|
|
def setUp(self):
|
|
super(TestServerReboot, self).setUp()
|
|
self.cmd = server.RebootServer(self.app, None)
|
|
self.action = 'reboot'
|
|
self.action_name = 'reboot'
|
|
|
|
def test_server_reboot_one(self, mock_update_all, mock_find):
|
|
self._test_server_power_action_one(mock_update_all, mock_find)
|
|
|
|
def test_server_reboot_multiple(self, mock_update_all, mock_find):
|
|
self._test_server_power_action_multiple(mock_update_all, mock_find)
|
|
|
|
def test_server_reboot_multiple_partly_failed(self,
|
|
mock_update_all, mock_find):
|
|
self._test_server_delete_more_than_one_partly_failed(
|
|
mock_update_all, mock_find)
|