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