Index: lucene/src/test/org/apache/lucene/search/TestLocaleMethods.java =================================================================== --- lucene/src/test/org/apache/lucene/search/TestLocaleMethods.java (revision 0) +++ lucene/src/test/org/apache/lucene/search/TestLocaleMethods.java (revision 0) @@ -0,0 +1,144 @@ +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); + } + } +} Property changes on: lucene\src\test\org\apache\lucene\search\TestLocaleMethods.java ___________________________________________________________________ Added: svn:eol-style + native