+++ /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 java.io.IOException;
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.Iterator;
-
-import junit.framework.Assert;
-
-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.store.Directory;
-import org.apache.lucene.util.LuceneTestCase;
-
-public class TestDocIdSet extends LuceneTestCase {
- public void testFilteredDocIdSet() throws Exception {
- final int maxdoc=10;
- final DocIdSet innerSet = new DocIdSet() {
-
- @Override
- public DocIdSetIterator iterator() {
- return new DocIdSetIterator() {
-
- int docid = -1;
-
- @Override
- public int docID() {
- return docid;
- }
-
- @Override
- public int nextDoc() throws IOException {
- docid++;
- return docid < maxdoc ? docid : (docid = NO_MORE_DOCS);
- }
-
- @Override
- public int advance(int target) throws IOException {
- while (nextDoc() < target) {}
- return docid;
- }
- };
- }
- };
-
-
- DocIdSet filteredSet = new FilteredDocIdSet(innerSet){
- @Override
- protected boolean match(int docid) {
- return docid%2 == 0; //validate only even docids
- }
- };
-
- DocIdSetIterator iter = filteredSet.iterator();
- ArrayList<Integer> list = new ArrayList<Integer>();
- int doc = iter.advance(3);
- if (doc != DocIdSetIterator.NO_MORE_DOCS) {
- list.add(Integer.valueOf(doc));
- while((doc = iter.nextDoc()) != DocIdSetIterator.NO_MORE_DOCS) {
- list.add(Integer.valueOf(doc));
- }
- }
-
- int[] docs = new int[list.size()];
- int c=0;
- Iterator<Integer> intIter = list.iterator();
- while(intIter.hasNext()) {
- docs[c++] = intIter.next().intValue();
- }
- int[] answer = new int[]{4,6,8};
- boolean same = Arrays.equals(answer, docs);
- if (!same) {
- System.out.println("answer: " + Arrays.toString(answer));
- System.out.println("gotten: " + Arrays.toString(docs));
- fail();
- }
- }
-
- public void testNullDocIdSet() throws Exception {
- // Tests that if a Filter produces a null DocIdSet, which is given to
- // IndexSearcher, everything works fine. This came up in LUCENE-1754.
- Directory dir = newDirectory();
- RandomIndexWriter writer = new RandomIndexWriter(random, dir);
- Document doc = new Document();
- doc.add(newField("c", "val", Store.NO, Index.NOT_ANALYZED_NO_NORMS));
- writer.addDocument(doc);
- IndexReader reader = writer.getReader();
- writer.close();
-
- // First verify the document is searchable.
- IndexSearcher searcher = newSearcher(reader);
- Assert.assertEquals(1, searcher.search(new MatchAllDocsQuery(), 10).totalHits);
-
- // Now search w/ a Filter which returns a null DocIdSet
- Filter f = new Filter() {
- @Override
- public DocIdSet getDocIdSet(IndexReader reader) throws IOException {
- return null;
- }
- };
-
- Assert.assertEquals(0, searcher.search(new MatchAllDocsQuery(), f, 10).totalHits);
- searcher.close();
- reader.close();
- dir.close();
- }
-
-}