--- /dev/null
+package org.apache.lucene.search;
+
+/**
+ * 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.
+ */
+
+import org.apache.lucene.document.Document;
+import org.apache.lucene.document.Field.Index;
+import org.apache.lucene.document.Field.Store;
+import org.apache.lucene.index.IndexReader;
+import org.apache.lucene.index.RandomIndexWriter;
+import org.apache.lucene.index.Term;
+import org.apache.lucene.search.BooleanClause.Occur;
+import org.apache.lucene.store.Directory;
+import org.apache.lucene.util.LuceneTestCase;
+
+public class TestQueryWrapperFilter extends LuceneTestCase {
+
+ public void testBasic() throws Exception {
+ Directory dir = newDirectory();
+ RandomIndexWriter writer = new RandomIndexWriter(random, dir);
+ Document doc = new Document();
+ doc.add(newField("field", "value", Store.NO, Index.ANALYZED));
+ writer.addDocument(doc);
+ IndexReader reader = writer.getReader();
+ writer.close();
+
+ TermQuery termQuery = new TermQuery(new Term("field", "value"));
+
+ // should not throw exception with primitive query
+ QueryWrapperFilter qwf = new QueryWrapperFilter(termQuery);
+
+ IndexSearcher searcher = newSearcher(reader);
+ TopDocs hits = searcher.search(new MatchAllDocsQuery(), qwf, 10);
+ assertEquals(1, hits.totalHits);
+ hits = searcher.search(new MatchAllDocsQuery(), new CachingWrapperFilter(qwf), 10);
+ assertEquals(1, hits.totalHits);
+
+ // should not throw exception with complex primitive query
+ BooleanQuery booleanQuery = new BooleanQuery();
+ booleanQuery.add(termQuery, Occur.MUST);
+ booleanQuery.add(new TermQuery(new Term("field", "missing")),
+ Occur.MUST_NOT);
+ qwf = new QueryWrapperFilter(termQuery);
+
+ hits = searcher.search(new MatchAllDocsQuery(), qwf, 10);
+ assertEquals(1, hits.totalHits);
+ hits = searcher.search(new MatchAllDocsQuery(), new CachingWrapperFilter(qwf), 10);
+ assertEquals(1, hits.totalHits);
+
+ // should not throw exception with non primitive Query (doesn't implement
+ // Query#createWeight)
+ qwf = new QueryWrapperFilter(new FuzzyQuery(new Term("field", "valu")));
+
+ hits = searcher.search(new MatchAllDocsQuery(), qwf, 10);
+ assertEquals(1, hits.totalHits);
+ hits = searcher.search(new MatchAllDocsQuery(), new CachingWrapperFilter(qwf), 10);
+ assertEquals(1, hits.totalHits);
+
+ // test a query with no hits
+ termQuery = new TermQuery(new Term("field", "not_exist"));
+ qwf = new QueryWrapperFilter(termQuery);
+ hits = searcher.search(new MatchAllDocsQuery(), qwf, 10);
+ assertEquals(0, hits.totalHits);
+ hits = searcher.search(new MatchAllDocsQuery(), new CachingWrapperFilter(qwf), 10);
+ assertEquals(0, hits.totalHits);
+ searcher.close();
+ reader.close();
+ dir.close();
+ }
+}