Cwikius-Spring/spring-data-elasticsearch
1
0
Fork 0
mirror of https://github.com/spring-projects/spring-data-elasticsearch.git synced 2025-06-23 12:32:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Updated Test (Scan and Scroll) which was causing build failure

Browse Source
This commit is contained in:
Mohsin Husen 2013-05-15 17:27:49 +01:00
parent b5873cb3e6
commit 7f1d26dac3
1 changed files with 35 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
src/test/java/org/springframework/data/elasticsearch/core/ElasticsearchTemplateTests.java
View File

@ -555,18 +555,20 @@ public class ElasticsearchTemplateTests {
indexQuery2.setObject(sampleEntity2); indexQuery2.setObject(sampleEntity2);
indexQueries.add(indexQuery2); indexQueries.add(indexQuery2);
List<IndexQuery> entities = createSampleEntitiesWithMessage("Test message", 30);
// when // when
elasticsearchTemplate.bulkIndex(indexQueries); elasticsearchTemplate.bulkIndex(entities);
elasticsearchTemplate.refresh(SampleEntity.class, true); elasticsearchTemplate.refresh(SampleEntity.class, true);
// then // then
SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(matchAllQuery()).withIndices("test-index") SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(matchAllQuery()).withIndices("test-index")
.withTypes("test-type").withPageable(new PageRequest(0, 1)).build(); .withTypes("test-type").withPageable(new PageRequest(0, 10)).build();
String scrollId = elasticsearchTemplate.scan(searchQuery, 1000, false); String scrollId = elasticsearchTemplate.scan(searchQuery, 1000, false);
List<SampleEntity> sampleEntities = new ArrayList<SampleEntity>(); List<SampleEntity> sampleEntities = new ArrayList<SampleEntity>();
boolean hasRecords = true; boolean hasRecords = true;
while (hasRecords) { while (hasRecords) {
Page<SampleEntity> page = elasticsearchTemplate.scroll(scrollId, 5000L, new ResultsMapper<SampleEntity>() { Page<SampleEntity> page = elasticsearchTemplate.scroll(scrollId, 5000L, new ResultsMapper<SampleEntity>() {
@Override @Override
public Page<SampleEntity> mapResults(SearchResponse response) { public Page<SampleEntity> mapResults(SearchResponse response) {
@ -580,21 +582,45 @@ public class ElasticsearchTemplateTests {
user.setMessage((String) searchHit.getSource().get("message")); user.setMessage((String) searchHit.getSource().get("message"));
chunk.add(user); chunk.add(user);
} }
return new PageImpl<SampleEntity>(chunk); if(chunk.size() > 0){
return new PageImpl<SampleEntity>(chunk);
}
return null;
} }
}); });
if (page != null) { if (page != null) {
sampleEntities.addAll(page.getContent()); sampleEntities.addAll(page.getContent());
hasRecords = page.hasNextPage(); } else {
} else { hasRecords = false;
hasRecords = false; }
}
} }
assertThat(sampleEntities.size(), is(equalTo(2))); assertThat(sampleEntities.size(), is(equalTo(30)));
} }
private static List<IndexQuery> createSampleEntitiesWithMessage(String message, int numberOfEntities) {
List<SampleEntity> sampleEntities = new ArrayList<SampleEntity>();
List<IndexQuery> indexQueries = new ArrayList<IndexQuery>();
for (int i = 0; i < numberOfEntities; i++) {
String documentId = randomNumeric(5);
SampleEntity sampleEntity = new SampleEntity();
sampleEntity.setId(documentId);
sampleEntity.setMessage(message);
sampleEntity.setRate(2);
sampleEntity.setVersion(System.currentTimeMillis());
IndexQuery indexQuery2 = new IndexQuery();
indexQuery2.setId(documentId);
indexQuery2.setObject(sampleEntity);
indexQueries.add(indexQuery2);
sampleEntities.add(sampleEntity);
}
return indexQueries;
}
@Test @Test
public void shouldReturnListForGivenCriteria() { public void shouldReturnListForGivenCriteria() {
// given // given