/*
* 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.query;
import static org.fest.assertions.Assertions.assertThat;
import java.util.Calendar;
import java.util.Date;
import java.util.List;
import java.util.Locale;
import javax.persistence.EntityTransaction;
import org.apache.lucene.queryparser.classic.ParseException;
import org.apache.lucene.queryparser.classic.QueryParser;
import org.apache.lucene.search.Sort;
import org.apache.lucene.search.SortField;
import org.hibernate.search.jpa.FullTextEntityManager;
import org.hibernate.search.jpa.FullTextQuery;
import org.hibernate.search.testsupport.TestConstants;
import org.hibernate.search.test.jpa.JPATestCase;
import org.junit.Test;
/**
* @author Davide D'Alto <davide@hibernate.org>
*/
public class SortUsingEntityManagerTest extends JPATestCase {
private static final boolean DESC = true;
private FullTextEntityManager em;
private QueryParser queryParser;
@Override
public void setUp() {
super.setUp();
em = org.hibernate.search.jpa.Search.getFullTextEntityManager( factory.createEntityManager() );
queryParser = new QueryParser( TestConstants.getTargetLuceneVersion(), "header", TestConstants.stopAnalyzer );
createArticles();
}
private void createArticles() {
EntityTransaction tx = em.getTransaction();
tx.begin();
try {
em.persist( article( 1, "Hibernate & Lucene", date( 4, Calendar.JULY, 2005 ) ) );
em.persist( article( 2, "Hibernate Search", date( 2, Calendar.SEPTEMBER, 2005 ) ) );
em.persist( article( 3, "Hibernate OGM", date( 4, Calendar.SEPTEMBER, 2005 ) ) );
em.persist( article( 4, "Hibernate", date( 4, Calendar.DECEMBER, 2005 ) ) );
em.persist( article( 5, "Hibernate Validator", date( 8, Calendar.SEPTEMBER, 2010 ) ) );
em.persist( article( 6, "Hibernate Core", date( 4, Calendar.SEPTEMBER, 2012 ) ) );
}
finally {
tx.commit();
em.clear();
}
}
@Test
@SuppressWarnings("unchecked")
public void testResultOrderedByDateDescending() throws Exception {
EntityTransaction tx = em.getTransaction();
tx.begin();
Sort dateDescending = new Sort( new SortField( "creationDate", SortField.Type.STRING, DESC ) );
List<ProductArticle> result = query( "Hib*" ).setSort( dateDescending ).setFirstResult( 3 ).getResultList();
assertThat( result ).as( "query result" ).hasSize( 3 );
assertThat( result.get( 0 ).getArticleId() ).as( "article id" ).isEqualTo( 3 );
assertThat( result.get( 1 ).getArticleId() ).as( "article id" ).isEqualTo( 2 );
assertThat( result.get( 2 ).getArticleId() ).as( "article id" ).isEqualTo( 1 );
tx.commit();
em.clear();
}
@Override
public void tearDown() {
deleteArticles();
super.tearDown();
}
private FullTextQuery query(String q) throws ParseException {
org.apache.lucene.search.Query query = queryParser.parse( q );
return em.createFullTextQuery( query, ProductArticle.class );
}
private Date date(int day, int month, int year) {
Calendar cal = Calendar.getInstance( Locale.US );
cal.set( year, month, day, 11, 05, 30 );
Date date = cal.getTime();
return date;
}
private ProductArticle article(int id, String header, Date date) {
ProductArticle article = new ProductArticle();
article.setCreationDate( date );
article.setHeader( header );
article.setArticleId( Integer.valueOf( id ) );
return article;
}
private void deleteArticles() {
EntityTransaction tx = em.getTransaction();
tx.begin();
em.createQuery( "delete " + ProductArticle.class.getName() ).executeUpdate();
tx.commit();
em.clear();
}
@Override
public Class[] getAnnotatedClasses() {
return new Class[] { ProductArticle.class };
}
}