def test_read_rows(mock_transport, client_under_test):
stream_position = types.StreamPosition()
client_under_test.read_rows(stream_position)
expected_request = types.ReadRowsRequest(read_position=stream_position)
mock_transport.create_read_session.read_rows(expected_request,
metadata=mock.ANY,
timeout=mock.ANY)
def _copy_stream_position(position):
"""Copy a StreamPosition.
Args:
position (Union[ \
dict, \
~google.cloud.bigquery_storage_v1beta1.types.StreamPosition \
]):
StreamPostion (or dictionary in StreamPosition format) to copy.
Returns:
~google.cloud.bigquery_storage_v1beta1.types.StreamPosition:
A copy of the input StreamPostion.
"""
if isinstance(position, types.StreamPosition):
output = types.StreamPosition()
output.CopyFrom(position)
return output
return types.StreamPosition(**position)
