pylucene 3.5.0-3

[pylucene.git] / lucene-java-3.5.0 / lucene / contrib / facet / src / test / org / apache / lucene / facet / search / TestCategoryListCache.java

diff --git a/lucene-java-3.5.0/lucene/contrib/facet/src/test/org/apache/lucene/facet/search/TestCategoryListCache.java b/lucene-java-3.5.0/lucene/contrib/facet/src/test/org/apache/lucene/facet/search/TestCategoryListCache.java
new file mode 100644 (file)
index 0000000..f77b709
--- /dev/null
+++ b/lucene-java-3.5.0/lucene/contrib/facet/src/test/org/apache/lucene/facet/search/TestCategoryListCache.java
@@ -0,0 +1,132 @@
+package org.apache.lucene.facet.search;
+
+import java.io.IOException;
+import java.util.List;
+import java.util.Map;
+
+import org.apache.lucene.search.MatchAllDocsQuery;
+
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+
+import org.apache.lucene.facet.FacetTestBase;
+import org.apache.lucene.facet.index.params.CategoryListParams;
+import org.apache.lucene.facet.index.params.FacetIndexingParams;
+import org.apache.lucene.facet.search.cache.CategoryListCache;
+import org.apache.lucene.facet.search.cache.CategoryListData;
+import org.apache.lucene.facet.search.params.CountFacetRequest;
+import org.apache.lucene.facet.search.params.FacetSearchParams;
+import org.apache.lucene.facet.search.results.FacetResult;
+import org.apache.lucene.facet.taxonomy.CategoryPath;
+
+/**
+ * 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 TestCategoryListCache extends FacetTestBase {
+
+  public TestCategoryListCache() {
+    super();
+  }
+  
+  @Before
+  @Override
+  public void setUp() throws Exception {
+    super.setUp();
+    initIndex();
+  }
+  
+  @After
+  @Override
+  public void tearDown() throws Exception {
+    closeAll();
+    super.tearDown();
+  }
+  
+  @Test
+  public void testNoClCache() throws Exception {
+    doTest(false,false);
+  }
+
+  @Test
+  public void testCorrectClCache() throws Exception {
+    doTest(true,false);
+  }
+  
+  @Test
+  public void testWrongClCache() throws Exception {
+    doTest(true,true);
+  }
+  
+  private void doTest(boolean withCache, boolean plantWrongData) throws IOException, Exception {
+    Map<CategoryPath,Integer> truth = facetCountsTruth();
+    CategoryPath cp = (CategoryPath) truth.keySet().toArray()[0]; // any category path will do for this test 
+    CountFacetRequest frq = new CountFacetRequest(cp, 10);
+    FacetSearchParams sParams = getFacetedSearchParams();
+    sParams.addFacetRequest(frq);
+    if (withCache) {
+      //let's use a cached cl data
+      FacetIndexingParams iparams = sParams.getFacetIndexingParams();
+      CategoryListParams clp = new CategoryListParams(); // default term ok as only single list
+      CategoryListCache clCache = new CategoryListCache();
+      clCache.loadAndRegister(clp, indexReader, taxoReader, iparams);
+      if (plantWrongData) {
+        // let's mess up the cached data and then expect a wrong result...
+        messCachedData(clCache, clp);
+      }
+      sParams.setClCache(clCache);
+    }
+    FacetsCollector fc = new FacetsCollector(sParams, indexReader, taxoReader);
+    searcher.search(new MatchAllDocsQuery(), fc);
+    List<FacetResult> res = fc.getFacetResults();
+    try {
+      assertCountsAndCardinality(truth, res);
+      assertFalse("Correct results not expected when wrong data was cached", plantWrongData);
+    } catch (Throwable e) {
+      assertTrue("Wrong results not expected unless wrong data was cached", withCache);
+      assertTrue("Wrong results not expected unless wrong data was cached", plantWrongData);
+    }
+  }
+
+  /** Mess the cached data for this {@link CategoryListParams} */
+  private void messCachedData(CategoryListCache clCache, CategoryListParams clp) {
+    final CategoryListData cld = clCache.get(clp);
+    CategoryListData badCld = new CategoryListData() {
+      @Override
+      public CategoryListIterator iterator(int partition)  throws IOException {
+        final CategoryListIterator it = cld.iterator(partition);
+        return new CategoryListIterator() {              
+          public boolean skipTo(int docId) throws IOException {
+            return it.skipTo(docId);
+          }
+          public long nextCategory() throws IOException {
+            long res = it.nextCategory();
+            if (res>Integer.MAX_VALUE) {
+              return res;
+            }
+            return res>1 ? res-1 : res+1;
+          }
+          public boolean init() throws IOException {
+            return it.init();
+          }
+        };
+      }
+    };
+    clCache.register(clp, badCld);
+  }
+  
+}