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)
