def test_execute(self, mock_hook):
update_maks = "instance.name"
op = CloudDataFusionUpdateInstanceOperator(
task_id="test_taks",
instance_name=INSTANCE_NAME,
instance=INSTANCE,
update_mask=update_maks,
location=LOCATION,
project_id=PROJECT_ID,
)
op.execute({})
mock_hook.return_value.patch_instance.assert_called_once_with(
instance_name=INSTANCE_NAME,
instance=INSTANCE,
update_mask=update_maks,
location=LOCATION,
project_id=PROJECT_ID,
)
assert mock_hook.return_value.wait_for_operation.call_count == 1
get_instance = CloudDataFusionGetInstanceOperator(
location=LOCATION, instance_name=INSTANCE_NAME, task_id="get_instance")
# [END howto_cloud_data_fusion_get_instance_operator]
# [START howto_cloud_data_fusion_restart_instance_operator]
restart_instance = CloudDataFusionRestartInstanceOperator(
location=LOCATION,
instance_name=INSTANCE_NAME,
task_id="restart_instance")
# [END howto_cloud_data_fusion_restart_instance_operator]
# [START howto_cloud_data_fusion_update_instance_operator]
update_instance = CloudDataFusionUpdateInstanceOperator(
location=LOCATION,
instance_name=INSTANCE_NAME,
instance=INSTANCE,
update_mask="instance.displayName",
task_id="update_instance",
)
# [END howto_cloud_data_fusion_update_instance_operator]
# [START howto_cloud_data_fusion_create_pipeline]
create_pipeline = CloudDataFusionCreatePipelineOperator(
location=LOCATION,
pipeline_name=PIPELINE_NAME,
pipeline=PIPELINE,
instance_name=INSTANCE_NAME,
task_id="create_pipeline",
)
# [END howto_cloud_data_fusion_create_pipeline]