1 package org.apache.lucene.search;
4 * Licensed to the Apache Software Foundation (ASF) under one or more
5 * contributor license agreements. See the NOTICE file distributed with
6 * this work for additional information regarding copyright ownership.
7 * The ASF licenses this file to You under the Apache License, Version 2.0
8 * (the "License"); you may not use this file except in compliance with
9 * the License. You may obtain a copy of the License at
11 * http://www.apache.org/licenses/LICENSE-2.0
13 * Unless required by applicable law or agreed to in writing, software
14 * distributed under the License is distributed on an "AS IS" BASIS,
15 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 * See the License for the specific language governing permissions and
17 * limitations under the License.
20 import org.apache.lucene.analysis.SimpleAnalyzer;
21 import org.apache.lucene.document.Document;
22 import org.apache.lucene.document.Field;
23 import org.apache.lucene.index.IndexWriter;
24 import org.apache.lucene.index.Term;
25 import org.apache.lucene.store.Directory;
26 import org.junit.AfterClass;
27 import org.junit.BeforeClass;
28 import org.junit.Test;
31 * Tests that the index is cached on the searcher side of things.
33 public class TestRemoteCachingWrapperFilter extends RemoteTestCase {
34 private static Directory indexStore;
35 private static Searchable local;
38 public static void beforeClass() throws Exception {
40 indexStore = newDirectory();
41 IndexWriter writer = new IndexWriter(indexStore, newIndexWriterConfig(
42 TEST_VERSION_CURRENT, new SimpleAnalyzer(
43 TEST_VERSION_CURRENT)));
44 Document doc = new Document();
45 doc.add(newField("test", "test text", Field.Store.YES, Field.Index.ANALYZED));
46 doc.add(newField("type", "A", Field.Store.YES, Field.Index.ANALYZED));
47 doc.add(newField("other", "other test text", Field.Store.YES, Field.Index.ANALYZED));
48 writer.addDocument(doc);
49 //Need a second document to search for
51 doc.add(newField("test", "test text", Field.Store.YES, Field.Index.ANALYZED));
52 doc.add(newField("type", "B", Field.Store.YES, Field.Index.ANALYZED));
53 doc.add(newField("other", "other test text", Field.Store.YES, Field.Index.ANALYZED));
54 writer.addDocument(doc);
57 local = new IndexSearcher(indexStore, true);
62 public static void afterClass() throws Exception {
68 private static void search(Query query, Filter filter, int hitNumber, String typeValue) throws Exception {
69 Searchable[] searchables = { lookupRemote() };
70 Searcher searcher = new MultiSearcher(searchables);
71 ScoreDoc[] result = searcher.search(query,filter, 1000).scoreDocs;
72 assertEquals(1, result.length);
73 Document document = searcher.doc(result[hitNumber].doc);
74 assertTrue("document is null and it shouldn't be", document != null);
75 assertEquals(typeValue, document.get("type"));
76 assertTrue("document.getFields() Size: " + document.getFields().size() + " is not: " + 3, document.getFields().size() == 3);
80 public void testTermRemoteFilter() throws Exception {
81 CachingWrapperFilterHelper cwfh = new CachingWrapperFilterHelper(new QueryWrapperFilter(new TermQuery(new Term("type", "a"))));
83 // This is what we are fixing - if one uses a CachingWrapperFilter(Helper) it will never
84 // cache the filter on the remote site
85 cwfh.setShouldHaveCache(false);
86 search(new TermQuery(new Term("test", "test")), cwfh, 0, "A");
87 cwfh.setShouldHaveCache(false);
88 search(new TermQuery(new Term("test", "test")), cwfh, 0, "A");
90 // This is how we fix caching - we wrap a Filter in the RemoteCachingWrapperFilter(Handler - for testing)
91 // to cache the Filter on the searcher (remote) side
92 RemoteCachingWrapperFilterHelper rcwfh = new RemoteCachingWrapperFilterHelper(cwfh, false);
93 search(new TermQuery(new Term("test", "test")), rcwfh, 0, "A");
95 // 2nd time we do the search, we should be using the cached Filter
96 rcwfh.shouldHaveCache(true);
97 search(new TermQuery(new Term("test", "test")), rcwfh, 0, "A");
99 // assert that we get the same cached Filter, even if we create a new instance of RemoteCachingWrapperFilter(Helper)
100 // this should pass because the Filter parameters are the same, and the cache uses Filter's hashCode() as cache keys,
101 // and Filters' hashCode() builds on Filter parameters, not the Filter instance itself
102 rcwfh = new RemoteCachingWrapperFilterHelper(new QueryWrapperFilter(new TermQuery(new Term("type", "a"))), false);
103 rcwfh.shouldHaveCache(false);
104 search(new TermQuery(new Term("test", "test")), rcwfh, 0, "A");
106 rcwfh = new RemoteCachingWrapperFilterHelper(new QueryWrapperFilter(new TermQuery(new Term("type", "a"))), false);
107 rcwfh.shouldHaveCache(true);
108 search(new TermQuery(new Term("test", "test")), rcwfh, 0, "A");
110 // assert that we get a non-cached version of the Filter because this is a new Query (type:b)
111 rcwfh = new RemoteCachingWrapperFilterHelper(new QueryWrapperFilter(new TermQuery(new Term("type", "b"))), false);
112 rcwfh.shouldHaveCache(false);
113 search(new TermQuery(new Term("type", "b")), rcwfh, 0, "B");