// $Id: WorkDuplicationTest.java 15605 2008-11-23 10:48:59Z hardy.ferentschik $
package org.hibernate.search.test.worker.duplication;
import java.util.List;
import java.util.ArrayList;
import org.apache.lucene.analysis.standard.StandardAnalyzer;
import org.apache.lucene.index.IndexReader;
import org.apache.lucene.queryParser.QueryParser;
import org.apache.lucene.search.Query;
import org.apache.lucene.search.IndexSearcher;
import org.apache.lucene.search.TopDocs;
import org.hibernate.Transaction;
import org.hibernate.search.FullTextQuery;
import org.hibernate.search.FullTextSession;
import org.hibernate.search.backend.WorkType;
import org.hibernate.search.backend.LuceneWork;
import org.hibernate.search.backend.AddLuceneWork;
import org.hibernate.search.backend.DeleteLuceneWork;
import org.hibernate.search.engine.DocumentBuilderIndexedEntity;
import org.hibernate.search.impl.SearchFactoryImpl;
import org.hibernate.search.reader.ReaderProvider;
import org.hibernate.search.store.DirectoryProvider;
import org.hibernate.search.test.SearchTestCase;
/**
* Testcase for HSEARCH-257.
*/
public class WorkDuplicationTest extends SearchTestCase {
/**
* This test assures that HSEARCH-257. Before the fix Search would issue another <code>AddLuceneWork</code> after
* the <code>DeleteLuceneWork</code>. This lead to the fact that after the deletion there was still a Lucene document
* in the index.
*
* @throws Exception in case the test fails.
*/
public void testNoWorkDuplication() throws Exception {
FullTextSession s = org.hibernate.search.Search.getFullTextSession( openSession() );
Transaction tx = s.beginTransaction();
// create new customer
SpecialPerson person = new SpecialPerson();
person.setName( "Joe Smith" );
EmailAddress emailAddress = new EmailAddress();
emailAddress.setAddress( "foo@foobar.com" );
emailAddress.setDefaultAddress(true);
person.addEmailAddress( emailAddress );
// persist the customer
s.persist( person );
tx.commit();
// search if the record made it into the index
tx = s.beginTransaction();
String searchQuery = "Joe";
QueryParser parser = new QueryParser( "Content", new StandardAnalyzer() );
Query luceneQuery = parser.parse( searchQuery );
FullTextQuery query = s.createFullTextQuery( luceneQuery );
List results = query.list();
assertTrue( "We should have a hit", results.size() == 1 );
tx.commit();
// Now try to delete
tx = s.beginTransaction();
int id = person.getId();
person = ( SpecialPerson ) s.get( SpecialPerson.class, id );
s.delete( person );
tx.commit();
// Search and the record via Lucene directly
tx = s.beginTransaction();
DirectoryProvider directoryProvider = s.getSearchFactory().getDirectoryProviders( SpecialPerson.class )[0];
ReaderProvider readerProvider = s.getSearchFactory().getReaderProvider();
IndexReader reader = readerProvider.openReader( directoryProvider );
IndexSearcher searcher = new IndexSearcher( reader );
try {
// we have to test using Lucene directly since query loaders will ignore hits for which there is no
// database entry
TopDocs topDocs = searcher.search( luceneQuery, null, 1 );
assertTrue( "We should have no hit", topDocs.totalHits == 0 );
}
finally {
readerProvider.closeReader( reader );
}
tx.commit();
s.close();
}
/**
* Tests that adding and deleting the same entity only results into a single delete in the work queue.
* See HSEARCH-293.
*
* @throws Exception in case the test fails.
*/
@SuppressWarnings( "unchecked" )
public void testAddWorkGetReplacedByDeleteWork() throws Exception {
FullTextSession fullTextSession = org.hibernate.search.Search.getFullTextSession( openSession() );
SearchFactoryImpl searchFactory = ( SearchFactoryImpl ) fullTextSession.getSearchFactory();
DocumentBuilderIndexedEntity builder = searchFactory.getDocumentBuilderIndexedEntity( SpecialPerson.class );
// create test entity
SpecialPerson person = new SpecialPerson();
person.setName( "Joe Smith" );
EmailAddress emailAddress = new EmailAddress();
emailAddress.setAddress( "foo@foobar.com" );
emailAddress.setDefaultAddress(true);
person.addEmailAddress( emailAddress );
List<LuceneWork> queue = new ArrayList<LuceneWork>();
builder.addWorkToQueue( SpecialPerson.class, person, 1, WorkType.ADD, queue, searchFactory );
assertEquals("There should only be one job in the queue", 1, queue.size());
assertTrue("Wrong job type", queue.get(0) instanceof AddLuceneWork );
builder.addWorkToQueue( SpecialPerson.class, person, 1, WorkType.DELETE, queue, searchFactory );
assertEquals("There should only be one job in the queue", 1, queue.size());
assertTrue("Wrong job type. Add job should have been replaced by delete.", queue.get(0) instanceof DeleteLuceneWork );
fullTextSession.close();
}
protected Class[] getMappings() {
return new Class[] { Person.class, EmailAddress.class, SpecialPerson.class };
}
}