434e8173fa
This adds five persistent fixtures for Scenario test cases; previously they had been variously mocked, patched, etc., in different ways individually in many test cases. It improves the way those fixtures are used, mostly by using the .return_value attribute of Mock objects instead of calling the Mock objects in the test cases and dealing with strange side effects of that. It also adds assertions for get_from_manager() calls. Change-Id: I81a69683ef8ca0fc00bb408c951ec2f9a98e7ba0
111 lines
3.9 KiB
Python
111 lines
3.9 KiB
Python
# Copyright 2014: 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.
|
|
|
|
import mock
|
|
|
|
from rally.plugins.openstack.scenarios.sahara import clusters
|
|
from tests.unit import test
|
|
|
|
SAHARA_CLUSTERS = ("rally.plugins.openstack.scenarios.sahara.clusters"
|
|
".SaharaClusters")
|
|
SAHARA_UTILS = "rally.plugins.openstack.scenarios.sahara.utils"
|
|
|
|
|
|
class SaharaClustersTestCase(test.ScenarioTestCase):
|
|
|
|
@mock.patch(SAHARA_CLUSTERS + "._delete_cluster")
|
|
@mock.patch(SAHARA_CLUSTERS + "._launch_cluster",
|
|
return_value=mock.MagicMock(id=42))
|
|
def test_create_and_delete_cluster(
|
|
self, mock__launch_cluster, mock__delete_cluster):
|
|
clusters_scenario = clusters.SaharaClusters()
|
|
|
|
clusters_scenario.context = {
|
|
"tenant": {
|
|
"sahara_image": "test_image"
|
|
}
|
|
}
|
|
clusters_scenario.create_and_delete_cluster(
|
|
flavor="test_flavor",
|
|
workers_count=5,
|
|
plugin_name="test_plugin",
|
|
hadoop_version="test_version")
|
|
|
|
mock__launch_cluster.assert_called_once_with(
|
|
flavor_id="test_flavor",
|
|
image_id="test_image",
|
|
workers_count=5,
|
|
plugin_name="test_plugin",
|
|
hadoop_version="test_version",
|
|
floating_ip_pool=None,
|
|
volumes_per_node=None,
|
|
volumes_size=None,
|
|
auto_security_group=None,
|
|
security_groups=None,
|
|
node_configs=None,
|
|
cluster_configs=None,
|
|
enable_anti_affinity=False)
|
|
|
|
mock__delete_cluster.assert_called_once_with(
|
|
mock__launch_cluster.return_value)
|
|
|
|
@mock.patch(SAHARA_CLUSTERS + "._delete_cluster")
|
|
@mock.patch(SAHARA_CLUSTERS + "._scale_cluster")
|
|
@mock.patch(SAHARA_CLUSTERS + "._launch_cluster",
|
|
return_value=mock.MagicMock(id=42))
|
|
def test_create_scale_delete_cluster(
|
|
self, mock__launch_cluster, mock__scale_cluster,
|
|
mock__delete_cluster):
|
|
self.clients("sahara").clusters.get.return_value = mock.MagicMock(
|
|
id=42, status="active"
|
|
)
|
|
clusters_scenario = clusters.SaharaClusters()
|
|
|
|
clusters_scenario.context = {
|
|
"tenant": {
|
|
"sahara_image": "test_image"
|
|
}
|
|
}
|
|
|
|
clusters_scenario.create_scale_delete_cluster(
|
|
flavor="test_flavor",
|
|
workers_count=5,
|
|
deltas=[1, -1],
|
|
plugin_name="test_plugin",
|
|
hadoop_version="test_version")
|
|
|
|
mock__launch_cluster.assert_called_once_with(
|
|
flavor_id="test_flavor",
|
|
image_id="test_image",
|
|
workers_count=5,
|
|
plugin_name="test_plugin",
|
|
hadoop_version="test_version",
|
|
floating_ip_pool=None,
|
|
volumes_per_node=None,
|
|
volumes_size=None,
|
|
auto_security_group=None,
|
|
security_groups=None,
|
|
node_configs=None,
|
|
cluster_configs=None,
|
|
enable_anti_affinity=False)
|
|
|
|
mock__scale_cluster.assert_has_calls([
|
|
mock.call(self.clients("sahara").clusters.get.return_value, 1),
|
|
mock.call(self.clients("sahara").clusters.get.return_value, -1),
|
|
])
|
|
|
|
mock__delete_cluster.assert_called_once_with(
|
|
self.clients("sahara").clusters.get.return_value)
|