def test_RRRM_build_updated_request_no_end_key():
from google.cloud.bigtable.row_filters import RowSampleFilter
from google.cloud.bigtable_v2 import types
row_filter = RowSampleFilter(0.33)
last_scanned_key = b"row_key25"
request = _ReadRowsRequestPB(filter=row_filter.to_pb(),
rows_limit=8,
table_name=TABLE_NAME)
row_range1 = types.RowRange(start_key_closed=b"row_key20")
request.rows.row_ranges.append(row_range1)
request_manager = _make_read_rows_request_manager(request,
last_scanned_key, 2)
result = request_manager.build_updated_request()
expected_result = _ReadRowsRequestPB(table_name=TABLE_NAME,
filter=row_filter.to_pb(),
rows_limit=6)
row_range2 = types.RowRange(start_key_open=last_scanned_key)
expected_result.rows.row_ranges.append(row_range2)
assert expected_result == result
def test_RRRM_build_updated_request_full_table():
from google.cloud.bigtable_v2 import types
last_scanned_key = b"row_key14"
request = _ReadRowsRequestPB(table_name=TABLE_NAME)
request_manager = _make_read_rows_request_manager(request,
last_scanned_key, 2)
result = request_manager.build_updated_request()
expected_result = _ReadRowsRequestPB(table_name=TABLE_NAME, filter={})
row_range1 = types.RowRange(start_key_open=last_scanned_key)
expected_result.rows.row_ranges.append(row_range1)
assert expected_result == result