com.liferay.portal.search.test.util.indexing.DocumentFixture.java Source code

Introduction

Here is the source code for com.liferay.portal.search.test.util.indexing.DocumentFixture.java

Source

/**
 * Copyright (c) 2000-present Liferay, Inc. All rights reserved.
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by the Free
 * Software Foundation; either version 2.1 of the License, or (at your option)
 * any later version.
 *
 * This library 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 Lesser General Public License for more
 * details.
 */

package com.liferay.portal.search.test.util.indexing;

import com.liferay.portal.kernel.search.Document;
import com.liferay.portal.kernel.search.DocumentHelper;
import com.liferay.portal.kernel.search.DocumentImpl;
import com.liferay.portal.kernel.search.Field;
import com.liferay.portal.kernel.test.util.PropsTestUtil;
import com.liferay.portal.kernel.test.util.RandomTestUtil;
import com.liferay.portal.kernel.util.FastDateFormatFactory;
import com.liferay.portal.kernel.util.FastDateFormatFactoryUtil;
import com.liferay.portal.kernel.util.HashMapBuilder;
import com.liferay.portal.kernel.util.Props;
import com.liferay.portal.kernel.util.PropsKeys;
import com.liferay.portal.kernel.util.PropsUtil;

import java.text.SimpleDateFormat;

import java.util.Map;

import org.mockito.Mockito;

/**
 * @author Miguel Angelo Caldas Gallindo
 */
public class DocumentFixture {

    public static Document newDocument(long companyId, long groupId, String entryClassName) {

        DocumentImpl documentImpl = new DocumentImpl();

        documentImpl.addKeyword(Field.COMPANY_ID, companyId);
        documentImpl.addKeyword(Field.GROUP_ID, groupId);

        long entryClassPK = RandomTestUtil.randomLong();

        documentImpl.addUID(entryClassName, entryClassPK);

        DocumentHelper documentHelper = new DocumentHelper(documentImpl);

        documentHelper.setEntryKey(entryClassName, entryClassPK);

        return documentImpl;
    }

    public void setUp() {
        setUpFastDateFormatFactoryUtil();
        setUpPropsUtil();
    }

    public void tearDown() {
        tearDownFastDateFormatFactoryUtil();
        tearDownPropsUtil();
    }

    protected void setUpFastDateFormatFactoryUtil() {
        _fastDateFormatFactory = FastDateFormatFactoryUtil.getFastDateFormatFactory();

        FastDateFormatFactoryUtil fastDateFormatFactoryUtil = new FastDateFormatFactoryUtil();

        FastDateFormatFactory fastDateFormatFactory = Mockito.mock(FastDateFormatFactory.class);

        Mockito.when(fastDateFormatFactory.getSimpleDateFormat("yyyyMMddHHmmss"))
                .thenReturn(new SimpleDateFormat("yyyyMMddHHmmss"));

        fastDateFormatFactoryUtil.setFastDateFormatFactory(fastDateFormatFactory);
    }

    protected void setUpPropsUtil() {
        Map<String, Object> properties = HashMapBuilder
                .<String, Object>put(PropsKeys.INDEX_DATE_FORMAT_PATTERN, "yyyyMMddHHmmss")
                .put(PropsKeys.INDEX_SEARCH_COLLATED_SPELL_CHECK_RESULT_ENABLED, "true")
                .put(PropsKeys.INDEX_SEARCH_COLLATED_SPELL_CHECK_RESULT_SCORES_THRESHOLD, "50")
                .put(PropsKeys.INDEX_SEARCH_HIGHLIGHT_FRAGMENT_SIZE, "80")
                .put(PropsKeys.INDEX_SEARCH_HIGHLIGHT_REQUIRE_FIELD_MATCH, "true")
                .put(PropsKeys.INDEX_SEARCH_HIGHLIGHT_SNIPPET_SIZE, "3")
                .put(PropsKeys.INDEX_SEARCH_QUERY_INDEXING_ENABLED, "true")
                .put(PropsKeys.INDEX_SEARCH_QUERY_INDEXING_THRESHOLD, "50")
                .put(PropsKeys.INDEX_SEARCH_QUERY_SUGGESTION_ENABLED, "true")
                .put(PropsKeys.INDEX_SEARCH_QUERY_SUGGESTION_MAX, "yyyyMMddHHmmss")
                .put(PropsKeys.INDEX_SEARCH_QUERY_SUGGESTION_SCORES_THRESHOLD, "0")
                .put(PropsKeys.INDEX_SEARCH_SCORING_ENABLED, "true")
                .put(PropsKeys.INDEX_SORTABLE_TEXT_FIELDS_TRUNCATED_LENGTH, "255").build();

        props = PropsTestUtil.setProps(properties);
    }

    protected void tearDownFastDateFormatFactoryUtil() {
        FastDateFormatFactoryUtil fastDateFormatFactoryUtil = new FastDateFormatFactoryUtil();

        fastDateFormatFactoryUtil.setFastDateFormatFactory(_fastDateFormatFactory);

        _fastDateFormatFactory = null;
    }

    protected void tearDownPropsUtil() {
        PropsUtil.setProps(_props);

        _props = null;

        props = null;
    }

    protected Props props;

    private FastDateFormatFactory _fastDateFormatFactory;
    private Props _props;

}