/* * 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. */ package org.apache.lucene.analysis.miscellaneous; import java.io.IOException; import java.io.StringReader; import org.apache.lucene.analysis.Analyzer; import org.apache.lucene.analysis.BaseTokenStreamTestCase; import org.apache.lucene.analysis.TokenStream; import org.apache.lucene.analysis.Tokenizer; import org.apache.lucene.analysis.core.KeywordTokenizer; import org.apache.lucene.util.TestUtil; import org.junit.Test; public class TestCodepointCountFilter extends BaseTokenStreamTestCase { public void testFilterWithPosIncr() throws Exception { TokenStream stream = whitespaceMockTokenizer("short toolong evenmuchlongertext a ab toolong foo"); CodepointCountFilter filter = new CodepointCountFilter(stream, 2, 6); assertTokenStreamContents(filter, new String[]{"short", "ab", "foo"}, new int[]{1, 4, 2} ); } public void testEmptyTerm() throws IOException { Analyzer a = new Analyzer() { @Override protected TokenStreamComponents createComponents(String fieldName) { Tokenizer tokenizer = new KeywordTokenizer(); return new TokenStreamComponents(tokenizer, new CodepointCountFilter(tokenizer, 0, 5)); } }; checkOneTerm(a, "", ""); a.close(); } public void testRandomStrings() throws IOException { for (int i = 0; i < 10000; i++) { String text = TestUtil.randomUnicodeString(random(), 100); int min = TestUtil.nextInt(random(), 0, 100); int max = TestUtil.nextInt(random(), 0, 100); int count = text.codePointCount(0, text.length()); if(min>max){ int temp = min; min = max; max = temp; } boolean expected = count >= min && count <= max; TokenStream stream = new KeywordTokenizer(); ((Tokenizer)stream).setReader(new StringReader(text)); stream = new CodepointCountFilter(stream, min, max); stream.reset(); assertEquals(expected, stream.incrementToken()); stream.end(); stream.close(); } } /** * checking the validity of constructor arguments */ @Test(expected = IllegalArgumentException.class) public void testIllegalArguments() throws Exception { new CodepointCountFilter(whitespaceMockTokenizer("accept only valid arguments"), 4, 1); } }