org.hibernate.search.test.dsl.embedded.DslEmbeddedSearchTest.java Source code

Introduction

Here is the source code for org.hibernate.search.test.dsl.embedded.DslEmbeddedSearchTest.java

Source

/*
 * 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.dsl.embedded;

import java.util.Calendar;
import java.util.Locale;
import java.util.TimeZone;

import org.apache.commons.lang.time.DateUtils;
import org.apache.lucene.search.Query;
import org.hibernate.search.query.dsl.QueryBuilder;
import org.hibernate.search.testsupport.TestForIssue;
import org.hibernate.search.testsupport.junit.SearchFactoryHolder;
import org.hibernate.search.testsupport.junit.SearchITHelper;
import org.hibernate.search.testsupport.junit.SearchITHelper.AssertBuildingHSQueryContext;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;

/**
 * @author Davide D'Alto
 */
public class DslEmbeddedSearchTest {

    private static Calendar initCalendar(int year, int month, int day) {
        Calendar instance = createCalendar();
        instance = DateUtils.truncate(instance, Calendar.DATE);
        instance.set(year, month, day);
        return instance;
    }

    private static Calendar createCalendar() {
        return Calendar.getInstance(TimeZone.getTimeZone("Europe/Rome"), Locale.ITALY);
    }

    @Rule
    public final SearchFactoryHolder sfHolder = new SearchFactoryHolder(ContainerEntity.class);

    private final SearchITHelper helper = new SearchITHelper(sfHolder);

    @Before
    public void setUp() throws Exception {
        EmbeddedEntity ee = new EmbeddedEntity();
        ee.setEmbeddedField("embedded");
        ee.setNumber(7);
        ee.setDate(initCalendar(2007, Calendar.JANUARY, 14).getTime());

        ContainerEntity pe = new ContainerEntity();
        pe.setId(1L);
        pe.setEmbeddedEntity(ee);
        pe.setParentStringValue("theparentvalue");

        helper.add(pe);

        EmbeddedEntity ee2 = new EmbeddedEntity();
        ee2.setEmbeddedField("otherembedded");
        ee2.setNumber(3);
        ee2.setDate(initCalendar(2007, Calendar.JANUARY, 12).getTime());

        ContainerEntity pe2 = new ContainerEntity();
        pe2.setId(2L);
        pe2.setEmbeddedEntity(ee2);
        pe2.setParentStringValue("theotherparentvalue");

        helper.add(pe2);
    }

    @Test
    public void testSearchString() throws Exception {
        QueryBuilder qb = helper.queryBuilder(ContainerEntity.class);
        Query q = qb.keyword().onField("emb.embeddedField").matching("embedded").createQuery();

        assertQuery(q).matchesExactlyIds(1L);
    }

    @Test
    public void testSearchNumberWithFieldBridge() throws Exception {
        QueryBuilder qb = helper.queryBuilder(ContainerEntity.class);
        Query q = qb.keyword().onField("emb.num").matching(7).createQuery();

        assertQuery(q).matchesExactlyIds(1L);
    }

    @Test
    @TestForIssue(jiraKey = "HSEARCH-2070")
    public void testSearchDateWithoutFieldBridge() throws Exception {
        QueryBuilder qb = helper.queryBuilder(ContainerEntity.class);
        Query q = qb.range().onField("emb.date").above(initCalendar(2007, Calendar.JANUARY, 14).getTime())
                .createQuery();

        assertQuery(q).matchesExactlyIds(1L);
    }

    private AssertBuildingHSQueryContext assertQuery(Query q) {
        return helper.assertThat(q).from(ContainerEntity.class);
    }
}