01: package org.apache.lucene.search;
02:
03: /**
04: * Licensed to the Apache Software Foundation (ASF) under one or more
05: * contributor license agreements. See the NOTICE file distributed with
06: * this work for additional information regarding copyright ownership.
07: * The ASF licenses this file to You under the Apache License, Version 2.0
08: * (the "License"); you may not use this file except in compliance with
09: * 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.util.LuceneTestCase;
21: import org.apache.lucene.store.RAMDirectory;
22: import org.apache.lucene.index.IndexWriter;
23: import org.apache.lucene.index.Term;
24: import org.apache.lucene.analysis.WhitespaceAnalyzer;
25: import org.apache.lucene.document.Document;
26: import org.apache.lucene.document.Field;
27:
28: /**
29: * Tests {@link PrefixQuery} class.
30: *
31: * @author Erik Hatcher
32: */
33: public class TestPrefixQuery extends LuceneTestCase {
34: public void testPrefixQuery() throws Exception {
35: RAMDirectory directory = new RAMDirectory();
36:
37: String[] categories = new String[] { "/Computers",
38: "/Computers/Mac", "/Computers/Windows" };
39: IndexWriter writer = new IndexWriter(directory,
40: new WhitespaceAnalyzer(), true);
41: for (int i = 0; i < categories.length; i++) {
42: Document doc = new Document();
43: doc.add(new Field("category", categories[i],
44: Field.Store.YES, Field.Index.UN_TOKENIZED));
45: writer.addDocument(doc);
46: }
47: writer.close();
48:
49: PrefixQuery query = new PrefixQuery(new Term("category",
50: "/Computers"));
51: IndexSearcher searcher = new IndexSearcher(directory);
52: Hits hits = searcher.search(query);
53: assertEquals("All documents in /Computers category and below",
54: 3, hits.length());
55:
56: query = new PrefixQuery(new Term("category", "/Computers/Mac"));
57: hits = searcher.search(query);
58: assertEquals("One in /Computers/Mac", 1, hits.length());
59: }
60: }
|