+++ /dev/null
-package org.apache.lucene.search;
-
-import java.io.IOException;
-import java.text.Collator;
-import java.util.Locale;
-
-import org.apache.lucene.document.Document;
-import org.apache.lucene.document.Field;
-import org.apache.lucene.index.IndexReader;
-import org.apache.lucene.index.RandomIndexWriter;
-import org.apache.lucene.search.BooleanClause.Occur;
-import org.apache.lucene.store.Directory;
-import org.apache.lucene.util.LuceneTestCase;
-import org.apache.lucene.util._TestUtil;
-import org.junit.AfterClass;
-import org.junit.BeforeClass;
-
-/**
- * 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.
- */
-
-/**
- * Tests Locale-based sort and range search
- */
-public class TestLocaleMethods extends LuceneTestCase {
- private static Locale locale;
- private static Collator collator;
- private static IndexSearcher searcher;
- private static IndexReader reader;
- private static Directory dir;
- private static int numDocs;
-
- @BeforeClass
- public static void beforeClass() throws Exception {
- locale = LuceneTestCase.randomLocale(random);
- collator = Collator.getInstance(locale);
- numDocs = 1000 * RANDOM_MULTIPLIER;
- dir = newDirectory();
- RandomIndexWriter iw = new RandomIndexWriter(random, dir);
- for (int i = 0; i < numDocs; i++) {
- Document doc = new Document();
- String value = _TestUtil.randomUnicodeString(random);
- Field field = newField("field", value, Field.Store.YES, Field.Index.NOT_ANALYZED_NO_NORMS);
- doc.add(field);
- iw.addDocument(doc);
- }
- reader = iw.getReader();
- iw.close();
-
- searcher = newSearcher(reader);
- }
-
- @AfterClass
- public static void afterClass() throws Exception {
- searcher.close();
- reader.close();
- dir.close();
- locale = null;
- collator = null;
- searcher = null;
- reader = null;
- dir = null;
- }
-
- public void testSort() throws Exception {
- SortField sf = new SortField("field", locale);
- TopFieldDocs docs = searcher.search(new MatchAllDocsQuery(), null, numDocs, new Sort(sf));
- String prev = "";
- for (ScoreDoc doc : docs.scoreDocs) {
- String value = reader.document(doc.doc).get("field");
- assertTrue(collator.compare(value, prev) >= 0);
- prev = value;
- }
- }
-
- public void testSort2() throws Exception {
- SortField sf = new SortField("field", new FieldComparatorSource() {
- @Override
- public FieldComparator newComparator(String fieldname, int numHits, int sortPos, boolean reversed) throws IOException {
- return new FieldComparator.StringComparatorLocale(numHits, fieldname, locale);
- }
- });
- TopFieldDocs docs = searcher.search(new MatchAllDocsQuery(), null, numDocs, new Sort(sf));
- String prev = "";
- for (ScoreDoc doc : docs.scoreDocs) {
- String value = reader.document(doc.doc).get("field");
- assertTrue(collator.compare(value, prev) >= 0);
- prev = value;
- }
- }
-
- private void doTestRanges(String startPoint, String endPoint, Query query) throws Exception {
- // positive test
- TopDocs docs = searcher.search(query, numDocs);
- for (ScoreDoc doc : docs.scoreDocs) {
- String value = reader.document(doc.doc).get("field");
- assertTrue(collator.compare(value, startPoint) >= 0);
- assertTrue(collator.compare(value, endPoint) <= 0);
- }
-
- // negative test
- BooleanQuery bq = new BooleanQuery();
- bq.add(new MatchAllDocsQuery(), Occur.SHOULD);
- bq.add(query, Occur.MUST_NOT);
- docs = searcher.search(bq, numDocs);
- for (ScoreDoc doc : docs.scoreDocs) {
- String value = reader.document(doc.doc).get("field");
- assertTrue(collator.compare(value, startPoint) < 0 || collator.compare(value, endPoint) > 0);
- }
- }
-
- public void testRangeQuery() throws Exception {
- int numQueries = 100*RANDOM_MULTIPLIER;
- for (int i = 0; i < numQueries; i++) {
- String startPoint = _TestUtil.randomUnicodeString(random);
- String endPoint = _TestUtil.randomUnicodeString(random);
- Query query = new TermRangeQuery("field", startPoint, endPoint, true, true, collator);
- doTestRanges(startPoint, endPoint, query);
- }
- }
-
- public void testRangeFilter() throws Exception {
- int numQueries = 100*RANDOM_MULTIPLIER;
- for (int i = 0; i < numQueries; i++) {
- String startPoint = _TestUtil.randomUnicodeString(random);
- String endPoint = _TestUtil.randomUnicodeString(random);
- Query query = new ConstantScoreQuery(new TermRangeFilter("field", startPoint, endPoint, true, true, collator));
- doTestRanges(startPoint, endPoint, query);
- }
- }
-}