org.hibernate.search.test.event.autoindexembeddable.EventBasedEmbeddableCollectionUpdateTest.java Source code

Introduction

Here is the source code for org.hibernate.search.test.event.autoindexembeddable.EventBasedEmbeddableCollectionUpdateTest.java

Source

/*
 * Hibernate Search, full-text search for your domain model
 *
 * License: GNU Lesser General Public License (LGPL), version 2.1 or later
 * See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
 */

package org.hibernate.search.test.event.autoindexembeddable;

import java.util.List;
import javax.persistence.EntityManager;

import org.apache.lucene.analysis.standard.StandardAnalyzer;
import org.apache.lucene.queryparser.classic.MultiFieldQueryParser;
import org.apache.lucene.queryparser.classic.ParseException;
import org.apache.lucene.queryparser.classic.QueryParser;
import org.hibernate.search.jpa.FullTextEntityManager;
import org.hibernate.search.jpa.FullTextQuery;
import org.hibernate.search.jpa.Search;
import org.hibernate.search.test.jpa.JPATestCase;
import org.hibernate.search.testsupport.TestForIssue;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import static org.junit.Assert.assertEquals;

/**
 * @author Hardy Ferentschik
 */
@TestForIssue(jiraKey = "HSEARCH-1358")
public class EventBasedEmbeddableCollectionUpdateTest extends JPATestCase {

    private EntityManager entityManager;

    @Before
    @Override
    public void setUp() {
        super.setUp();
        entityManager = factory.createEntityManager();
    }

    @After
    @Override
    public void tearDown() {
        FullTextEntityManager fullTextEntityManager = Search.getFullTextEntityManager(entityManager);
        fullTextEntityManager.purgeAll(Book.class);
        fullTextEntityManager.flushToIndexes();
        fullTextEntityManager.close();
        super.tearDown();
    }

    @Test
    public void testUpdateOfEmbeddedElementCollectionTriggersIndexUpdate() throws Exception {
        indexBookAndEnsureItIsIndexed();

        // find and update book by adding "Bar" into Embedded ElementCollection
        entityManager.getTransaction().begin();

        Book book = entityManager.find(Book.class, 1234L);
        book.getEmbeddableCategories().getCategories().remove(12L);
        book.getEmbeddableCategories().getCategories().put(13L, "Bar");

        entityManager.persist(book);
        entityManager.getTransaction().commit();

        assertEquals("Foo should have been removed by indexed update", 0,
                search("embeddableCategories.categories:Foo").size());
        assertEquals("Bar should have been added by indexed update", 1,
                search("embeddableCategories.categories:Bar").size());
    }

    @Override
    public Class[] getAnnotatedClasses() {
        return new Class[] {}; // configured in persistence.xml
    }

    private void indexBookAndEnsureItIsIndexed() throws ParseException {
        entityManager.getTransaction().begin();

        Book book = new Book();
        book.setId(1234L);
        book.getEmbeddableCategories().getCategories().put(12L, "Foo");

        entityManager.persist(book);
        entityManager.getTransaction().commit();

        assertEquals("Foo should have been added during indexing", 1,
                search("embeddableCategories.categories:Foo").size());
        assertEquals("Bar was not yet added", 0, search("embeddableCategories.categories:Bar").size());
    }

    @SuppressWarnings("unchecked")
    private List<Book> search(String searchQuery) throws ParseException {
        QueryParser parser = new MultiFieldQueryParser(new String[] {}, new StandardAnalyzer());
        FullTextQuery query = Search.getFullTextEntityManager(entityManager)
                .createFullTextQuery(parser.parse(searchQuery), Book.class);
        return (List<Book>) query.getResultList();
    }
}