com.openkm.hibernate.search.SearchTest.java Source code

Introduction

Here is the source code for com.openkm.hibernate.search.SearchTest.java

Source

/**
 *  OpenKM, Open Document Management System (http://www.openkm.com)
 *  Copyright (c) 2006-2017  Paco Avila & Josep Llort
 *
 *  No bytes were intentionally harmed during the development of this application.
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *  
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License along
 *  with this program; if not, write to the Free Software Foundation, Inc.,
 *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

package com.openkm.hibernate.search;

import com.openkm.hibernate.HibernateUtil;
import junit.framework.TestCase;
import org.apache.lucene.analysis.Analyzer;
import org.apache.lucene.analysis.standard.StandardAnalyzer;
import org.apache.lucene.queryParser.ParseException;
import org.apache.lucene.queryParser.QueryParser;
import org.apache.lucene.search.Query;
import org.hibernate.HibernateException;
import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.hibernate.Transaction;
import org.hibernate.cfg.Configuration;
import org.hibernate.search.FullTextQuery;
import org.hibernate.search.FullTextSession;
import org.hibernate.search.Search;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Iterator;
import java.util.UUID;

/**
 * @see http://www.theserverside.com/news/1365222/Unit-Testing-Hibernate-With-HSQLDB
 * @author pavila
 */
public class SearchTest extends TestCase {
    private static Logger log = LoggerFactory.getLogger(SearchTest.class);
    private static String uuid = UUID.randomUUID().toString();

    public SearchTest(String name) {
        super(name);
    }

    public static void main(String[] args) throws Exception {
        SearchTest test = new SearchTest("main");
        test.setUp();
        test.testCreate();
        test.testUpdate();
        test.testSelect();
        test.testSearch();
        test.tearDown();
    }

    public void testCreate() {
        log.info("testCreate()");
        Session session = null;
        Transaction tx = null;

        try {
            session = HibernateUtil.getSessionFactory().openSession();
            tx = session.beginTransaction();

            Document doc = new Document();
            doc.setUuid(uuid);
            doc.setName("pruebas.txt");
            session.save(doc);

            Document dbDoc = (Document) session.get(Document.class, uuid);
            assertNotNull(dbDoc);
            assertEquals(dbDoc.getName(), doc.getName());
            HibernateUtil.commit(tx);
        } catch (HibernateException e) {
            log.error(e.getMessage(), e);
            HibernateUtil.rollback(tx);
            throw e;
        } finally {
            HibernateUtil.close(session);
        }
    }

    public void testUpdate() {
        log.info("testUpdate()");
        Session session = null;
        Transaction tx = null;

        try {
            session = HibernateUtil.getSessionFactory().openSession();
            tx = session.beginTransaction();
            Document doc = (Document) session.load(Document.class, uuid);
            assertNotNull(doc);

            doc.getKeywords().add("alfa");
            doc.getKeywords().add("beta");
            session.update(doc);
            HibernateUtil.commit(tx);
        } catch (HibernateException e) {
            log.error(e.getMessage(), e);
            HibernateUtil.rollback(tx);
            throw e;
        } finally {
            HibernateUtil.close(session);
        }
    }

    public void testSelect() {
        log.info("testSelect()");
        Session session = null;

        try {
            session = HibernateUtil.getSessionFactory().openSession();
            Document doc = (Document) session.load(Document.class, uuid);
            assertNotNull(doc);
            assertEquals(2, doc.getKeywords().size());
            log.info("Keywords: {}", doc.getKeywords());
        } catch (HibernateException e) {
            log.error(e.getMessage(), e);
            throw e;
        } finally {
            HibernateUtil.close(session);
        }
    }

    @SuppressWarnings("unchecked")
    public void testSearch() throws ParseException {
        log.info("testSearch()");
        FullTextSession ftSession = null;
        Session session = null;
        Transaction tx = null;

        try {
            session = HibernateUtil.getSessionFactory().openSession();
            ftSession = Search.getFullTextSession(session);
            tx = ftSession.beginTransaction();

            Analyzer analyzer = new StandardAnalyzer(HibernateUtil.LUCENE_VERSION);
            QueryParser parser = new QueryParser(HibernateUtil.LUCENE_VERSION, "name", analyzer);
            Query query = parser.parse("pruebas.txt");
            log.info("Query: {}", query);
            FullTextQuery ftq = ftSession.createFullTextQuery(query, Document.class);
            assertEquals(1, ftq.list().size());

            for (Iterator<Document> it = ftq.list().iterator(); it.hasNext();) {
                Document doc = it.next();
                log.info("Results: {}", doc);
            }

            query = parser.parse("keyword:alfa");
            log.info("Query: {}", query);
            ftq = ftSession.createFullTextQuery(query, Document.class);
            assertEquals(1, ftq.list().size());

            for (Iterator<Document> it = ftq.list().iterator(); it.hasNext();) {
                Document doc = it.next();
                log.info("Results: {}", doc);
            }

            HibernateUtil.commit(tx);
        } catch (ParseException e) {
            log.error(e.getMessage(), e);
            HibernateUtil.rollback(tx);
            throw e;
        } catch (HibernateException e) {
            log.error(e.getMessage(), e);
            HibernateUtil.rollback(tx);
            throw e;
        } finally {
            HibernateUtil.close(session);
        }
    }

    @Override
    protected void setUp() {
        log.debug("setUp()");
        Configuration cfg = new Configuration();
        cfg.addAnnotatedClass(Document.class);
        SessionFactory sf = HibernateUtil.getSessionFactory(cfg);
        assertNotNull(sf);
    }

    @Override
    protected void tearDown() {
        log.debug("tearDown()");
        //FileUtils.deleteQuietly(new File("DocIndex"));
    }
}