def test_read_rows(self):
# Setup Expected Response
last_scanned_row_key = b"-126"
expected_response = {"last_scanned_row_key": last_scanned_row_key}
expected_response = bigtable_pb2.ReadRowsResponse(**expected_response)
# Mock the API response
channel = ChannelStub(responses=[iter([expected_response])])
patch = mock.patch("google.api_core.grpc_helpers.create_channel")
with patch as create_channel:
create_channel.return_value = channel
client = bigtable_v2.BigtableClient()
# Setup Request
table_name = client.table_path("[PROJECT]", "[INSTANCE]", "[TABLE]")
response = client.read_rows(table_name)
resources = list(response)
assert len(resources) == 1
assert expected_response == resources[0]
assert len(channel.requests) == 1
expected_request = bigtable_pb2.ReadRowsRequest(table_name=table_name)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def test_read_rows(self):
# Setup Expected Response
last_scanned_row_key = b'-126'
expected_response = {'last_scanned_row_key': last_scanned_row_key}
expected_response = bigtable_pb2.ReadRowsResponse(**expected_response)
# Mock the API response
channel = ChannelStub(responses=[iter([expected_response])])
client = bigtable_v2.BigtableClient(channel=channel)
# Setup Request
table_name = client.table_path('[PROJECT]', '[INSTANCE]', '[TABLE]')
response = client.read_rows(table_name)
resources = list(response)
assert len(resources) == 1
assert expected_response == resources[0]
assert len(channel.requests) == 1
expected_request = bigtable_pb2.ReadRowsRequest(table_name=table_name)
actual_request = channel.requests[0][1]
assert expected_request == actual_request
def _ReadRowsResponsePB(*args, **kw):
from google.cloud.bigtable_v2.proto import (
bigtable_pb2 as messages_v2_pb2)
return messages_v2_pb2.ReadRowsResponse(*args, **kw)