def test_assert_valid_hook_call(self, mock_hook) -> None: task = CloudDataCatalogCreateTagTemplateOperator( task_id="task_id", location=TEST_LOCATION, tag_template_id=TEST_TAG_TEMPLATE_ID, tag_template=TEST_TAG_TEMPLATE, 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, ) ti = mock.MagicMock() result = task.execute(context={"task_instance": ti}) mock_hook.assert_called_once_with( gcp_conn_id=TEST_GCP_CONN_ID, impersonation_chain=TEST_IMPERSONATION_CHAIN, ) mock_hook.return_value.create_tag_template.assert_called_once_with( location=TEST_LOCATION, tag_template_id=TEST_TAG_TEMPLATE_ID, tag_template=TEST_TAG_TEMPLATE, project_id=TEST_PROJECT_ID, retry=TEST_RETRY, timeout=TEST_TIMEOUT, metadata=TEST_METADATA, ) ti.xcom_push.assert_called_once_with(key="tag_template_id", value=TEST_TAG_TEMPLATE_ID) self.assertEqual(TEST_TAG_TEMPLATE_DICT, result)
# [END howto_operator_gcp_datacatalog_create_tag_result] # [START howto_operator_gcp_datacatalog_create_tag_result2] create_tag_result2 = BashOperator( task_id="create_tag_result2", bash_command="echo \"{{ task_instance.xcom_pull('create_tag') }}\"") # [END howto_operator_gcp_datacatalog_create_tag_result2] # [START howto_operator_gcp_datacatalog_create_tag_template] create_tag_template = CloudDataCatalogCreateTagTemplateOperator( task_id="create_tag_template", location=LOCATION, tag_template_id=TEMPLATE_ID, tag_template={ "display_name": "Awesome Tag Template", "fields": { FIELD_NAME_1: TagTemplateField(display_name="first-field", type=FieldType(primitive_type="STRING")) }, }, ) # [END howto_operator_gcp_datacatalog_create_tag_template] # [START howto_operator_gcp_datacatalog_create_tag_template_result] create_tag_template_result = BashOperator( task_id="create_tag_template_result", bash_command= "echo \"{{ task_instance.xcom_pull('create_tag_template', key='tag_template_id') }}\"", ) # [END howto_operator_gcp_datacatalog_create_tag_template_result]