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)
