@Test
public void testSampleRowKeys() {

  List<KeyOffset> expectedKeyOff =
      ImmutableList.of(KeyOffset.create(ByteString.copyFromUtf8(ROW_KEY), 10));
  when(dataClientV2.sampleRowKeys(TABLE_ID)).thenReturn(expectedKeyOff);
  List<KeyOffset> keyOffSets = dataGCJClient.sampleRowKeys(TABLE_ID);
  assertEquals(expectedKeyOff, keyOffSets);
  verify(dataClientV2).sampleRowKeys(TABLE_ID);
}