/*
* 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.jpa;
import org.apache.lucene.index.Term;
import org.apache.lucene.queryparser.classic.QueryParser;
import org.apache.lucene.search.Query;
import org.apache.lucene.search.TermQuery;
import org.hibernate.search.jpa.FullTextEntityManager;
import org.hibernate.search.jpa.FullTextQuery;
import org.hibernate.search.jpa.Search;
import org.hibernate.search.query.dsl.QueryBuilder;
import org.hibernate.search.testsupport.TestConstants;
import org.junit.Test;
import static org.junit.Assert.assertEquals;
/**
* @author Emmanuel Bernard
*/
public class EntityManagerTest extends JPATestCase {
@Test
public void testMassIndexer() throws Exception {
// index a Bretzel
FullTextEntityManager em = Search.getFullTextEntityManager( factory.createEntityManager() );
em.getTransaction().begin();
Bretzel bretzel = new Bretzel( 23, 34 );
em.persist( bretzel );
em.getTransaction().commit();
em.clear();
// verify against index
assertEquals( 1, countBretzelsViaIndex( em ) );
em.purgeAll( Bretzel.class );
// clear index
em.flushToIndexes();
// verify Bretzel removed from index
assertEquals( 0, countBretzelsViaIndex( em ) );
// re-index
em.createIndexer( Bretzel.class ).startAndWait();
assertEquals( 1, countBretzelsViaIndex( em ) );
}
@Test
public void testQuery() throws Exception {
FullTextEntityManager em = Search.getFullTextEntityManager( factory.createEntityManager() );
em.getTransaction().begin();
Bretzel bretzel = new Bretzel( 23, 34 );
em.persist( bretzel );
em.getTransaction().commit();
em.clear();
em.getTransaction().begin();
QueryParser parser = new QueryParser( getTargetLuceneVersion(), "title", TestConstants.stopAnalyzer );
Query query = parser.parse( "saltQty:noword" );
assertEquals( 0, em.createFullTextQuery( query ).getResultList().size() );
query = new TermQuery( new Term( "saltQty", "23" ) );
assertEquals( "getResultList", 1, em.createFullTextQuery( query ).getResultList().size() );
assertEquals(
"getSingleResult and object retrieval", 23,
( (Bretzel) em.createFullTextQuery( query ).getSingleResult() ).getSaltQty()
);
assertEquals( 1, em.createFullTextQuery( query ).getResultSize() );
em.getTransaction().commit();
em.clear();
em.getTransaction().begin();
em.remove( em.find( Bretzel.class, bretzel.getId() ) );
em.getTransaction().commit();
em.close();
}
@Test
public void testIndex() {
FullTextEntityManager em = Search.getFullTextEntityManager( factory.createEntityManager() );
em.getTransaction().begin();
Bretzel bretzel = new Bretzel( 23, 34 );
em.persist( bretzel );
em.getTransaction().commit();
em.clear();
//Not really a unit test but a test that shows the method call without failing
//FIXME port the index test
em.getTransaction().begin();
em.index( em.find( Bretzel.class, bretzel.getId() ) );
em.getTransaction().commit();
em.getTransaction().begin();
em.remove( em.find( Bretzel.class, bretzel.getId() ) );
em.getTransaction().commit();
em.close();
}
@Override
public Class[] getAnnotatedClasses() {
return new Class[] {
Bretzel.class
};
}
private int countBretzelsViaIndex(FullTextEntityManager em) {
QueryBuilder queryBuilder = em.getSearchFactory().buildQueryBuilder().forEntity( Bretzel.class ).get();
Query allQuery = queryBuilder.all().createQuery();
FullTextQuery fullTextQuery = em.createFullTextQuery( allQuery, Bretzel.class );
return fullTextQuery.getResultSize();
}
}