def test_cfn_cluster_update_fail(self): template_url = setup_configurations() args = UpdateClusterArgs(template_url, True, False) with self.assertRaises(SystemExit) as sys_ex: pcluster.update(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_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_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 update(args): pcluster.update(args)