def test_assert_valid_hook_call(self, mock_hook) -> None:
task = CloudDataCatalogUpdateEntryOperator(
task_id="task_id",
entry=TEST_ENTRY,
update_mask=TEST_UPDATE_MASK,
location=TEST_LOCATION,
entry_group=TEST_ENTRY_GROUP_ID,
entry_id=TEST_ENTRY_ID,
project_id=TEST_PROJECT_ID,
retry=TEST_RETRY,
timeout=TEST_TIMEOUT,
metadata=TEST_METADATA,
gcp_conn_id=TEST_GCP_CONN_ID,
impersonation_chain=TEST_IMPERSONATION_CHAIN,
)
task.execute(context=mock.MagicMock())
mock_hook.assert_called_once_with(
gcp_conn_id=TEST_GCP_CONN_ID,
impersonation_chain=TEST_IMPERSONATION_CHAIN,
)
mock_hook.return_value.update_entry.assert_called_once_with(
entry=TEST_ENTRY,
update_mask=TEST_UPDATE_MASK,
location=TEST_LOCATION,
entry_group=TEST_ENTRY_GROUP_ID,
entry_id=TEST_ENTRY_ID,
project_id=TEST_PROJECT_ID,
retry=TEST_RETRY,
timeout=TEST_TIMEOUT,
metadata=TEST_METADATA,
)
query=f"projectid:{PROJECT_ID}")
# [END howto_operator_gcp_datacatalog_search_catalog]
# [START howto_operator_gcp_datacatalog_search_catalog_result]
search_catalog_result = BashOperator(
task_id="search_catalog_result",
bash_command="echo \"{{ task_instance.xcom_pull('search_catalog') }}\"",
)
# [END howto_operator_gcp_datacatalog_search_catalog_result]
# Update
# [START howto_operator_gcp_datacatalog_update_entry]
update_entry = CloudDataCatalogUpdateEntryOperator(
task_id="update_entry",
entry={"display_name": "New Wizard"},
update_mask={"paths": ["display_name"]},
location=LOCATION,
entry_group=ENTRY_GROUP_ID,
entry_id=ENTRY_ID,
)
# [END howto_operator_gcp_datacatalog_update_entry]
# [START howto_operator_gcp_datacatalog_update_tag]
update_tag = CloudDataCatalogUpdateTagOperator(
task_id="update_tag",
tag={
"fields": {
FIELD_NAME_1: TagField(string_value="new-value-string")
}
},
update_mask={"paths": ["fields"]},
location=LOCATION,
