def test_execute(self, mock_hook, mock_uuid): mock_hook.return_value.project_id = GCP_PROJECT mock_uuid.return_value = self.job_id op = DataprocSubmitPySparkJobOperator( task_id=TASK_ID, region=GCP_LOCATION, gcp_conn_id=GCP_CONN_ID, main=self.uri ) job = op.generate_job() self.assertDictEqual(self.job, job)
def test_deprecation_warning(self, mock_hook): with self.assertWarns(DeprecationWarning) as warning: DataprocSubmitPySparkJobOperator( task_id=TASK_ID, region=GCP_LOCATION, main=self.uri, ) assert_warning("DataprocSubmitJobOperator", warning)