def test_create_tenant(self):
# Setup Expected Response
name = "name3373707"
external_id = "externalId-1153075697"
expected_response = {"name": name, "external_id": external_id}
expected_response = tenant_pb2.Tenant(**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 = talent_v4beta1.TenantServiceClient()
# Setup Request
parent = client.project_path("[PROJECT]")
tenant = {}
response = client.create_tenant(parent, tenant)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = tenant_service_pb2.CreateTenantRequest(
parent=parent, tenant=tenant)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_update_tenant(self):
# Setup Expected Response
name = 'name3373707'
external_id = 'externalId-1153075697'
expected_response = {'name': name, 'external_id': external_id}
expected_response = tenant_pb2.Tenant(**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 = talent_v4beta1.TenantServiceClient()
# Setup Request
tenant = {}
response = client.update_tenant(tenant)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = tenant_service_pb2.UpdateTenantRequest(
tenant=tenant)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_get_tenant(self):
# Setup Expected Response
name_2 = "name2-1052831874"
external_id = "externalId-1153075697"
expected_response = {"name": name_2, "external_id": external_id}
expected_response = tenant_pb2.Tenant(**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 = talent_v4beta1.TenantServiceClient()
# Setup Request
name = client.tenant_path("[PROJECT]", "[TENANT]")
response = client.get_tenant(name)
assert expected_response == response
assert len(channel.requests) == 1
expected_request = tenant_service_pb2.GetTenantRequest(name=name)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
