def test_get_agent(self):
# Setup Expected Response
parent_2 = 'parent21175163357'
display_name = 'displayName1615086568'
default_language_code = 'defaultLanguageCode856575222'
time_zone = 'timeZone36848094'
description = 'description-1724546052'
avatar_uri = 'avatarUri-402824826'
enable_logging = False
classification_threshold = 1.11581064E8
expected_response = {
'parent': parent_2,
'display_name': display_name,
'default_language_code': default_language_code,
'time_zone': time_zone,
'description': description,
'avatar_uri': avatar_uri,
'enable_logging': enable_logging,
'classification_threshold': classification_threshold
}
expected_response = agent_pb2.Agent(**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 = dialogflow_v2.AgentsClient()
# Setup Request
parent = client.project_path('[PROJECT]')
response = client.get_agent(parent)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = agent_pb2.GetAgentRequest(parent=parent)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_get_agent(self):
# Setup Expected Response
parent_2 = "parent21175163357"
display_name = "displayName1615086568"
default_language_code = "defaultLanguageCode856575222"
time_zone = "timeZone36848094"
description = "description-1724546052"
avatar_uri = "avatarUri-402824826"
enable_logging = False
classification_threshold = 1.11581064e8
expected_response = {
"parent": parent_2,
"display_name": display_name,
"default_language_code": default_language_code,
"time_zone": time_zone,
"description": description,
"avatar_uri": avatar_uri,
"enable_logging": enable_logging,
"classification_threshold": classification_threshold,
}
expected_response = agent_pb2.Agent(**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 = dialogflow_v2.AgentsClient()
# Setup Request
parent = client.project_path("[PROJECT]")
response = client.get_agent(parent)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = agent_pb2.GetAgentRequest(parent=parent)
actual_request = channel.requests[0][1]
assert expected_request == actual_request