SearchTestUtil.java

/**
 * 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;

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.Hits;
import com.liferay.portal.kernel.search.HitsImpl;
import com.liferay.portal.kernel.search.SearchResult;
import com.liferay.portal.kernel.search.result.SearchResultTranslator;
import com.liferay.portal.kernel.test.util.RandomTestUtil;

import java.util.List;

/**
 * @author Manuel de la Peña
 */
public class SearchTestUtil {

	public static final String ATTACHMENT_OWNER_CLASS_NAME =
		RandomTestUtil.randomString();

	public static final long ATTACHMENT_OWNER_CLASS_NAME_ID =
		RandomTestUtil.randomLong();

	public static final long ATTACHMENT_OWNER_CLASS_PK =
		RandomTestUtil.randomLong();

	public static final long ENTRY_CLASS_PK = RandomTestUtil.randomLong();

	public static final String SUMMARY_CONTENT = RandomTestUtil.randomString();

	public static final String SUMMARY_TITLE = RandomTestUtil.randomString();

	public static Document createAttachmentDocument(String entryClassName) {
		return createAttachmentDocument(entryClassName, ENTRY_CLASS_PK);
	}

	public static Document createAttachmentDocument(
		String entryClassName, long entryClassPK) {

		Document document = createDocument(entryClassName, entryClassPK);

		DocumentHelper documentHelper = new DocumentHelper(document);

		documentHelper.setAttachmentOwnerKey(
			ATTACHMENT_OWNER_CLASS_NAME_ID, ATTACHMENT_OWNER_CLASS_PK);

		return document;
	}

	public static Document createDocument(String entryClassName) {
		return createDocument(entryClassName, ENTRY_CLASS_PK);
	}

	public static Document createDocument(
		String entryClassName, long entryClassPK) {

		Document document = new DocumentImpl();

		DocumentHelper documentHelper = new DocumentHelper(document);

		documentHelper.setEntryKey(entryClassName, entryClassPK);

		return document;
	}

	public static List<SearchResult> getSearchResults(
		SearchResultTranslator searchResultTranslator, Document... documents) {

		Hits hits = new HitsImpl();

		hits.setDocs(documents);

		return searchResultTranslator.translate(hits, null, null, null);
	}

}