def test_setup_and_cleanup(self, mock_sahara_scenario__launch_cluster,
mock_cleanup):
sahara_ctx = sahara_cluster.SaharaCluster(self.context)
launch_cluster_calls = []
for i in self.tenants:
launch_cluster_calls.append(
mock.call(plugin_name="test_plugin",
hadoop_version="test_version",
flavor_id="test_flavor",
workers_count=2,
image_id=self.context["tenants"][i]["sahara_image"],
floating_ip_pool=None,
volumes_per_node=None,
volumes_size=1,
auto_security_group=True,
security_groups=None,
node_configs=None,
cluster_configs=None,
enable_anti_affinity=False,
enable_proxy=False,
wait_active=False))
self.clients("sahara").clusters.get.side_effect = [
mock.MagicMock(status="not-active"),
mock.MagicMock(status="active")
]
sahara_ctx.setup()
mock_sahara_scenario__launch_cluster.assert_has_calls(
launch_cluster_calls)
sahara_ctx.cleanup()
mock_cleanup.assert_called_once_with(names=["sahara.clusters"],
users=self.context["users"])
def test_setup_and_cleanup_error(self,
mock_sahara_scenario__launch_cluster):
sahara_ctx = sahara_cluster.SaharaCluster(self.context)
launch_cluster_calls = []
for i in self.tenants:
launch_cluster_calls.append(
mock.call(
flavor_id=None,
plugin_name="test_plugin",
hadoop_version="test_version",
master_flavor_id="test_flavor_m",
worker_flavor_id="test_flavor_w",
workers_count=2,
image_id=self.context["tenants"][i]["sahara"]["image"],
floating_ip_pool=None,
volumes_per_node=None,
volumes_size=1,
auto_security_groups=True,
security_groups=None,
node_configs=None,
cluster_configs=None,
wait_active=False,
use_autoconfig=True))
self.clients("sahara").clusters.get.side_effect = [
mock.MagicMock(status="not-active"),
mock.MagicMock(status="error")
]
self.assertRaises(exceptions.SaharaClusterFailure, sahara_ctx.setup)
def test_setup_and_cleanup_error(self, mock_osclients,
mock_sahara_scenario__launch_cluster):
mock_sahara = mock_osclients.Clients(mock.MagicMock()).sahara()
ctx = self.context_without_cluster_keys
sahara_ctx = sahara_cluster.SaharaCluster(ctx)
launch_cluster_calls = []
for i in self.tenants:
launch_cluster_calls.append(
mock.call(plugin_name="test_plugin",
hadoop_version="test_version",
flavor_id="test_flavor",
workers_count=2,
image_id=ctx["tenants"][i]["sahara_image"],
floating_ip_pool=None,
volumes_per_node=None,
volumes_size=1,
auto_security_groups=True,
security_groups=None,
node_configs=None,
cluster_configs=None,
wait_active=False))
mock_sahara.clusters.get.side_effect = [
mock.MagicMock(status="not-active"),
mock.MagicMock(status="error")
]
self.assertRaises(exceptions.SaharaClusterFailure, sahara_ctx.setup)
