def test_execute(self, mock_hook): mock_hook.return_value.get_instance.return_value = {"apiEndpoint": INSTANCE_URL} op = CloudDataFusionStopPipelineOperator( task_id="test_taks", pipeline_name=PIPELINE_NAME, instance_name=INSTANCE_NAME, namespace=NAMESPACE, location=LOCATION, project_id=PROJECT_ID, ) op.execute({}) mock_hook.return_value.get_instance.assert_called_once_with( instance_name=INSTANCE_NAME, location=LOCATION, project_id=PROJECT_ID ) mock_hook.return_value.stop_pipeline.assert_called_once_with( instance_url=INSTANCE_URL, pipeline_name=PIPELINE_NAME, namespace=NAMESPACE )
# [START howto_cloud_data_fusion_start_pipeline_sensor] start_pipeline_sensor = CloudDataFusionPipelineStateSensor( task_id="pipeline_state_sensor", pipeline_name=PIPELINE_NAME, pipeline_id=start_pipeline_async.output, expected_statuses=["COMPLETED"], instance_name=INSTANCE_NAME, location=LOCATION, ) # [END howto_cloud_data_fusion_start_pipeline_sensor] # [START howto_cloud_data_fusion_stop_pipeline] stop_pipeline = CloudDataFusionStopPipelineOperator( location=LOCATION, pipeline_name=PIPELINE_NAME, instance_name=INSTANCE_NAME, task_id="stop_pipeline", ) # [END howto_cloud_data_fusion_stop_pipeline] # [START howto_cloud_data_fusion_delete_pipeline] delete_pipeline = CloudDataFusionDeletePipelineOperator( location=LOCATION, pipeline_name=PIPELINE_NAME, instance_name=INSTANCE_NAME, task_id="delete_pipeline", ) # [END howto_cloud_data_fusion_delete_pipeline] # [START howto_cloud_data_fusion_delete_instance_operator] delete_instance = CloudDataFusionDeleteInstanceOperator(