LPS-83506 portal-search-test-util: added searchCount test case for Fa…

…cetSelection https://issues.liferay.com/browse/LPS-83506
arboliveira · Aug 6, 2018 · 193a587 · 193a587
1 parent 7522ba6
commit 193a587
Show file tree

Hide file tree

Showing 3 changed files with 54 additions and 0 deletions.
diff --git a/...-test-util/src/main/java/com/liferay/portal/search/test/util/facet/BaseFacetTestCase.java b/...-test-util/src/main/java/com/liferay/portal/search/test/util/facet/BaseFacetTestCase.java
@@ -83,6 +83,10 @@ protected Hits doSearch(
 		}
 	}
 
+	protected long doSearchCount(SearchContext searchContext) {
+		return searchCount(searchContext);
+	}
+
 	protected abstract String getField();
 
 	protected Facet initFacet(Facet facet) {
@@ -131,6 +135,10 @@ public void search(QueryContributor queryContributor) {
 			_hits = doSearch(_searchContext, queryContributor);
 		}
 
+		public long searchCount() {
+			return doSearchCount(_searchContext);
+		}
+
 		public void setSearchContextAttribute(String name, Serializable value) {
 			_searchContext.setAttribute(name, value);
 		}

diff --git a/...util/src/main/java/com/liferay/portal/search/test/util/facet/BaseSimpleFacetTestCase.java b/...util/src/main/java/com/liferay/portal/search/test/util/facet/BaseSimpleFacetTestCase.java
@@ -25,6 +25,7 @@
 
 import java.util.Arrays;
 
+import org.junit.Assert;
 import org.junit.Test;
 
 /**
@@ -135,6 +136,26 @@ public void testMaxTermsZero() throws Exception {
 			});
 	}
 
+	@Test
+	public void testSearchCountSelection() throws Exception {
+		addDocuments(6, "one");
+		addDocuments(5, "two");
+		addDocuments(4, "three");
+		addDocuments(3, "four");
+		addDocuments(2, "five");
+
+		assertSearchFacet(
+			helper -> {
+				Facet facet = helper.addFacet(this::createFacet);
+
+				select(facet, "three", helper);
+
+				long count = helper.searchCount();
+
+				Assert.assertEquals(4, count);
+			});
+	}
+
 	@Test
 	public void testSelection() throws Exception {
 		addDocuments(6, "one");

diff --git a/...util/src/main/java/com/liferay/portal/search/test/util/indexing/BaseIndexingTestCase.java b/...util/src/main/java/com/liferay/portal/search/test/util/indexing/BaseIndexingTestCase.java
@@ -195,6 +195,31 @@ protected Hits search(
 		return search(searchContext, _getQuery(queryContributor));
 	}
 
+	protected long searchCount(SearchContext searchContext) {
+		return searchCount(searchContext, getDefaultQuery());
+	}
+
+	protected long searchCount(SearchContext searchContext, Query query) {
+		try {
+			return _indexSearcher.searchCount(searchContext, query);
+		}
+		catch (SearchException se) {
+			Throwable t = se.getCause();
+
+			if (t instanceof RuntimeException) {
+				throw (RuntimeException)t;
+			}
+
+			throw new RuntimeException(se);
+		}
+	}
+
+	protected long searchCount(
+		SearchContext searchContext, QueryContributor queryContributor) {
+
+		return searchCount(searchContext, _getQuery(queryContributor));
+	}
+
 	protected void setPreBooleanFilter(Filter filter, Query query) {
 		BooleanFilter booleanFilter = new BooleanFilter();