--- /dev/null
+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 = "<NumericRangeFilter fieldName='AGE' type='int' lowerTerm='-1' upperTerm='NaN'/>";
+ 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 = "<NumericRangeFilter fieldName='AGE' type='int' lowerTerm='-1' upperTerm='NaN'/>";
+ 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 = "<NumericRangeFilter fieldName='AGE' type='int' lowerTerm='-1' upperTerm='10'/>";
+ Document doc = getDocumentFromString(xml);
+ Filter filter = filterBuilder.getFilter(doc.getDocumentElement());
+ assertTrue(filter instanceof NumericRangeFilter<?>);
+ @SuppressWarnings("unchecked")
+ NumericRangeFilter<Integer> numRangeFilter = (NumericRangeFilter<Integer>) 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 = "<NumericRangeFilter fieldName='AGE' type='int' lowerTerm='-1' upperTerm='10' includeUpper='false'/>";
+ Document doc2 = getDocumentFromString(xml2);
+ Filter filter2 = filterBuilder.getFilter(doc2.getDocumentElement());
+ assertTrue(filter2 instanceof NumericRangeFilter<?>);
+ @SuppressWarnings("unchecked")
+ NumericRangeFilter<Integer> 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 = "<NumericRangeFilter fieldName='AGE' type='LoNg' lowerTerm='-2321' upperTerm='60000000'/>";
+ Document doc = getDocumentFromString(xml);
+ Filter filter = filterBuilder.getFilter(doc.getDocumentElement());
+ assertTrue(filter instanceof NumericRangeFilter<?>);
+ @SuppressWarnings("unchecked")
+ NumericRangeFilter<Long> 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 = "<NumericRangeFilter fieldName='AGE' type='LoNg' lowerTerm='-2321' upperTerm='60000000' includeUpper='false'/>";
+ Document doc2 = getDocumentFromString(xml2);
+ Filter filter2 = filterBuilder.getFilter(doc2.getDocumentElement());
+ assertTrue(filter2 instanceof NumericRangeFilter<?>);
+ @SuppressWarnings("unchecked")
+ NumericRangeFilter<Long> 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 = "<NumericRangeFilter fieldName='AGE' type='doubLe' lowerTerm='-23.21' upperTerm='60000.00023'/>";
+ Document doc = getDocumentFromString(xml);
+
+ Filter filter = filterBuilder.getFilter(doc.getDocumentElement());
+ assertTrue(filter instanceof NumericRangeFilter<?>);
+ @SuppressWarnings("unchecked")
+ NumericRangeFilter<Double> 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 = "<NumericRangeFilter fieldName='AGE' type='doubLe' lowerTerm='-23.21' upperTerm='60000.00023' includeUpper='false'/>";
+ Document doc2 = getDocumentFromString(xml2);
+ Filter filter2 = filterBuilder.getFilter(doc2.getDocumentElement());
+ assertTrue(filter2 instanceof NumericRangeFilter<?>);
+ @SuppressWarnings("unchecked")
+ NumericRangeFilter<Double> 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 = "<NumericRangeFilter fieldName='AGE' type='FLOAT' lowerTerm='-2.321432' upperTerm='32432.23'/>";
+ Document doc = getDocumentFromString(xml);
+
+ Filter filter = filterBuilder.getFilter(doc.getDocumentElement());
+ assertTrue(filter instanceof NumericRangeFilter<?>);
+ @SuppressWarnings("unchecked")
+ NumericRangeFilter<Float> 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 = "<NumericRangeFilter fieldName='AGE' type='FLOAT' lowerTerm='-2.321432' upperTerm='32432.23' includeUpper='false' precisionStep='2' />";
+ Document doc2 = getDocumentFromString(xml2);
+
+ Filter filter2 = filterBuilder.getFilter(doc2.getDocumentElement());
+ assertTrue(filter2 instanceof NumericRangeFilter<?>);
+ @SuppressWarnings("unchecked")
+ NumericRangeFilter<Float> 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;
+ }
+
+}
fnp / pylucene.git / blobdiff