def test_rename_tag_template_field(self):
# Setup Expected Response
name_2 = "name2-1052831874"
display_name = "displayName1615086568"
is_required = True
expected_response = {
"name": name_2,
"display_name": display_name,
"is_required": is_required,
}
expected_response = tags_pb2.TagTemplateField(**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = datacatalog_v1beta1.DataCatalogClient()
# Setup Request
name = client.field_path("[PROJECT]", "[LOCATION]", "[TAG_TEMPLATE]",
"[FIELD]")
new_tag_template_field_id = "newTagTemplateFieldId-1668354591"
response = client.rename_tag_template_field(name,
new_tag_template_field_id)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = datacatalog_pb2.RenameTagTemplateFieldRequest(
name=name, new_tag_template_field_id=new_tag_template_field_id)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_create_tag_template_field(self):
# Setup Expected Response
name = "name3373707"
display_name = "displayName1615086568"
expected_response = {"name": name, "display_name": display_name}
expected_response = tags_pb2.TagTemplateField(**expected_response)
# Mock the API response
channel = ChannelStub(responses=[expected_response])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = datacatalog_v1beta1.DataCatalogClient()
# Setup Request
parent = client.tag_template_path("[PROJECT]", "[LOCATION]",
"[TAG_TEMPLATE]")
tag_template_field_id = "tagTemplateFieldId-92144832"
tag_template_field = {}
response = client.create_tag_template_field(parent,
tag_template_field_id,
tag_template_field)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = datacatalog_pb2.CreateTagTemplateFieldRequest(
parent=parent,
tag_template_field_id=tag_template_field_id,
tag_template_field=tag_template_field,
)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
