--- /dev/null
+package org.apache.lucene.facet.index.params;
+
+import org.apache.lucene.index.Term;
+import org.junit.Test;
+
+import org.apache.lucene.util.LuceneTestCase;
+import org.apache.lucene.facet.index.categorypolicy.DefaultOrdinalPolicy;
+import org.apache.lucene.facet.index.categorypolicy.DefaultPathPolicy;
+import org.apache.lucene.facet.index.categorypolicy.OrdinalPolicy;
+import org.apache.lucene.facet.index.categorypolicy.PathPolicy;
+import org.apache.lucene.facet.index.params.CategoryListParams;
+import org.apache.lucene.facet.index.params.DefaultFacetIndexingParams;
+import org.apache.lucene.facet.index.params.FacetIndexingParams;
+import org.apache.lucene.facet.search.DrillDown;
+import org.apache.lucene.facet.taxonomy.CategoryPath;
+import org.apache.lucene.facet.taxonomy.TaxonomyReader;
+import org.apache.lucene.facet.util.PartitionsUtils;
+
+/**
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+public class DefaultFacetIndexingParamsTest extends LuceneTestCase {
+
+ @Test
+ public void testDefaultSettings() {
+ FacetIndexingParams dfip = new DefaultFacetIndexingParams();
+ assertNotNull("Missing default category list", dfip
+ .getAllCategoryListParams());
+ assertEquals(
+ "all categories have the same CategoryListParams by default",
+ dfip.getCategoryListParams(null), dfip
+ .getCategoryListParams(new CategoryPath("a")));
+ assertEquals(
+ "Expected default category list term is $facets:$fulltree$",
+ new Term("$facets", "$fulltree$"), dfip.getCategoryListParams(
+ null).getTerm());
+ String expectedDDText = "a"
+ + dfip.getFacetDelimChar() + "b";
+ CategoryPath cp = new CategoryPath("a", "b");
+ assertEquals("wrong drill-down term", new Term("$facets",
+ expectedDDText), DrillDown.term(dfip,cp));
+ char[] buf = new char[20];
+ int numchars = dfip.drillDownTermText(cp, buf);
+ assertEquals("3 characters should be written", 3, numchars);
+ assertEquals("wrong drill-down term text", expectedDDText, new String(
+ buf, 0, numchars));
+ CategoryListParams clParams = dfip.getCategoryListParams(null);
+ assertEquals("partition for all ordinals is the first", "$fulltree$",
+ PartitionsUtils.partitionNameByOrdinal(dfip, clParams , 250));
+ assertEquals("for partition 0, the same name should be returned",
+ "$fulltree$", PartitionsUtils.partitionName(clParams, 0));
+ assertEquals(
+ "for any other, it's the concatenation of name + partition",
+ "$fulltree$1", PartitionsUtils.partitionName(clParams, 1));
+ assertEquals("default partition number is always 0", 0,
+ PartitionsUtils.partitionNumber(dfip,100));
+ assertEquals("default partition size is unbounded", Integer.MAX_VALUE,
+ dfip.getPartitionSize());
+ }
+
+ @Test
+ public void testCategoryListParamsWithDefaultIndexingParams() {
+ CategoryListParams clp = new CategoryListParams(
+ new Term("clp", "value"));
+ FacetIndexingParams dfip = new DefaultFacetIndexingParams(clp);
+ assertEquals("Expected default category list term is " + clp.getTerm(),
+ clp.getTerm(), dfip.getCategoryListParams(null).getTerm());
+ }
+
+ @Test
+ public void testCategoryPolicies() {
+ FacetIndexingParams dfip = new DefaultFacetIndexingParams();
+ long seed = System.currentTimeMillis();
+ // check path policy
+ CategoryPath cp = new CategoryPath();
+ PathPolicy pathPolicy = new DefaultPathPolicy();
+ assertEquals("path policy does not match default for root" + "(seed "
+ + seed + ")", pathPolicy.shouldAdd(cp), dfip.getPathPolicy()
+ .shouldAdd(cp));
+ for (int i = 0; i < 30; i++) {
+ int nComponents = random.nextInt(10);
+ String[] components = new String[nComponents];
+ for (int j = 0; j < components.length; j++) {
+ components[j] = (Integer.valueOf(random.nextInt(30))).toString();
+ }
+ cp = new CategoryPath(components);
+ assertEquals("path policy does not match default for "
+ + cp.toString('/') + "(seed " + seed + ")", pathPolicy
+ .shouldAdd(cp), dfip.getPathPolicy().shouldAdd(cp));
+ }
+
+ // check ordinal policy
+ OrdinalPolicy ordinalPolicy = new DefaultOrdinalPolicy();
+ assertEquals("ordinal policy does not match default for root"
+ + "(seed " + seed + ")", ordinalPolicy
+ .shouldAdd(TaxonomyReader.ROOT_ORDINAL), dfip
+ .getOrdinalPolicy().shouldAdd(TaxonomyReader.ROOT_ORDINAL));
+ for (int i = 0; i < 30; i++) {
+ int ordinal = random.nextInt();
+ assertEquals("ordinal policy does not match default for " + ordinal
+ + "(seed " + seed + ")", ordinalPolicy.shouldAdd(ordinal),
+ dfip.getOrdinalPolicy().shouldAdd(ordinal));
+ }
+ }
+
+}
\ No newline at end of file
fnp / pylucene.git / blobdiff