3f79951a11
cloud-init writes a result.json when finished[1]. When wait_for_cloud_init=true is provided, this change uses result.json to detect when cloud-init has finished, and raises an exception if it finished with errors. [1] http://bazaar.launchpad.net/~cloud-init-dev/cloud-init/trunk/view/head:/doc/status.txt Change-Id: I7e0bcce733b008b488c8e1cf9bb74b5b6192cfbc
285 lines
11 KiB
Python
285 lines
11 KiB
Python
# Copyright 2013: Mirantis 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.
|
|
|
|
"""Tests for OpenStack VM provider."""
|
|
|
|
import textwrap
|
|
|
|
import jsonschema
|
|
import mock
|
|
from oslotest import mockpatch
|
|
|
|
from rally.deployment.serverprovider.providers import openstack as provider
|
|
from rally import exceptions
|
|
from tests.unit import fakes
|
|
from tests.unit import test
|
|
|
|
MOD_NAME = "rally.deployment.serverprovider.providers.openstack"
|
|
OSProvider = provider.OpenStackProvider
|
|
|
|
|
|
class FakeOSClients(object):
|
|
|
|
def nova(self):
|
|
return "nova"
|
|
|
|
def glance(self):
|
|
return "glance"
|
|
|
|
|
|
class OpenStackProviderTestCase(test.TestCase):
|
|
|
|
def setUp(self):
|
|
super(OpenStackProviderTestCase, self).setUp()
|
|
self.useFixture(mockpatch.Patch(
|
|
"rally.deployment.serverprovider.provider.ResourceManager"))
|
|
|
|
def _get_valid_config(self):
|
|
return {
|
|
"image": {
|
|
"url": "http://example.net/img.qcow2",
|
|
"format": "qcow2",
|
|
"name": "Image",
|
|
"checksum": "0123456789abcdef",
|
|
},
|
|
"deployment_name": "rally-dep-1",
|
|
"auth_url": "urlto",
|
|
"user": "name",
|
|
"password": "mypass",
|
|
"tenant": "tenant",
|
|
"flavor_id": "22"}
|
|
|
|
def _init_mock_clients(self):
|
|
self.clients = mock.MagicMock()
|
|
|
|
self.image = mock.MagicMock()
|
|
self.image.checksum = "0123456789abcdef"
|
|
self.image.get = mock.MagicMock(return_value=self.image)
|
|
self.image.id = "fake-uuid"
|
|
self.glance_client = mock.Mock(return_value=self.image)
|
|
self.glance_client.images.create = mock.Mock(return_value=self.image)
|
|
self.glance_client.images.list = mock.Mock(return_value=[self.image])
|
|
self.clients.glance = mock.Mock(return_value=self.glance_client)
|
|
|
|
self.instance = mock.MagicMock()
|
|
self.instance.status = "ACTIVE"
|
|
|
|
self.nova_client = mock.MagicMock()
|
|
self.nova_client.servers.create = mock.MagicMock(
|
|
return_value=self.instance)
|
|
|
|
self.clients.nova = mock.MagicMock(return_value=self.nova_client)
|
|
|
|
@mock.patch(
|
|
"rally.deployment.serverprovider.providers.openstack.osclients")
|
|
def test_openstack_provider_init(self, mock_osclients):
|
|
cfg = self._get_valid_config()
|
|
mock_osclients.Clients = mock.MagicMock(return_value=FakeOSClients())
|
|
os_provider = OSProvider(mock.MagicMock(), cfg)
|
|
self.assertEqual("nova", os_provider.nova)
|
|
self.assertEqual("glance", os_provider.glance)
|
|
|
|
@mock.patch("rally.osclients.Clients")
|
|
def test_init_no_glance(self, mock_clients):
|
|
mock_clients.return_value.glance.side_effect = KeyError("image")
|
|
cfg = self._get_valid_config()
|
|
provider = OSProvider(mock.MagicMock(), cfg)
|
|
self.assertIsNone(provider.glance)
|
|
|
|
@mock.patch(
|
|
"rally.deployment.serverprovider.providers.openstack.osclients")
|
|
def test_openstack_provider_init_with_invalid_conf_no_user(self,
|
|
mock_osclients):
|
|
cfg = self._get_valid_config()
|
|
cfg.pop("user")
|
|
self.assertRaises(jsonschema.ValidationError, OSProvider,
|
|
mock.MagicMock(), cfg)
|
|
|
|
@mock.patch(
|
|
"rally.deployment.serverprovider.providers.openstack.osclients")
|
|
def test_openstack_provider_init_with_invalid_conf_no_url(self,
|
|
mock_osclients):
|
|
cfg = self._get_valid_config()
|
|
del cfg["image"]["url"]
|
|
del cfg["image"]["checksum"]
|
|
self.assertRaises(jsonschema.ValidationError, OSProvider,
|
|
mock.MagicMock(), cfg)
|
|
|
|
@mock.patch(
|
|
"rally.deployment.serverprovider.providers.openstack.osclients")
|
|
def test_openstack_provider_init_with_invalid_conf_extra_key(
|
|
self, mock_osclients):
|
|
cfg = self._get_valid_config()
|
|
cfg["aaaaa"] = "bbbbb"
|
|
self.assertRaises(jsonschema.ValidationError, OSProvider,
|
|
mock.MagicMock(), cfg)
|
|
|
|
@mock.patch(
|
|
"rally.deployment.serverprovider.providers.openstack.osclients")
|
|
def test_openstack_provider_init_with_invalid_conf_flavor_(self,
|
|
mock_osclients):
|
|
cfg = self._get_valid_config()["user"] = 1111
|
|
self.assertRaises(jsonschema.ValidationError, OSProvider,
|
|
mock.MagicMock(), cfg)
|
|
|
|
@mock.patch(
|
|
"rally.deployment.serverprovider.providers.openstack.osclients")
|
|
def test_openstack_provider_with_valid_config(self,
|
|
mock_osclients):
|
|
cfg = self._get_valid_config()
|
|
OSProvider(mock.MagicMock(), cfg)
|
|
|
|
@mock.patch(
|
|
"rally.deployment.serverprovider.providers.openstack.osclients")
|
|
def test_openstack_provider_with_valid_config_uuid(self, mock_osclients):
|
|
cfg = self._get_valid_config()
|
|
cfg["image"] = dict(uuid="289D7A51-1A0C-43C4-800D-706EA8A3CDF3")
|
|
OSProvider(mock.MagicMock(), cfg)
|
|
|
|
@mock.patch(
|
|
"rally.deployment.serverprovider.providers.openstack.osclients")
|
|
def test_openstack_provider_with_valid_config_checksum(self,
|
|
mock_osclients):
|
|
cfg = self._get_valid_config()
|
|
cfg["image"] = dict(checksum="checksum")
|
|
OSProvider(mock.MagicMock(), cfg)
|
|
|
|
def test_cloud_init_success_notready(self):
|
|
fake_server = mock.Mock()
|
|
fake_server.ssh.execute.return_value = (1, "", "")
|
|
|
|
# Not ready yet -> False
|
|
self.assertFalse(provider._cloud_init_success(fake_server))
|
|
|
|
def test_cloud_init_success_completed(self):
|
|
fake_server = mock.Mock()
|
|
result_json_text = textwrap.dedent("""
|
|
{
|
|
"v1": {
|
|
"errors": [],
|
|
"datasource": "DataSourceFoo"
|
|
}
|
|
}
|
|
""")
|
|
fake_server.ssh.execute.return_value = (0, result_json_text, "")
|
|
# Completed (with no errors) -> True
|
|
self.assertTrue(provider._cloud_init_success(fake_server))
|
|
|
|
def test_cloud_init_success_errors(self):
|
|
fake_server = mock.Mock()
|
|
result_json_text = textwrap.dedent("""
|
|
{
|
|
"v1": {
|
|
"errors": ["omg!"],
|
|
"datasource": "DataSourceFoo"
|
|
}
|
|
}
|
|
""")
|
|
fake_server.ssh.execute.return_value = (0, result_json_text, "")
|
|
# Completed with errors -> Exception
|
|
self.assertRaises(RuntimeError,
|
|
provider._cloud_init_success, fake_server)
|
|
|
|
@mock.patch("time.sleep")
|
|
@mock.patch(MOD_NAME + ".provider.Server")
|
|
@mock.patch(MOD_NAME + ".osclients")
|
|
@mock.patch(MOD_NAME + ".utils")
|
|
def test_create_servers(self, mock_utils, mock_osclients,
|
|
mock_server, mock_sleep):
|
|
fake_keypair = mock.Mock()
|
|
fake_keypair.name = "fake_key_name"
|
|
provider = OSProvider(mock.Mock(), self._get_valid_config())
|
|
provider.nova = mock.Mock()
|
|
provider.get_image_uuid = mock.Mock(return_value="fake_image_uuid")
|
|
provider.get_userdata = mock.Mock(return_value="fake_userdata")
|
|
provider.get_nics = mock.Mock(return_value="fake_nics")
|
|
provider.create_keypair = mock.Mock(return_value=(fake_keypair,
|
|
"fake_path"))
|
|
mock_utils.wait_for = lambda x, **kw: x
|
|
mock_server.return_value = fake_server = mock.Mock()
|
|
provider.nova.servers.create.return_value = fake_instance = mock.Mock()
|
|
fake_instance.accessIPv4 = None
|
|
fake_instance.accessIPv6 = None
|
|
fake_instance.addresses = {"private": [{"addr": "1.2.3.4"}]}
|
|
|
|
servers = provider.create_servers()
|
|
|
|
mock_server.assert_called_once_with(host="1.2.3.4", user="root",
|
|
key="fake_path")
|
|
self.assertEqual([fake_server], servers)
|
|
fake_server.ssh.wait.assert_called_once_with(interval=5, timeout=120)
|
|
provider.nova.servers.create.assert_called_once_with(
|
|
"rally-dep-1-0", "fake_image_uuid", "22", userdata="fake_userdata",
|
|
nics="fake_nics", key_name="fake_key_name", config_drive=False)
|
|
|
|
@mock.patch(MOD_NAME + ".osclients")
|
|
def test_get_image_found_by_checksum(self, mock_osclients):
|
|
self._init_mock_clients()
|
|
mock_osclients.Clients = mock.MagicMock(return_value=self.clients)
|
|
prov = OSProvider(mock.MagicMock(), self._get_valid_config())
|
|
image_uuid = prov.get_image_uuid()
|
|
self.assertEqual(image_uuid, "fake-uuid")
|
|
|
|
@mock.patch(MOD_NAME + ".osclients")
|
|
def test_get_image_download(self, mock_osclients):
|
|
self._init_mock_clients()
|
|
self.glance_client.images.list = mock.Mock(return_value=[])
|
|
mock_osclients.Clients = mock.MagicMock(return_value=self.clients)
|
|
prov = OSProvider(mock.MagicMock(), self._get_valid_config())
|
|
image_uuid = prov.get_image_uuid()
|
|
self.assertEqual(image_uuid, "fake-uuid")
|
|
|
|
@mock.patch(MOD_NAME + ".osclients")
|
|
def test_get_image_no_glance_exception(
|
|
self, mock_osclients):
|
|
prov = OSProvider(mock.MagicMock(), self._get_valid_config())
|
|
prov.glance = None
|
|
self.assertRaises(exceptions.InvalidConfigException,
|
|
prov.get_image_uuid)
|
|
|
|
@mock.patch(MOD_NAME + ".osclients")
|
|
def test_get_image_from_uuid_no_glance(self, mock_osclients):
|
|
conf = self._get_valid_config()
|
|
conf["image"]["uuid"] = "EC7A1DB7-C5BD-49A2-8066-613809CB22F5"
|
|
prov = OSProvider(mock.MagicMock(), conf)
|
|
prov.glance = True
|
|
self.assertEqual(conf["image"]["uuid"], prov.get_image_uuid())
|
|
|
|
@mock.patch(MOD_NAME + ".osclients")
|
|
def test_destroy_servers(self, mock_osclients):
|
|
prov = OSProvider(mock.MagicMock(), self._get_valid_config())
|
|
prov.resources.get_all.side_effect = [
|
|
[fakes.FakeResource(
|
|
id=1,
|
|
items={"info": {"id": "35FC0503-FED6-419F-B6EE-B704198CE642"}}
|
|
)],
|
|
[fakes.FakeResource(
|
|
id=2,
|
|
items={"info": {"id": "keypair_name"}}
|
|
)],
|
|
]
|
|
prov.destroy_servers()
|
|
prov.resources.get_all.assert_has_calls([
|
|
mock.call(type="server"),
|
|
mock.call(type="keypair"),
|
|
])
|
|
prov.nova.servers.delete.assert_called_once_with(
|
|
"35FC0503-FED6-419F-B6EE-B704198CE642")
|
|
prov.nova.keypairs.delete.assert_called_once_with("keypair_name")
|
|
prov.resources.delete.assert_has_calls([
|
|
mock.call(1),
|
|
mock.call(2),
|
|
])
|