Add Elasticsearch Spring Data test cases

spring-data-elasticsearch/src/test/java/com/baeldung/ElasticSearchTest.java

@@ -0,0 +1,130 @@
+package com.baeldung;
+
+import com.baeldung.config.Config;
+import com.baeldung.model.Article;
+import com.baeldung.model.Author;
+import com.baeldung.service.ArticleService;
+import org.elasticsearch.index.query.QueryBuilder;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.data.domain.Page;
+import org.springframework.data.domain.PageRequest;
+import org.springframework.data.elasticsearch.core.ElasticsearchTemplate;
+import org.springframework.data.elasticsearch.core.query.NativeSearchQueryBuilder;
+import org.springframework.data.elasticsearch.core.query.SearchQuery;
+import org.springframework.test.context.ContextConfiguration;
+import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
+import org.springframework.test.context.support.AnnotationConfigContextLoader;
+
+import java.util.List;
+
+import static java.util.Arrays.asList;
+import static org.elasticsearch.index.query.FilterBuilders.regexpFilter;
+import static org.elasticsearch.index.query.QueryBuilders.*;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+
+@RunWith(SpringJUnit4ClassRunner.class)
+@ContextConfiguration(classes = {Config.class}, loader = AnnotationConfigContextLoader.class)
+public class ElasticSearchTest {
+
+    @Autowired
+    private ElasticsearchTemplate elasticsearchTemplate;
+    @Autowired
+    private ArticleService articleService;
+
+    private final Author johnSmith = new Author("John Smith");
+    private final Author johnDoe = new Author("John Doe");
+
+    @Before
+    public void before() {
+        elasticsearchTemplate.deleteIndex(Article.class);
+        elasticsearchTemplate.createIndex(Article.class);
+
+        Article article = new Article("Spring Data Elasticsearch");
+        article.setAuthors(asList(johnSmith, johnDoe));
+        articleService.save(article);
+
+        article = new Article("Search engines");
+        article.setAuthors(asList(johnDoe));
+        articleService.save(article);
+
+        article = new Article("Second Article About Elasticsearch");
+        article.setAuthors(asList(johnSmith));
+        articleService.save(article);
+    }
+
+    @Test
+    public void givenArticleService_whenSaveArticle_thenIdIsAssigned() {
+        List<Author> authors = asList(
+                new Author("John Smith"), johnDoe);
+
+        Article article = new Article("Making Search Elastic");
+        article.setAuthors(authors);
+
+        article = articleService.save(article);
+        assertNotNull(article.getId());
+    }
+
+    @Test
+    public void givenPersistedArticles_whenSearchByAuthorsName_thenRightFound() {
+
+        Page<Article> articleByAuthorName = articleService.findByAuthorName(johnSmith.getName(), new PageRequest(0, 10));
+        assertEquals(2L, articleByAuthorName.getTotalElements());
+    }
+
+    @Test
+    public void givenCustomQuery_whenSearchByAuthorsName_thenArticleIsFound() {
+
+        Page<Article> articleByAuthorName = articleService.findByAuthorNameUsingCustomQuery("John Smith", new PageRequest(0, 10));
+        assertEquals(3L, articleByAuthorName.getTotalElements());
+    }
+
+
+    @Test
+    public void givenPersistedArticles_whenUseRegexQuery_thenRightArticlesFound() {
+
+        SearchQuery searchQuery = new NativeSearchQueryBuilder()
+                .withFilter(regexpFilter("title", ".*data.*"))
+                .build();
+        List<Article> articles = elasticsearchTemplate.queryForList(searchQuery, Article.class);
+
+        assertEquals(1, articles.size());
+    }
+
+    @Test
+    public void givenSavedDoc_whenTitleUpdated_thenCouldFindByUpdatedTitle() {
+        SearchQuery searchQuery = new NativeSearchQueryBuilder()
+                .withQuery(fuzzyQuery("title", "serch"))
+                .build();
+        List<Article> articles = elasticsearchTemplate.queryForList(searchQuery, Article.class);
+
+        assertEquals(1, articles.size());
+
+        Article article = articles.get(0);
+        final String newTitle = "Getting started with Search Engines";
+        article.setTitle(newTitle);
+        articleService.save(article);
+
+        assertEquals(newTitle, articleService.findOne(article.getId()).getTitle());
+    }
+
+    @Test
+    public void givenSavedDoc_whenDelete_thenRemovedFromIndex() {
+
+        final String articleTitle = "Spring Data Elasticsearch";
+
+        SearchQuery searchQuery = new NativeSearchQueryBuilder()
+                .withQuery(matchQuery("title", articleTitle).minimumShouldMatch("75%"))
+                .build();
+        List<Article> articles = elasticsearchTemplate.queryForList(searchQuery, Article.class);
+        assertEquals(1, articles.size());
+        final long count = articleService.count();
+
+        articleService.delete(articles.get(0));
+
+        assertEquals(count - 1, articleService.count());
+    }
+}