d46c3aa023
To pull nova-baremetal node registration into line with the previous change for Ironic, switch to novaclient with the baremetal extension to register nodes. Change-Id: I13e6ae61defa3194ea406c97c36be9633abfd317
113 lines
5.2 KiB
Python
113 lines
5.2 KiB
Python
# Copyright (c) 2014 Hewlett-Packard Development Company, L.P.
|
|
#
|
|
# 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 collections
|
|
|
|
import mock
|
|
|
|
from ironicclient.openstack.common.apiclient import exceptions as ironicexp
|
|
from os_cloud_config import nodes
|
|
from os_cloud_config.tests import base
|
|
|
|
|
|
class NodesTest(base.TestCase):
|
|
|
|
def _get_node(self):
|
|
return {'cpu': '1', 'memory': '2048', 'disk': '30', 'arch': 'amd64',
|
|
'mac': ['aaa'], 'pm_addr': 'foo.bar', 'pm_user': 'test',
|
|
'pm_password': 'random', 'pm_type': 'pxe_ssh'}
|
|
|
|
def test_check_output(self):
|
|
self.assertEqual("/dev/null\n",
|
|
nodes._check_output(["ls", "/dev/null"]))
|
|
|
|
@mock.patch('os.environ')
|
|
@mock.patch('novaclient.v1_1.client.Client')
|
|
def test_get_nova_bm_client(self, client_mock, environ):
|
|
nodes._get_nova_bm_client()
|
|
client_mock.assert_called_once_with(environ["OS_USERNAME"],
|
|
environ["OS_PASSWORD"],
|
|
environ["OS_AUTH_URL"],
|
|
environ["OS_TENANT_NAME"],
|
|
extensions=[mock.ANY])
|
|
|
|
@mock.patch('os_cloud_config.nodes.using_ironic', return_value=False)
|
|
def test_register_all_nodes_nova_bm(self, ironic_mock):
|
|
node_list = [self._get_node(), self._get_node()]
|
|
node_list[0]["mac"].append("bbb")
|
|
client = mock.MagicMock()
|
|
nodes.register_all_nodes('servicehost', node_list, client=client)
|
|
nova_bm_call = mock.call(
|
|
"servicehost", "1", "2048", "30", "aaa", pm_address="foo.bar",
|
|
pm_user="test", pm_password="random")
|
|
client.baremetal.create.has_calls([nova_bm_call, nova_bm_call])
|
|
client.baremetal.add_interface.assert_called_once_with(mock.ANY, "bbb")
|
|
|
|
@mock.patch('os.environ')
|
|
@mock.patch('ironicclient.client.get_client')
|
|
def test_get_ironic_client(self, client_mock, environ):
|
|
nodes._get_ironic_client()
|
|
client_mock.assert_called_once_with(
|
|
1, os_username=environ["OS_USERNAME"],
|
|
os_password=environ["OS_PASSWORD"],
|
|
os_auth_url=environ["OS_AUTH_URL"],
|
|
os_tenant_name=environ["OS_TENANT_NAME"])
|
|
|
|
@mock.patch('os_cloud_config.nodes.using_ironic', return_value=True)
|
|
def test_register_all_nodes_ironic(self, using_ironic):
|
|
node_list = [self._get_node(), self._get_node()]
|
|
node_list[1]["pm_type"] = "ipmi"
|
|
node_properties = {"cpus": "1",
|
|
"memory_mb": "2048",
|
|
"local_gb": "30",
|
|
"cpu_arch": "amd64"}
|
|
ironic = mock.MagicMock()
|
|
nodes.register_all_nodes('servicehost', node_list, client=ironic)
|
|
ssh_key = "/mnt/state/var/lib/ironic/virtual-power-key"
|
|
pxe_node_driver_info = {"ssh_address": "foo.bar",
|
|
"ssh_username": "test",
|
|
"ssh_key_filename": ssh_key,
|
|
"ssh_virt_type": "virsh"}
|
|
ipmi_node_driver_info = {"ipmi_address": "foo.bar",
|
|
"ipmi_username": "test",
|
|
"ipmi_password": "random"}
|
|
pxe_node = mock.call(driver="pxe_ssh",
|
|
driver_info=pxe_node_driver_info,
|
|
properties=node_properties)
|
|
port_call = mock.call(node_uuid=ironic.node.create.return_value.uuid,
|
|
address='aaa')
|
|
power_off_call = mock.call(ironic.node.create.return_value.uuid, 'off')
|
|
ipmi_node = mock.call(driver="ipmi",
|
|
driver_info=ipmi_node_driver_info,
|
|
properties=node_properties)
|
|
ironic.node.create.assert_has_calls([pxe_node, ipmi_node])
|
|
ironic.port.create.assert_has_calls([port_call, port_call])
|
|
ironic.node.set_power_state.assert_has_calls(
|
|
[power_off_call, power_off_call])
|
|
|
|
@mock.patch('time.sleep')
|
|
def test_register_ironic_node_retry(self, sleep):
|
|
ironic = mock.MagicMock()
|
|
ironic_node = collections.namedtuple('node', ['uuid'])
|
|
side_effect = (ironicexp.ConnectionRefused,
|
|
ironicexp.ServiceUnavailable, ironic_node('1'))
|
|
ironic.node.create.side_effect = side_effect
|
|
nodes.register_ironic_node(None, self._get_node(), client=ironic)
|
|
sleep.assert_has_calls([mock.call(10), mock.call(10)])
|
|
node_create = mock.call(driver='pxe_ssh',
|
|
driver_info=mock.ANY,
|
|
properties=mock.ANY)
|
|
ironic.node.create.assert_has_calls(node_create)
|