if (i == start + limit - 1) {
endKey = entity.getId();
}
allKeys.add(entity.getId());
}
Query query = Query.get(select(allColumns(), from(FirstEntity.class), where(range(startKey, endKey)), limit(limit)));
log.debug(query.getQuery());
List<FirstEntity> results = factory.getResultList(FirstEntity.class, query);
assertEquals(limit, results.size());
assertEquals(startKey, results.get(0).getId());
assertEquals(endKey, results.get(results.size() - 1).getId());
query = Query.get(select(allColumns(), from(FirstEntity.class), where(startAt(startKey)), limit(limit)));
log.debug(query.getQuery());
results = factory.getResultList(FirstEntity.class, query);
assertEquals(limit, results.size());
assertEquals(startKey, results.get(0).getId());
assertEquals(endKey, results.get(results.size() - 1).getId());
}