X-Git-Url: https://git.mdrn.pl/pylucene.git/blobdiff_plain/a2e61f0c04805cfcb8706176758d1283c7e3a55c..aaeed5504b982cf3545252ab528713250aa33eed:/lucene-java-3.5.0/lucene/contrib/xml-query-parser/src/test/org/apache/lucene/xmlparser/builders/TestNumericRangeFilterBuilder.java diff --git a/lucene-java-3.5.0/lucene/contrib/xml-query-parser/src/test/org/apache/lucene/xmlparser/builders/TestNumericRangeFilterBuilder.java b/lucene-java-3.5.0/lucene/contrib/xml-query-parser/src/test/org/apache/lucene/xmlparser/builders/TestNumericRangeFilterBuilder.java new file mode 100644 index 0000000..dca574d --- /dev/null +++ b/lucene-java-3.5.0/lucene/contrib/xml-query-parser/src/test/org/apache/lucene/xmlparser/builders/TestNumericRangeFilterBuilder.java @@ -0,0 +1,215 @@ +package org.apache.lucene.xmlparser.builders; + +/** + * 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. + */ + +import java.io.ByteArrayInputStream; +import java.io.IOException; +import java.io.InputStream; + +import javax.xml.parsers.DocumentBuilder; +import javax.xml.parsers.DocumentBuilderFactory; +import javax.xml.parsers.ParserConfigurationException; + +import org.apache.lucene.util.LuceneTestCase; + +import org.apache.lucene.index.IndexReader; +import org.apache.lucene.index.IndexWriter; +import org.apache.lucene.search.Filter; +import org.apache.lucene.search.NumericRangeFilter; +import org.apache.lucene.store.Directory; +import org.apache.lucene.xmlparser.ParserException; +import org.w3c.dom.Document; +import org.xml.sax.SAXException; + +public class TestNumericRangeFilterBuilder extends LuceneTestCase { + + public void testGetFilterHandleNumericParseErrorStrict() throws Exception { + NumericRangeFilterBuilder filterBuilder = new NumericRangeFilterBuilder(); + filterBuilder.setStrictMode(true); + + String xml = ""; + Document doc = getDocumentFromString(xml); + try { + filterBuilder.getFilter(doc.getDocumentElement()); + } catch (ParserException e) { + return; + } + fail("Expected to throw " + ParserException.class); + } + + public void testGetFilterHandleNumericParseError() throws Exception { + NumericRangeFilterBuilder filterBuilder = new NumericRangeFilterBuilder(); + filterBuilder.setStrictMode(false); + + String xml = ""; + Document doc = getDocumentFromString(xml); + Filter filter = filterBuilder.getFilter(doc.getDocumentElement()); + Directory ramDir = newDirectory(); + IndexWriter writer = new IndexWriter(ramDir, newIndexWriterConfig(TEST_VERSION_CURRENT, null)); + writer.commit(); + try + { + IndexReader reader = IndexReader.open(ramDir, true); + try + { + assertNull(filter.getDocIdSet(reader)); + } + finally + { + reader.close(); + } + } + finally + { + writer.commit(); + writer.close(); + ramDir.close(); + } + } + + public void testGetFilterInt() throws Exception { + NumericRangeFilterBuilder filterBuilder = new NumericRangeFilterBuilder(); + filterBuilder.setStrictMode(true); + + String xml = ""; + Document doc = getDocumentFromString(xml); + Filter filter = filterBuilder.getFilter(doc.getDocumentElement()); + assertTrue(filter instanceof NumericRangeFilter); + @SuppressWarnings("unchecked") + NumericRangeFilter numRangeFilter = (NumericRangeFilter) filter; + assertEquals(Integer.valueOf(-1), numRangeFilter.getMin()); + assertEquals(Integer.valueOf(10), numRangeFilter.getMax()); + assertEquals("AGE", numRangeFilter.getField()); + assertTrue(numRangeFilter.includesMin()); + assertTrue(numRangeFilter.includesMax()); + + String xml2 = ""; + Document doc2 = getDocumentFromString(xml2); + Filter filter2 = filterBuilder.getFilter(doc2.getDocumentElement()); + assertTrue(filter2 instanceof NumericRangeFilter); + @SuppressWarnings("unchecked") + NumericRangeFilter numRangeFilter2 = (NumericRangeFilter) filter2; + assertEquals(Integer.valueOf(-1), numRangeFilter2.getMin()); + assertEquals(Integer.valueOf(10), numRangeFilter2.getMax()); + assertEquals("AGE", numRangeFilter2.getField()); + assertTrue(numRangeFilter2.includesMin()); + assertFalse(numRangeFilter2.includesMax()); + } + + public void testGetFilterLong() throws Exception { + NumericRangeFilterBuilder filterBuilder = new NumericRangeFilterBuilder(); + filterBuilder.setStrictMode(true); + + String xml = ""; + Document doc = getDocumentFromString(xml); + Filter filter = filterBuilder.getFilter(doc.getDocumentElement()); + assertTrue(filter instanceof NumericRangeFilter); + @SuppressWarnings("unchecked") + NumericRangeFilter numRangeFilter = (NumericRangeFilter) filter; + assertEquals(Long.valueOf(-2321L), numRangeFilter.getMin()); + assertEquals(Long.valueOf(60000000L), numRangeFilter.getMax()); + assertEquals("AGE", numRangeFilter.getField()); + assertTrue(numRangeFilter.includesMin()); + assertTrue(numRangeFilter.includesMax()); + + String xml2 = ""; + Document doc2 = getDocumentFromString(xml2); + Filter filter2 = filterBuilder.getFilter(doc2.getDocumentElement()); + assertTrue(filter2 instanceof NumericRangeFilter); + @SuppressWarnings("unchecked") + NumericRangeFilter numRangeFilter2 = (NumericRangeFilter) filter2; + assertEquals(Long.valueOf(-2321L), numRangeFilter2.getMin()); + assertEquals(Long.valueOf(60000000L), numRangeFilter2.getMax()); + assertEquals("AGE", numRangeFilter2.getField()); + assertTrue(numRangeFilter2.includesMin()); + assertFalse(numRangeFilter2.includesMax()); + } + + public void testGetFilterDouble() throws Exception { + NumericRangeFilterBuilder filterBuilder = new NumericRangeFilterBuilder(); + filterBuilder.setStrictMode(true); + + String xml = ""; + Document doc = getDocumentFromString(xml); + + Filter filter = filterBuilder.getFilter(doc.getDocumentElement()); + assertTrue(filter instanceof NumericRangeFilter); + @SuppressWarnings("unchecked") + NumericRangeFilter numRangeFilter = (NumericRangeFilter) filter; + assertEquals(Double.valueOf(-23.21d), numRangeFilter.getMin()); + assertEquals(Double.valueOf(60000.00023d), numRangeFilter.getMax()); + assertEquals("AGE", numRangeFilter.getField()); + assertTrue(numRangeFilter.includesMin()); + assertTrue(numRangeFilter.includesMax()); + + String xml2 = ""; + Document doc2 = getDocumentFromString(xml2); + Filter filter2 = filterBuilder.getFilter(doc2.getDocumentElement()); + assertTrue(filter2 instanceof NumericRangeFilter); + @SuppressWarnings("unchecked") + NumericRangeFilter numRangeFilter2 = (NumericRangeFilter) filter2; + assertEquals(Double.valueOf(-23.21d), numRangeFilter2.getMin()); + assertEquals(Double.valueOf(60000.00023d), numRangeFilter2.getMax()); + assertEquals("AGE", numRangeFilter2.getField()); + assertTrue(numRangeFilter2.includesMin()); + assertFalse(numRangeFilter2.includesMax()); + } + + public void testGetFilterFloat() throws Exception { + NumericRangeFilterBuilder filterBuilder = new NumericRangeFilterBuilder(); + filterBuilder.setStrictMode(true); + + String xml = ""; + Document doc = getDocumentFromString(xml); + + Filter filter = filterBuilder.getFilter(doc.getDocumentElement()); + assertTrue(filter instanceof NumericRangeFilter); + @SuppressWarnings("unchecked") + NumericRangeFilter numRangeFilter = (NumericRangeFilter) filter; + assertEquals(Float.valueOf(-2.321432f), numRangeFilter.getMin()); + assertEquals(Float.valueOf(32432.23f), numRangeFilter.getMax()); + assertEquals("AGE", numRangeFilter.getField()); + assertTrue(numRangeFilter.includesMin()); + assertTrue(numRangeFilter.includesMax()); + + String xml2 = ""; + Document doc2 = getDocumentFromString(xml2); + + Filter filter2 = filterBuilder.getFilter(doc2.getDocumentElement()); + assertTrue(filter2 instanceof NumericRangeFilter); + @SuppressWarnings("unchecked") + NumericRangeFilter numRangeFilter2 = (NumericRangeFilter) filter2; + assertEquals(Float.valueOf(-2.321432f), numRangeFilter2.getMin()); + assertEquals(Float.valueOf(32432.23f), numRangeFilter2.getMax()); + assertEquals("AGE", numRangeFilter2.getField()); + assertTrue(numRangeFilter2.includesMin()); + assertFalse(numRangeFilter2.includesMax()); + } + + private static Document getDocumentFromString(String str) + throws SAXException, IOException, ParserConfigurationException { + InputStream is = new ByteArrayInputStream(str.getBytes()); + DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance(); + factory.setNamespaceAware(true); + DocumentBuilder builder = factory.newDocumentBuilder(); + Document doc = builder.parse(is); + is.close(); + return doc; + } + +}