lucene-java-3.5.0/lucene/src/test/org/apache/lucene/search/TestPrefixQuery.java

   1 package org.apache.lucene.search;
   2
   3 /**
   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
  10  *
  11  *     http://www.apache.org/licenses/LICENSE-2.0
  12  *
  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.
  18  */
  19
  20 import org.apache.lucene.store.Directory;
  21 import org.apache.lucene.util.LuceneTestCase;
  22 import org.apache.lucene.index.IndexReader;
  23 import org.apache.lucene.index.RandomIndexWriter;
  24 import org.apache.lucene.index.Term;
  25 import org.apache.lucene.document.Document;
  26 import org.apache.lucene.document.Field;
  27
  28 /**
  29  * Tests {@link PrefixQuery} class.
  30  *
  31  */
  32 public class TestPrefixQuery extends LuceneTestCase {
  33   public void testPrefixQuery() throws Exception {
  34     Directory directory = newDirectory();
  35
  36     String[] categories = new String[] {"/Computers",
  37                                         "/Computers/Mac",
  38                                         "/Computers/Windows"};
  39     RandomIndexWriter writer = new RandomIndexWriter(random, directory);
  40     for (int i = 0; i < categories.length; i++) {
  41       Document doc = new Document();
  42       doc.add(newField("category", categories[i], Field.Store.YES, Field.Index.NOT_ANALYZED));
  43       writer.addDocument(doc);
  44     }
  45     IndexReader reader = writer.getReader();
  46
  47     PrefixQuery query = new PrefixQuery(new Term("category", "/Computers"));
  48     IndexSearcher searcher = newSearcher(reader);
  49     ScoreDoc[] hits = searcher.search(query, null, 1000).scoreDocs;
  50     assertEquals("All documents in /Computers category and below", 3, hits.length);
  51
  52     query = new PrefixQuery(new Term("category", "/Computers/Mac"));
  53     hits = searcher.search(query, null, 1000).scoreDocs;
  54     assertEquals("One in /Computers/Mac", 1, hits.length);
  55     writer.close();
  56     searcher.close();
  57     reader.close();
  58     directory.close();
  59   }
  60 }