def setup_cluster(self, cluster_type='base'): cluster_types = ['presto', 'base'] config_filename = ConfigurableCluster.check_for_cluster_config() try: if cluster_type == 'presto': if config_filename: self.cluster = ConfigurableCluster.start_presto_cluster( config_filename, self.install_default_presto, self.assert_installed ) else: self.cluster = DockerCluster.start_presto_cluster( self.install_default_presto) elif cluster_type == 'base': if config_filename: self.cluster = ConfigurableCluster.start_base_cluster( config_filename, self.assert_installed ) else: self.cluster = DockerCluster.start_base_cluster() else: self.fail('{0} is not a supported cluster type. Must choose ' 'one from {1}'.format(cluster_type, cluster_types)) except DockerClusterException as e: self.fail(e.msg)
def setup_docker_cluster(self, cluster_type="centos"): cluster_types = ["presto", "centos"] if cluster_type not in cluster_types: self.fail( "{0} is not a supported cluster type. Must choose one" " from {1}".format(cluster_type, cluster_types) ) try: are_presto_images_present = DockerCluster.check_for_presto_images() if cluster_type == "presto" and are_presto_images_present: self.docker_cluster = DockerCluster.start_presto_cluster() return self.docker_cluster = DockerCluster.start_centos_cluster() if cluster_type == "presto" and not are_presto_images_present: self.install_presto_admin(self.docker_cluster) self.upload_topology() self.server_install() self.docker_client.commit(self.docker_cluster.master, INSTALLED_PRESTO_TEST_MASTER_IMAGE) self.docker_client.commit(self.docker_cluster.slaves[0], INSTALLED_PRESTO_TEST_SLAVE_IMAGE) except DockerClusterException as e: self.fail(e.msg)