/*
* 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.embedded.path.defaultdepth;
import java.util.List;
import org.apache.lucene.search.Query;
import org.hibernate.Session;
import org.hibernate.Transaction;
import org.hibernate.search.FullTextSession;
import org.hibernate.search.Search;
import org.hibernate.search.exception.SearchException;
import org.hibernate.search.query.dsl.QueryBuilder;
import org.hibernate.search.test.SearchTestBase;
import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import static org.junit.Assert.fail;
/**
* @author Davide D'Alto
*/
public class DefaultDepthPathCaseEmbeddedTest extends SearchTestBase {
private Session s = null;
private EntityA entityA = null;
@Override
@Before
public void setUp() throws Exception {
super.setUp();
EntityC indexedC = new EntityC( "indexed" );
EntityC skippedC = new EntityC( "indexed" );
EntityB indexedB = new EntityB( indexedC, skippedC );
entityA = new EntityA( indexedB );
s = openSession();
persistEntity( s, indexedC, skippedC, indexedB, entityA );
}
@Override
@After
public void tearDown() throws Exception {
s.clear();
deleteAll( s, EntityA.class, EntityB.class, EntityC.class );
s.close();
super.tearDown();
}
@Test
public void testFieldIsIndexedIfInPath() throws Exception {
List<EntityA> result = search( s, "b.indexed.field", "indexed" );
Assert.assertEquals( 1, result.size() );
Assert.assertEquals( entityA.id, result.get( 0 ).id );
}
@Test
public void testEmbeddedNotIndexedIfNotInPath() throws Exception {
try {
search( s, "b.skipped.indexed", "indexed" );
fail( "Should not index embedded property if not in path and not in depth limit" );
}
catch (SearchException e) {
}
}
@Test
public void testFieldNotIndexedIfNotInPath() throws Exception {
try {
search( s, "b.indexed.skipped", "skipped" );
fail( "Should not index embedded property if not in path and not in depth limit" );
}
catch (SearchException e) {
}
}
private List<EntityA> search(Session s, String field, String value) {
FullTextSession session = Search.getFullTextSession( s );
QueryBuilder queryBuilder = session.getSearchFactory().buildQueryBuilder().forEntity( EntityA.class ).get();
Query query = queryBuilder.keyword().onField( field ).matching( value ).createQuery();
@SuppressWarnings("unchecked")
List<EntityA> result = session.createFullTextQuery( query ).list();
return result;
}
private void deleteAll(Session s, Class<?>... classes) {
Transaction tx = s.beginTransaction();
for ( Class<?> each : classes ) {
List<?> list = s.createCriteria( each ).list();
for ( Object object : list ) {
s.delete( object );
}
}
tx.commit();
}
private void persistEntity(Session s, Object... entities) {
Transaction tx = s.beginTransaction();
for ( Object entity : entities ) {
s.persist( entity );
}
tx.commit();
}
@Override
protected Class<?>[] getAnnotatedClasses() {
return new Class<?>[] { EntityA.class, EntityB.class, EntityC.class };
}
}