def test_list_log_entries(self): # Setup Expected Response next_page_token = "" entries_element = {} entries = [entries_element] expected_response = { "next_page_token": next_page_token, "entries": entries } expected_response = logging_pb2.ListLogEntriesResponse( **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 = logging_v2.LoggingServiceV2Client() # Setup Request resource_names = [] paged_list_response = client.list_log_entries(resource_names) resources = list(paged_list_response) assert len(resources) == 1 assert expected_response.entries[0] == resources[0] assert len(channel.requests) == 1 expected_request = logging_pb2.ListLogEntriesRequest( resource_names=resource_names) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_list_log_entries(self): # Setup Expected Response next_page_token = '' entries_element = {} entries = [entries_element] expected_response = { 'next_page_token': next_page_token, 'entries': entries } expected_response = logging_pb2.ListLogEntriesResponse( **expected_response) # Mock the API response channel = ChannelStub(responses=[expected_response]) client = logging_v2.LoggingServiceV2Client(channel=channel) # Setup Request resource_names = [] paged_list_response = client.list_log_entries(resource_names) resources = list(paged_list_response) assert len(resources) == 1 assert expected_response.entries[0] == resources[0] assert len(channel.requests) == 1 expected_request = logging_pb2.ListLogEntriesRequest( resource_names=resource_names) actual_request = channel.requests[0][1] assert expected_request == actual_request
def test_list_entries(self): channel, api = self.make_logging_api() log_entry_msg = log_entry_pb2.LogEntry( log_name=self.LOG_PATH, text_payload="text" ) channel.ListLogEntries.response = logging_pb2.ListLogEntriesResponse( entries=[log_entry_msg] ) result = api.list_entries([PROJECT], FILTER, google.cloud.logging.DESCENDING) entries = list(result) # Check the response assert len(entries) == 1 entry = entries[0] assert isinstance(entry, google.cloud.logging.entries.TextEntry) assert entry.payload == "text" # Check the request assert len(channel.ListLogEntries.requests) == 1 request = channel.ListLogEntries.requests[0] assert request.project_ids == [PROJECT] assert request.filter == FILTER assert request.order_by == google.cloud.logging.DESCENDING
def test_list_entries_with_options(self): channel, api = self.make_logging_api() channel.ListLogEntries.response = logging_pb2.ListLogEntriesResponse(entries=[]) result = api.list_entries( [PROJECT], FILTER, google.cloud.logging.ASCENDING, page_size=42, page_token="token", ) list(result) # Check the request assert len(channel.ListLogEntries.requests) == 1 request = channel.ListLogEntries.requests[0] assert request.project_ids == [PROJECT] assert request.filter == FILTER assert request.order_by == google.cloud.logging.ASCENDING assert request.page_size == 42 assert request.page_token == "token"