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.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;
29 * Tests {@link PrefixQuery} class.
32 public class TestPrefixQuery extends LuceneTestCase {
33 public void testPrefixQuery() throws Exception {
34 Directory directory = newDirectory();
36 String[] categories = new String[] {"/Computers",
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);
45 IndexReader reader = writer.getReader();
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);
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);