com.liferay.portal.search.test.util.FacetsAssert Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of com.liferay.portal.search.test.util Show documentation

Liferay Portal Search Test Utilities

There is a newer version: 11.1.1

Show newest version

/**
 * SPDX-FileCopyrightText: (c) 2000 Liferay, Inc. https://liferay.com
 * SPDX-License-Identifier: LGPL-2.1-or-later OR LicenseRef-Liferay-DXP-EULA-2.0.0-2023-06
 */

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

import com.liferay.petra.function.transform.TransformUtil;
import com.liferay.petra.string.StringBundler;
import com.liferay.portal.kernel.search.Hits;
import com.liferay.portal.kernel.search.SearchContext;
import com.liferay.portal.kernel.search.facet.Facet;
import com.liferay.portal.kernel.search.facet.collector.FacetCollector;
import com.liferay.portal.kernel.search.facet.collector.TermCollector;
import com.liferay.portal.kernel.util.StringUtil;

import java.util.List;
import java.util.Map;

import org.junit.Assert;

/**
 * @author André de Oliveira
 */
public class FacetsAssert {

	public static void assertFrequencies(
		String message, Facet facet, String expected) {

		FacetCollector facetCollector = facet.getFacetCollector();

		List termCollectors = facetCollector.getTermCollectors();

		Assert.assertNotNull(termCollectors);

		Assert.assertEquals(
			message, expected,
			String.valueOf(
				TransformUtil.transform(
					termCollectors, FacetsAssert::toString)));
	}

	public static void assertFrequencies(
		String facetName, SearchContext searchContext, Hits hits,
		Map expected) {

		Map facets = searchContext.getFacets();

		Facet facet = facets.get(facetName);

		FacetCollector facetCollector = facet.getFacetCollector();

		AssertUtils.assertEquals(
			StringBundler.concat(
				searchContext.getAttribute("queryString"), "->",
				StringUtil.merge(hits.getDocs())),
			expected,
			TermCollectorUtil.toMap(facetCollector.getTermCollectors()));
	}

	public static void assertFrequencies(
		String facetName, SearchContext searchContext, List expected) {

		assertFrequencies(
			(String)searchContext.getAttribute("queryString"),
			searchContext.getFacet(facetName), String.valueOf(expected));
	}

	protected static String toString(TermCollector termCollector) {
		return termCollector.getTerm() + "=" + termCollector.getFrequency();
	}

}