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

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

@ -555,13 +555,15 @@ 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>();
@ -580,19 +582,43 @@ public class ElasticsearchTemplateTests {
user.setMessage((String) searchHit.getSource().get("message")); user.setMessage((String) searchHit.getSource().get("message"));
chunk.add(user); chunk.add(user);
} }
if(chunk.size() > 0){
return new PageImpl<SampleEntity>(chunk); 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