def test_terminate_cluster(self): cluster_id = self.make_cluster(pool_clusters=True) self.monkey_patch_argv('--quiet', '--no-conf', 'j-MOCKCLUSTER0') terminate_main() emr_conn = EMRJobRunner(conf_paths=[]).make_emr_conn() cluster = _patched_describe_cluster(emr_conn, cluster_id) emr_conn.simulate_progress(cluster_id) self.assertEqual(cluster.status.state, 'TERMINATED')
def test_terminate_pool(self): jf_id = self.make_job_flow(pool_emr_job_flows=True) emr_conn = EMRJobRunner(conf_paths=[]).make_emr_conn() for i in range(3): emr_conn.simulate_progress(jf_id) self.monkey_patch_argv('--quiet', '--no-conf', '-t', jf_id) self.monkey_patch_stdout() pool_main() value = self.stdout.getvalue() self.assertIn('j-MOCKJOBFLOW0', value)
def test_terminate_pool(self): jf_id = self.make_job_flow(pool_emr_job_flows=True) emr_conn = EMRJobRunner(conf_path=False).make_emr_conn() for i in range(3): emr_conn.simulate_progress(jf_id) self.monkey_patch_argv("--quiet", "--no-conf", "-t", jf_id) self.monkey_patch_stdout() pool_main() value = self.stdout.getvalue() self.assertIn("j-MOCKJOBFLOW0", value)
def test_find_job_flow(self): jf_id = self.make_job_flow(pool_emr_job_flows=True) emr_conn = EMRJobRunner(conf_paths=[]).make_emr_conn() for i in range(3): emr_conn.simulate_progress(jf_id) self.monkey_patch_argv( '--quiet', '--no-conf', '-f') self.monkey_patch_stdout() pool_main() value = self.stdout.getvalue() self.assertIn('j-MOCKJOBFLOW0', value)