def test_cfn_cluster_create_nowait(self):
template_url = setup_configurations()
args = CreateClusterArgs(template_url, True)
pcluster.create(args)
log = test_log_stream.getvalue()
success_message = "INFO:parallelcluster.parallelcluster:Status: CREATE_COMPLETE"
error_prefix = "CRITICAL:"
self.assertTrue(success_message in log)
self.assertFalse(error_prefix in log)
def test_cfn_cluster_create_wait(self):
template_url = setup_configurations()
args = CreateClusterArgs(template_url, False)
pcluster.create(args)
log = test_log_stream.getvalue()
success_message = 'INFO:parallelcluster.parallelcluster:MasterPublicIP:'
error_prefix = "CRITICAL:"
self.assertTrue(success_message in log)
self.assertFalse(error_prefix in log)
def test_cfn_cluster_create_wait(self):
template_url = setup_configurations()
args = CreateClusterArgs(template_url, False)
pcluster.create(args)
log = test_log_stream.getvalue()
success_message = "INFO:parallelcluster.parallelcluster:MasterPublicIP:"
error_prefix = "CRITICAL:"
self.assertTrue(success_message in log)
self.assertFalse(error_prefix in log)
def test_cfn_cluster_start(self):
template_url = setup_configurations()
args = CreateClusterArgs(template_url, True)
pcluster.create(args)
pcluster.start(args)
log = test_log_stream.getvalue()
error_prefix = "CRITICAL:"
success_message = "Starting compute fleet"
self.assertFalse(error_prefix in log)
self.assertTrue(success_message in log)
def test_cfn_cluster_delete(self):
template_url = setup_configurations()
args = CreateClusterArgs(template_url, True)
pcluster.create(args)
with self.assertRaises(SystemExit) as sys_ex:
pcluster.delete(args)
self.assertEqual(sys_ex.exception.code, 0)
success_message = "Cluster deleted successfully"
log = test_log_stream.getvalue()
self.assertTrue(success_message in log)
def test_cfn_cluster_update_with_reset(self):
template_url = setup_configurations()
args = UpdateClusterArgs(template_url, True, True)
pcluster.create(args)
pcluster.update(args)
success_message = 'INFO:parallelcluster.parallelcluster:Status: UPDATE_COMPLETE'
log = test_log_stream.getvalue()
self.assertTrue(success_message in log)
error_prefix = "CRITICAL:"
self.assertFalse(error_prefix in log)
def test_cfn_cluster_create_fail(self):
template_url = setup_configurations()
args = CreateClusterArgs('', True)
with self.assertRaises(SystemExit) as sys_ex:
pcluster.create(args)
self.assertEqual(sys_ex.exception.code, 1)
log = test_log_stream.getvalue()
error_prefix = "CRITICAL:"
self.assertTrue(error_prefix in log)
def test_cfn_cluster_start(self):
template_url = setup_configurations()
args = CreateClusterArgs(template_url, True)
pcluster.create(args)
pcluster.start(args)
log = test_log_stream.getvalue()
error_prefix = "CRITICAL:"
success_message = "Starting compute fleet"
self.assertFalse(error_prefix in log)
self.assertTrue(success_message in log)
def test_cfn_cluster_delete(self):
template_url = setup_configurations()
args = CreateClusterArgs(template_url, True)
pcluster.create(args)
with self.assertRaises(SystemExit) as sys_ex:
pcluster.delete(args)
self.assertEqual(sys_ex.exception.code, 0)
success_message = "Cluster deleted successfully"
log = test_log_stream.getvalue()
self.assertTrue(success_message in log)
def test_cfn_cluster_update_with_reset(self):
template_url = setup_configurations()
args = UpdateClusterArgs(template_url, True, True)
pcluster.create(args)
pcluster.update(args)
success_message = "INFO:parallelcluster.parallelcluster:Status: UPDATE_COMPLETE"
log = test_log_stream.getvalue()
self.assertTrue(success_message in log)
error_prefix = "CRITICAL:"
self.assertFalse(error_prefix in log)
def test_cfn_cluster_list_nonempty(self):
template_url = setup_configurations()
args = CreateClusterArgs(template_url, False)
pcluster.create(args)
# reset the logger
self.tearDown()
pcluster.list(args)
log = test_log_stream.getvalue()
cluster_name = re.match(r"INFO:parallelcluster.parallelcluster:(\w+)", log).group(1)
self.assertEqual(cluster_name, args.cluster_name)
def test_cfn_cluster_create_fail(self):
setup_configurations()
args = CreateClusterArgs("", True)
with self.assertRaises(SystemExit) as sys_ex:
pcluster.create(args)
self.assertEqual(sys_ex.exception.code, 1)
log = test_log_stream.getvalue()
error_prefix = "CRITICAL:"
self.assertTrue(error_prefix in log)
def test_cfn_cluster_list_nonempty(self):
template_url = setup_configurations()
args = CreateClusterArgs(template_url, False)
pcluster.create(args)
# reset the logger
self.tearDown()
pcluster.list(args)
log = test_log_stream.getvalue()
cluster_name = re.match(r"INFO:parallelcluster.parallelcluster:(\w+)",
log).group(1)
self.assertEqual(cluster_name, args.cluster_name)
def create(args):
pcluster.create(args)
def create(args):
pcluster.create(args)
