def test_job_type_supported(self, job_get): job, job_exec = u.create_job_exec(edp.JOB_TYPE_PIG) job_get.return_value = job self.assertIsNotNone(job_manager._get_job_engine(u.create_cluster(), job_exec)) job.type = "unsupported_type" self.assertIsNone(job_manager._get_job_engine(u.create_cluster(), job_exec))