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

DATAES-55 - polish test cases

Browse Source
This commit is contained in:
mohsin.husen 2014-02-14 09:29:48 +00:00
parent 08a6a9b26f
commit ea50046f45
1 changed files with 42 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -60,6 +60,9 @@ import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
@ContextConfiguration("classpath:elasticsearch-template-test.xml") @ContextConfiguration("classpath:elasticsearch-template-test.xml")
public class ElasticsearchTemplateTests { public class ElasticsearchTemplateTests {
private static final String INDEX_NAME = "test-index";
private static final String TYPE_NAME = "test-type";
@Autowired @Autowired
private ElasticsearchTemplate elasticsearchTemplate; private ElasticsearchTemplate elasticsearchTemplate;
@ -189,7 +192,7 @@ public class ElasticsearchTemplateTests {
elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.index(indexQuery);
// when // when
elasticsearchTemplate.delete("test-index", "test-type", documentId); elasticsearchTemplate.delete(INDEX_NAME, TYPE_NAME, documentId);
elasticsearchTemplate.refresh(SampleEntity.class, true); elasticsearchTemplate.refresh(SampleEntity.class, true);
// then // then
SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(fieldQuery("id", documentId)).build(); SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(fieldQuery("id", documentId)).build();
@ -467,8 +470,8 @@ public class ElasticsearchTemplateTests {
elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.index(indexQuery);
elasticsearchTemplate.refresh(SampleEntity.class, true); elasticsearchTemplate.refresh(SampleEntity.class, true);
SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(matchAllQuery()).withIndices("test-index") SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(matchAllQuery()).withIndices(INDEX_NAME)
.withTypes("test-type").withFields("message").build(); .withTypes(TYPE_NAME).withFields("message").build();
// when // when
Page<String> page = elasticsearchTemplate.queryForPage(searchQuery, String.class, new SearchResultMapper() { Page<String> page = elasticsearchTemplate.queryForPage(searchQuery, String.class, new SearchResultMapper() {
@Override @Override
@ -541,8 +544,8 @@ public class ElasticsearchTemplateTests {
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(INDEX_NAME)
.withTypes("test-type").withPageable(new PageRequest(0, 10)).build(); .withTypes(TYPE_NAME).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>();
@ -567,8 +570,8 @@ public class ElasticsearchTemplateTests {
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(INDEX_NAME)
.withTypes("test-type").withPageable(new PageRequest(0, 10)).build(); .withTypes(TYPE_NAME).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>();
@ -871,10 +874,10 @@ public class ElasticsearchTemplateTests {
elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.index(indexQuery);
elasticsearchTemplate.refresh(SampleEntity.class, true); elasticsearchTemplate.refresh(SampleEntity.class, true);
// when // when
elasticsearchTemplate.deleteType("test-index", "test-type"); elasticsearchTemplate.deleteType(INDEX_NAME, TYPE_NAME);
//then //then
boolean typeExists = elasticsearchTemplate.typeExists("test-index", "test-type"); boolean typeExists = elasticsearchTemplate.typeExists(INDEX_NAME, TYPE_NAME);
assertThat(typeExists, is(false)); assertThat(typeExists, is(false));
} }
@ -896,8 +899,8 @@ public class ElasticsearchTemplateTests {
// when // when
DeleteQuery deleteQuery = new DeleteQuery(); DeleteQuery deleteQuery = new DeleteQuery();
deleteQuery.setQuery(fieldQuery("id", documentId)); deleteQuery.setQuery(fieldQuery("id", documentId));
deleteQuery.setIndex("test-index"); deleteQuery.setIndex(INDEX_NAME);
deleteQuery.setType("test-type"); deleteQuery.setType(TYPE_NAME);
elasticsearchTemplate.delete(deleteQuery); elasticsearchTemplate.delete(deleteQuery);
// then // then
SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(fieldQuery("id", documentId)).build(); SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(fieldQuery("id", documentId)).build();
@ -909,22 +912,23 @@ public class ElasticsearchTemplateTests {
public void shouldAddAlias() { public void shouldAddAlias() {
// given // given
elasticsearchTemplate.createIndex(SampleEntity.class); elasticsearchTemplate.createIndex(SampleEntity.class);
String aliasName = "test-alias";
AliasQuery aliasQuery = new AliasBuilder() AliasQuery aliasQuery = new AliasBuilder()
.withIndexName("test-index") .withIndexName(INDEX_NAME)
.withAliasName("test-alias").build(); .withAliasName(aliasName).build();
// when // when
elasticsearchTemplate.addAlias(aliasQuery); elasticsearchTemplate.addAlias(aliasQuery);
// then // then
Set<String> aliases = elasticsearchTemplate.queryForAlias("test-index"); Set<String> aliases = elasticsearchTemplate.queryForAlias(INDEX_NAME);
assertThat(aliases, is(notNullValue())); assertThat(aliases, is(notNullValue()));
assertThat(aliases.contains("test-alias"), is(true)); assertThat(aliases.contains(aliasName), is(true));
} }
@Test @Test
public void shouldRemoveAlias() { public void shouldRemoveAlias() {
// given // given
elasticsearchTemplate.createIndex(SampleEntity.class); elasticsearchTemplate.createIndex(SampleEntity.class);
String indexName = "test-index"; String indexName = INDEX_NAME;
String aliasName = "test-alias"; String aliasName = "test-alias";
AliasQuery aliasQuery = new AliasBuilder() AliasQuery aliasQuery = new AliasBuilder()
.withIndexName(indexName) .withIndexName(indexName)
@ -950,14 +954,14 @@ public class ElasticsearchTemplateTests {
IndexQuery indexQuery = new IndexQuery(); IndexQuery indexQuery = new IndexQuery();
indexQuery.setId("2333343434"); indexQuery.setId("2333343434");
indexQuery.setSource(documentSource); indexQuery.setSource(documentSource);
indexQuery.setIndexName("test-index"); indexQuery.setIndexName(INDEX_NAME);
indexQuery.setType("test-type"); indexQuery.setType(TYPE_NAME);
// when // when
elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.index(indexQuery);
elasticsearchTemplate.refresh(SampleEntity.class, true); elasticsearchTemplate.refresh(SampleEntity.class, true);
SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(termQuery("id", indexQuery.getId())) SearchQuery searchQuery = new NativeSearchQueryBuilder().withQuery(termQuery("id", indexQuery.getId()))
.withIndices("test-index") .withIndices(INDEX_NAME)
.withTypes("test-type") .withTypes(TYPE_NAME)
.build(); .build();
// then // then
Page<SampleEntity> page = elasticsearchTemplate.queryForPage(searchQuery, SampleEntity.class, new SearchResultMapper() { Page<SampleEntity> page = elasticsearchTemplate.queryForPage(searchQuery, SampleEntity.class, new SearchResultMapper() {
@ -983,8 +987,8 @@ public class ElasticsearchTemplateTests {
// given // given
IndexQuery indexQuery = new IndexQuery(); IndexQuery indexQuery = new IndexQuery();
indexQuery.setId("2333343434"); indexQuery.setId("2333343434");
indexQuery.setIndexName("test-index"); indexQuery.setIndexName(INDEX_NAME);
indexQuery.setType("test-type"); indexQuery.setType(TYPE_NAME);
//when //when
elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.index(indexQuery);
@ -999,8 +1003,8 @@ public class ElasticsearchTemplateTests {
elasticsearchTemplate.refresh(SampleEntity.class, true); elasticsearchTemplate.refresh(SampleEntity.class, true);
SearchQuery searchQuery = new NativeSearchQueryBuilder() SearchQuery searchQuery = new NativeSearchQueryBuilder()
.withQuery(termQuery("message", "message")) .withQuery(termQuery("message", "message"))
.withIndices("test-index") .withIndices(INDEX_NAME)
.withTypes("test-type") .withTypes(TYPE_NAME)
.withPageable(new PageRequest(0, 100)) .withPageable(new PageRequest(0, 100))
.build(); .build();
// then // then
@ -1024,8 +1028,8 @@ public class ElasticsearchTemplateTests {
// when // when
SearchQuery searchQuery = new NativeSearchQueryBuilder() SearchQuery searchQuery = new NativeSearchQueryBuilder()
.withQuery(boolQuery().must(wildcardQuery("message", "*a*")).should(wildcardQuery("message", "*b*"))) .withQuery(boolQuery().must(wildcardQuery("message", "*a*")).should(wildcardQuery("message", "*b*")))
.withIndices("test-index") .withIndices(INDEX_NAME)
.withTypes("test-type") .withTypes(TYPE_NAME)
.withMinScore(0.5F) .withMinScore(0.5F)
.build(); .build();
@ -1110,14 +1114,14 @@ public class ElasticsearchTemplateTests {
IndexQuery indexQuery1 = new IndexQuery(); IndexQuery indexQuery1 = new IndexQuery();
indexQuery1.setId("1"); indexQuery1.setId("1");
indexQuery1.setObject(person1); indexQuery1.setObject(person1);
indexQuery1.setIndexName("test-index"); indexQuery1.setIndexName(INDEX_NAME);
indexQuery1.setType("test-type"); indexQuery1.setType(TYPE_NAME);
IndexQuery indexQuery2 = new IndexQuery(); IndexQuery indexQuery2 = new IndexQuery();
indexQuery2.setId("2"); indexQuery2.setId("2");
indexQuery2.setObject(person2); indexQuery2.setObject(person2);
indexQuery2.setIndexName("test-index"); indexQuery2.setIndexName(INDEX_NAME);
indexQuery2.setType("test-type"); indexQuery2.setType(TYPE_NAME);
List<IndexQuery> indexQueries = new ArrayList<IndexQuery>(); List<IndexQuery> indexQueries = new ArrayList<IndexQuery>();
indexQueries.add(indexQuery1); indexQueries.add(indexQuery1);
@ -1125,11 +1129,11 @@ public class ElasticsearchTemplateTests {
//when //when
elasticsearchTemplate.bulkIndex(indexQueries); elasticsearchTemplate.bulkIndex(indexQueries);
elasticsearchTemplate.refresh("test-index", true); elasticsearchTemplate.refresh(INDEX_NAME, true);
// then // then
SearchQuery searchQuery = new NativeSearchQueryBuilder().withIndices("test-index") SearchQuery searchQuery = new NativeSearchQueryBuilder().withIndices(INDEX_NAME)
.withTypes("test-type").withQuery(matchAllQuery()).build(); .withTypes(TYPE_NAME).withQuery(matchAllQuery()).build();
Page<Map> sampleEntities = elasticsearchTemplate.queryForPage(searchQuery, Map.class, new SearchResultMapper() { Page<Map> sampleEntities = elasticsearchTemplate.queryForPage(searchQuery, Map.class, new SearchResultMapper() {
@Override @Override
public <T> FacetedPage<T> mapResults(SearchResponse response, Class<T> clazz, Pageable pageable) { public <T> FacetedPage<T> mapResults(SearchResponse response, Class<T> clazz, Pageable pageable) {
@ -1168,15 +1172,15 @@ public class ElasticsearchTemplateTests {
IndexQuery indexQuery = new IndexQuery(); IndexQuery indexQuery = new IndexQuery();
indexQuery.setId(documentId); indexQuery.setId(documentId);
indexQuery.setIndexName("test-index"); indexQuery.setIndexName(INDEX_NAME);
indexQuery.setType("test-type"); indexQuery.setType(TYPE_NAME);
indexQuery.setObject(sampleEntity); indexQuery.setObject(sampleEntity);
elasticsearchTemplate.index(indexQuery); elasticsearchTemplate.index(indexQuery);
elasticsearchTemplate.refresh("test-index", true); elasticsearchTemplate.refresh(INDEX_NAME, true);
SearchQuery searchQuery = new NativeSearchQueryBuilder().withIndices("test-index") SearchQuery searchQuery = new NativeSearchQueryBuilder().withIndices(INDEX_NAME)
.withTypes("test-type").withQuery(matchAllQuery()).build(); .withTypes(TYPE_NAME).withQuery(matchAllQuery()).build();
// when // when
Page<SampleEntity> sampleEntities = elasticsearchTemplate.queryForPage(searchQuery, SampleEntity.class); Page<SampleEntity> sampleEntities = elasticsearchTemplate.queryForPage(searchQuery, SampleEntity.class);
// then // then