def test_search_all_resources(self): # Setup Expected Response next_page_token = "" results_element = {} results = [results_element] expected_response = { "next_page_token": next_page_token, "results": results } expected_response = asset_service_pb2.SearchAllResourcesResponse( **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 = asset_v1p1beta1.AssetServiceClient() # Setup Request scope = "scope109264468" paged_list_response = client.search_all_resources(scope) resources = list(paged_list_response) assert len(resources) == 1 assert expected_response.results[0] == resources[0] assert len(channel.requests) == 1 expected_request = asset_service_pb2.SearchAllResourcesRequest( scope=scope) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_search_all_resources_exception(self): channel = ChannelStub(responses=[CustomException()]) patch = mock.patch("google.api_core.grpc_helpers.create_channel") with patch as create_channel: create_channel.return_value = channel client = asset_v1p1beta1.AssetServiceClient() # Setup request scope = "scope109264468" paged_list_response = client.search_all_resources(scope) with pytest.raises(CustomException): list(paged_list_response)
def sample_search_all_iam_policies(): # Create a client client = asset_v1p1beta1.AssetServiceClient() # Initialize request argument(s) request = asset_v1p1beta1.SearchAllIamPoliciesRequest( scope="scope_value", ) # Make the request page_result = client.search_all_iam_policies(request=request) # Handle the response for response in page_result: print(response)