1 package org.apache.lucene.facet.index.categorypolicy;
3 import org.apache.lucene.store.Directory;
6 import org.apache.lucene.util.LuceneTestCase;
7 import org.apache.lucene.facet.index.categorypolicy.DefaultOrdinalPolicy;
8 import org.apache.lucene.facet.index.categorypolicy.NonTopLevelOrdinalPolicy;
9 import org.apache.lucene.facet.index.categorypolicy.OrdinalPolicy;
10 import org.apache.lucene.facet.taxonomy.CategoryPath;
11 import org.apache.lucene.facet.taxonomy.TaxonomyReader;
12 import org.apache.lucene.facet.taxonomy.TaxonomyWriter;
13 import org.apache.lucene.facet.taxonomy.directory.DirectoryTaxonomyWriter;
16 * Licensed to the Apache Software Foundation (ASF) under one or more
17 * contributor license agreements. See the NOTICE file distributed with
18 * this work for additional information regarding copyright ownership.
19 * The ASF licenses this file to You under the Apache License, Version 2.0
20 * (the "License"); you may not use this file except in compliance with
21 * the License. You may obtain a copy of the License at
23 * http://www.apache.org/licenses/LICENSE-2.0
25 * Unless required by applicable law or agreed to in writing, software
26 * distributed under the License is distributed on an "AS IS" BASIS,
27 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
28 * See the License for the specific language governing permissions and
29 * limitations under the License.
32 public class OrdinalPolicyTest extends LuceneTestCase {
35 public void testDefaultOrdinalPolicy() {
36 // check ordinal policy
37 OrdinalPolicy ordinalPolicy = new DefaultOrdinalPolicy();
38 assertFalse("default ordinal policy should not match root", ordinalPolicy
39 .shouldAdd(TaxonomyReader.ROOT_ORDINAL));
40 for (int i = 0; i < 300; i++) {
41 int ordinal = 1 + random.nextInt(Integer.MAX_VALUE - 1);
42 assertTrue("default ordinal policy should match " + ordinal,
43 ordinalPolicy.shouldAdd(ordinal));
48 public void testNonTopLevelOrdinalPolicy() throws Exception {
49 Directory dir = newDirectory();
50 TaxonomyWriter taxonomy = null;
51 taxonomy = new DirectoryTaxonomyWriter(dir);
53 int[] topLevelOrdinals = new int[10];
54 String[] topLevelStrings = new String[10];
55 for (int i = 0; i < 10; i++) {
56 topLevelStrings[i] = Integer.valueOf(random.nextInt(30)).toString();
57 topLevelOrdinals[i] = taxonomy.addCategory(new CategoryPath(
60 int[] nonTopLevelOrdinals = new int[300];
61 for (int i = 0; i < 300; i++) {
62 int nComponents = 2 + random.nextInt(10);
63 String[] components = new String[nComponents];
64 components[0] = topLevelStrings[i % 10];
65 for (int j = 1; j < components.length; j++) {
66 components[j] = (Integer.valueOf(random.nextInt(30))).toString();
68 nonTopLevelOrdinals[i] = taxonomy.addCategory(new CategoryPath(
71 // check ordinal policy
72 OrdinalPolicy ordinalPolicy = new NonTopLevelOrdinalPolicy();
73 ordinalPolicy.init(taxonomy);
74 assertFalse("top level ordinal policy should not match root", ordinalPolicy
75 .shouldAdd(TaxonomyReader.ROOT_ORDINAL));
76 for (int i = 0; i < 10; i++) {
77 assertFalse("top level ordinal policy should not match "
78 + topLevelOrdinals[i],
79 ordinalPolicy.shouldAdd(topLevelOrdinals[i]));
81 for (int i = 0; i < 300; i++) {
82 assertTrue("top level ordinal policy should match "
83 + nonTopLevelOrdinals[i],
84 ordinalPolicy.shouldAdd(nonTopLevelOrdinals[i]));
87 // check illegal ordinal
88 assertFalse("Should not add illegal ordinal", ordinalPolicy.shouldAdd(100000));