diff --git hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestRestoreSnapshotFromClientWithRegionReplicas.java hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestRestoreSnapshotFromClientWithRegionReplicas.java index 94cf44d..399d1db 100644 --- hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestRestoreSnapshotFromClientWithRegionReplicas.java +++ hbase-server/src/test/java/org/apache/hadoop/hbase/client/TestRestoreSnapshotFromClientWithRegionReplicas.java @@ -21,6 +21,8 @@ import org.apache.hadoop.hbase.testclassification.ClientTests; import org.apache.hadoop.hbase.testclassification.LargeTests; import org.junit.experimental.categories.Category; +import java.io.IOException; + @Category({LargeTests.class, ClientTests.class}) public class TestRestoreSnapshotFromClientWithRegionReplicas extends TestRestoreSnapshotFromClient { @@ -28,4 +30,15 @@ public class TestRestoreSnapshotFromClientWithRegionReplicas extends protected int getNumReplicas() { return 3; } + + @Override + protected int countRows(final Table table, final byte[]... families) throws IOException { + Scan scan = new Scan(); + scan.setConsistency(Consistency.TIMELINE); + for (byte[] family: families) { + scan.addFamily(family); + } + + return TEST_UTIL.countRows(table, scan); + } }