def test_list_realms_flattened_error(): client = RealmsServiceClient( credentials=credentials.AnonymousCredentials()) # Attempting to call a method with both a request object and flattened # fields is an error. with pytest.raises(ValueError): client.list_realms(realms.ListRealmsRequest(), parent="parent_value")
def test_list_realms(transport: str = "grpc"): client = RealmsServiceClient( credentials=credentials.AnonymousCredentials(), transport=transport) # Everything is optional in proto3 as far as the runtime is concerned, # and we are mocking out the actual API, so just send an empty request. request = realms.ListRealmsRequest() # Mock the actual call within the gRPC stub, and fake the request. with mock.patch.object(type(client._transport.list_realms), "__call__") as call: # Designate an appropriate return value for the call. call.return_value = realms.ListRealmsResponse( next_page_token="next_page_token_value", unreachable=["unreachable_value"]) response = client.list_realms(request) # Establish that the underlying gRPC stub method was called. assert len(call.mock_calls) == 1 _, args, _ = call.mock_calls[0] assert args[0] == request # Establish that the response is the type that we expect. assert isinstance(response, pagers.ListRealmsPager) assert response.next_page_token == "next_page_token_value" assert response.unreachable == ["unreachable_value"]
def test_list_realms_field_headers(): client = RealmsServiceClient( credentials=credentials.AnonymousCredentials()) # Any value that is part of the HTTP/1.1 URI should be sent as # a field header. Set these to a non-empty value. request = realms.ListRealmsRequest(parent="parent/value") # Mock the actual call within the gRPC stub, and fake the request. with mock.patch.object(type(client._transport.list_realms), "__call__") as call: call.return_value = realms.ListRealmsResponse() client.list_realms(request) # Establish that the underlying gRPC stub method was called. assert len(call.mock_calls) == 1 _, args, _ = call.mock_calls[0] assert args[0] == request # Establish that the field header was sent. _, _, kw = call.mock_calls[0] assert ("x-goog-request-params", "parent=parent/value") in kw["metadata"]
def test_list_realms_flattened(): client = RealmsServiceClient( credentials=credentials.AnonymousCredentials()) # Mock the actual call within the gRPC stub, and fake the request. with mock.patch.object(type(client._transport.list_realms), "__call__") as call: # Designate an appropriate return value for the call. call.return_value = realms.ListRealmsResponse() # Call the method with a truthy value for each flattened field, # using the keyword arguments to the method. response = client.list_realms(parent="parent_value") # Establish that the underlying call was made with the expected # request object values. assert len(call.mock_calls) == 1 _, args, _ = call.mock_calls[0] assert args[0].parent == "parent_value"
def test_list_realms_pages(): client = RealmsServiceClient(credentials=credentials.AnonymousCredentials) # Mock the actual call within the gRPC stub, and fake the request. with mock.patch.object(type(client._transport.list_realms), "__call__") as call: # Set the response to a series of pages. call.side_effect = ( realms.ListRealmsResponse( realms=[realms.Realm(), realms.Realm(), realms.Realm()], next_page_token="abc", ), realms.ListRealmsResponse(realms=[], next_page_token="def"), realms.ListRealmsResponse(realms=[realms.Realm()], next_page_token="ghi"), realms.ListRealmsResponse( realms=[realms.Realm(), realms.Realm()]), RuntimeError, ) pages = list(client.list_realms(request={}).pages) for page, token in zip(pages, ["abc", "def", "ghi", ""]): assert page.raw_page.next_page_token == token
def test_list_realms_pager(): client = RealmsServiceClient(credentials=credentials.AnonymousCredentials) # Mock the actual call within the gRPC stub, and fake the request. with mock.patch.object(type(client._transport.list_realms), "__call__") as call: # Set the response to a series of pages. call.side_effect = ( realms.ListRealmsResponse( realms=[realms.Realm(), realms.Realm(), realms.Realm()], next_page_token="abc", ), realms.ListRealmsResponse(realms=[], next_page_token="def"), realms.ListRealmsResponse(realms=[realms.Realm()], next_page_token="ghi"), realms.ListRealmsResponse( realms=[realms.Realm(), realms.Realm()]), RuntimeError, ) results = [i for i in client.list_realms(request={})] assert len(results) == 6 assert all(isinstance(i, realms.Realm) for i in results)