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)