You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@lucene.apache.org by da...@apache.org on 2018/11/02 11:16:29 UTC
[20/56] lucene-solr:jira/gradle: Add :solr:contrib:analytics module
http://git-wip-us.apache.org/repos/asf/lucene-solr/blob/f0366b94/solr/contrib/analytics/src/test/java/org/apache/solr/analytics/function/mapping/FillMissingFunctionTest.java
----------------------------------------------------------------------
diff --git a/solr/contrib/analytics/src/test/java/org/apache/solr/analytics/function/mapping/FillMissingFunctionTest.java b/solr/contrib/analytics/src/test/java/org/apache/solr/analytics/function/mapping/FillMissingFunctionTest.java
new file mode 100644
index 0000000..0ddce31
--- /dev/null
+++ b/solr/contrib/analytics/src/test/java/org/apache/solr/analytics/function/mapping/FillMissingFunctionTest.java
@@ -0,0 +1,717 @@
+/*
+ * 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.solr.analytics.function.mapping;
+
+import java.time.Instant;
+import java.time.format.DateTimeParseException;
+import java.util.Arrays;
+import java.util.Date;
+import java.util.Iterator;
+
+import org.apache.solr.SolrTestCaseJ4;
+import org.apache.solr.analytics.value.AnalyticsValue;
+import org.apache.solr.analytics.value.AnalyticsValueStream;
+import org.apache.solr.analytics.value.BooleanValue;
+import org.apache.solr.analytics.value.BooleanValueStream;
+import org.apache.solr.analytics.value.DateValue;
+import org.apache.solr.analytics.value.DateValueStream;
+import org.apache.solr.analytics.value.DoubleValue;
+import org.apache.solr.analytics.value.DoubleValueStream;
+import org.apache.solr.analytics.value.FloatValue;
+import org.apache.solr.analytics.value.FloatValueStream;
+import org.apache.solr.analytics.value.IntValue;
+import org.apache.solr.analytics.value.IntValueStream;
+import org.apache.solr.analytics.value.LongValue;
+import org.apache.solr.analytics.value.LongValueStream;
+import org.apache.solr.analytics.value.StringValue;
+import org.apache.solr.analytics.value.StringValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestAnalyticsValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestAnalyticsValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestBooleanValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestBooleanValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestDateValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestDateValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestDoubleValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestDoubleValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestFloatValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestFloatValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestIntValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestIntValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestLongValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestLongValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestStringValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestStringValueStream;
+import org.junit.Test;
+
+public class FillMissingFunctionTest extends SolrTestCaseJ4 {
+
+ @Test
+ public void castingTest() {
+ assertTrue(FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {new TestBooleanValue(), new TestStringValue()}) instanceof StringValue);
+
+ assertFalse(FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {new TestIntValue(), new TestFloatValue()}) instanceof IntValue);
+
+ assertFalse(FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {new TestBooleanValue(), new TestStringValueStream()}) instanceof AnalyticsValue);
+
+ assertFalse(FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {new TestLongValue(), new TestAnalyticsValue()}) instanceof LongValue);
+
+ assertTrue(FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {new TestIntValue(), new TestLongValue()}) instanceof DoubleValue);
+
+ assertFalse(FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {new TestDateValue(), new TestStringValue()}) instanceof DateValue);
+
+ assertFalse(FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {new TestBooleanValueStream(), new TestStringValue()}) instanceof AnalyticsValue);
+
+ assertTrue(FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {new TestDoubleValue(), new TestIntValueStream()}) instanceof DoubleValueStream);
+ }
+
+ @Test
+ public void singleValueBooleanTest() {
+ TestBooleanValue val = new TestBooleanValue();
+ TestBooleanValue filler = new TestBooleanValue();
+
+ AnalyticsValueStream uncasted = FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filler});
+ assertTrue(uncasted instanceof BooleanValue);
+ BooleanValue func = (BooleanValue) uncasted;
+
+ // Value doesn't exist
+ val.setExists(false);
+ filler.setExists(false);
+ func.getBoolean();
+ assertFalse(func.exists());
+
+ // Value exists
+ val.setValue(true).setExists(true);
+ filler.setValue(false).setExists(true);
+ assertEquals(true, func.getBoolean());
+ assertTrue(func.exists());
+
+ val.setValue(false).setExists(true);
+ filler.setValue(false).setExists(true);
+ assertEquals(false, func.getBoolean());
+ assertTrue(func.exists());
+
+ val.setValue(true).setExists(true);
+ filler.setExists(false);
+ assertEquals(true, func.getBoolean());
+ assertTrue(func.exists());
+
+ val.setValue(false).setExists(true);
+ filler.setExists(false);
+ assertEquals(false, func.getBoolean());
+ assertTrue(func.exists());
+
+ val.setExists(false);
+ filler.setValue(false).setExists(true);
+ assertEquals(false, func.getBoolean());
+ assertTrue(func.exists());
+
+ val.setExists(false);
+ filler.setValue(true).setExists(true);
+ assertEquals(true, func.getBoolean());
+ assertTrue(func.exists());
+ }
+
+ @Test
+ public void singleValueIntTest() {
+ TestIntValue val = new TestIntValue();
+ TestIntValue filler = new TestIntValue();
+
+ AnalyticsValueStream uncasted = FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filler});
+ assertTrue(uncasted instanceof IntValue);
+ IntValue func = (IntValue) uncasted;
+
+ // Value doesn't exist
+ val.setExists(false);
+ filler.setExists(false);
+ func.getInt();
+ assertFalse(func.exists());
+
+ // Value exists
+ val.setValue(21).setExists(true);
+ filler.setValue(234).setExists(true);
+ assertEquals(21, func.getInt());
+ assertTrue(func.exists());
+
+ val.setExists(false);
+ filler.setValue(-234).setExists(true);
+ assertEquals(-234, func.getInt());
+ assertTrue(func.exists());
+
+ val.setValue(52334).setExists(true);
+ filler.setExists(false);
+ assertEquals(52334, func.getInt());
+ assertTrue(func.exists());
+ }
+
+ @Test
+ public void singleValueLongTest() {
+ TestLongValue val = new TestLongValue();
+ TestLongValue filler = new TestLongValue();
+
+ AnalyticsValueStream uncasted = FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filler});
+ assertTrue(uncasted instanceof LongValue);
+ LongValue func = (LongValue) uncasted;
+
+ // Value doesn't exist
+ val.setExists(false);
+ filler.setExists(false);
+ func.getLong();
+ assertFalse(func.exists());
+
+ // Value exists
+ val.setValue(21L).setExists(true);
+ filler.setValue(234L).setExists(true);
+ assertEquals(21L, func.getLong());
+ assertTrue(func.exists());
+
+ val.setExists(false);
+ filler.setValue(234L).setExists(true);
+ assertEquals(234L, func.getLong());
+ assertTrue(func.exists());
+
+ val.setValue(-52334L).setExists(true);
+ filler.setExists(false);
+ assertEquals(-52334L, func.getLong());
+ assertTrue(func.exists());
+ }
+
+ @Test
+ public void singleValueFloatTest() {
+ TestFloatValue val = new TestFloatValue();
+ TestFloatValue filler = new TestFloatValue();
+
+ AnalyticsValueStream uncasted = FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filler});
+ assertTrue(uncasted instanceof FloatValue);
+ FloatValue func = (FloatValue) uncasted;
+
+ // Value doesn't exist
+ val.setExists(false);
+ filler.setExists(false);
+ func.getFloat();
+ assertFalse(func.exists());
+
+ // Value exists
+ val.setValue(-21.324F).setExists(true);
+ filler.setValue(23423.423342F).setExists(true);
+ assertEquals(-21.324F, func.getFloat(), .00000001);
+ assertTrue(func.exists());
+
+ val.setExists(false);
+ filler.setValue(3124123.32F).setExists(true);
+ assertEquals(3124123.32F, func.getFloat(), .00000001);
+ assertTrue(func.exists());
+
+ val.setValue(2345.345543F).setExists(true);
+ filler.setExists(false);
+ assertEquals(2345.345543F, func.getFloat(), .00000001);
+ assertTrue(func.exists());
+ }
+
+ @Test
+ public void singleValueDoubleTest() {
+ TestDoubleValue val = new TestDoubleValue();
+ TestDoubleValue filler = new TestDoubleValue();
+
+ AnalyticsValueStream uncasted = FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filler});
+ assertTrue(uncasted instanceof DoubleValue);
+ DoubleValue func = (DoubleValue) uncasted;
+
+ // Value doesn't exist
+ val.setExists(false);
+ filler.setExists(false);
+ func.getDouble();
+ assertFalse(func.exists());
+
+ // Value exists
+ val.setValue(-21.324).setExists(true);
+ filler.setValue(23423.423342).setExists(true);
+ assertEquals(-21.324, func.getDouble(), .00000001);
+ assertTrue(func.exists());
+
+ val.setExists(false);
+ filler.setValue(3124123.32).setExists(true);
+ assertEquals(3124123.32, func.getDouble(), .00000001);
+ assertTrue(func.exists());
+
+ val.setValue(2345.345543).setExists(true);
+ filler.setExists(false);
+ assertEquals(2345.345543, func.getDouble(), .00000001);
+ assertTrue(func.exists());
+ }
+
+ @Test
+ public void singleValueDateTest() throws DateTimeParseException {
+ Date date1 = Date.from(Instant.parse("1810-12-02T10:30:15Z"));
+ Date date2 = Date.from(Instant.parse("1950-02-23T14:54:34Z"));
+ Date date3 = Date.from(Instant.parse("2023-11-01T20:30:15Z"));
+
+ TestDateValue val = new TestDateValue();
+ TestDateValue filler = new TestDateValue();
+
+ AnalyticsValueStream uncasted = FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filler});
+ assertTrue(uncasted instanceof DateValue);
+ DateValue func = (DateValue) uncasted;
+
+ // Value doesn't exist
+ val.setExists(false);
+ filler.setExists(false);
+ func.getLong();
+ assertFalse(func.exists());
+
+ // Value exists
+ val.setValue("1810-12-02T10:30:15Z").setExists(true);
+ filler.setValue("2023-11-01T20:30:15Z").setExists(true);
+ assertEquals(date1.getTime(), func.getLong());
+ assertTrue(func.exists());
+
+ val.setExists(false);
+ filler.setValue("1950-02-23T14:54:34Z").setExists(true);
+ assertEquals(date2.getTime(), func.getLong());
+ assertTrue(func.exists());
+
+ val.setValue("2023-11-01T20:30:15Z").setExists(true);
+ filler.setExists(false);
+ assertEquals(date3.getTime(), func.getLong());
+ assertTrue(func.exists());
+ }
+
+ @Test
+ public void singleValueStringTest() {
+ TestStringValue val = new TestStringValue();
+ TestStringValue filler = new TestStringValue();
+
+ AnalyticsValueStream uncasted = FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filler});
+ assertTrue(uncasted instanceof StringValue);
+ StringValue func = (StringValue) uncasted;
+
+ // Value doesn't exist
+ val.setExists(false);
+ filler.setExists(false);
+ func.getString();
+ assertFalse(func.exists());
+
+ // Value exists
+ val.setValue("abc123").setExists(true);
+ filler.setValue("def456").setExists(true);
+ assertEquals("abc123", func.getString());
+ assertTrue(func.exists());
+
+ val.setExists(false);
+ filler.setValue("abc456").setExists(true);
+ assertEquals("abc456", func.getString());
+ assertTrue(func.exists());
+
+ val.setValue("def123").setExists(true);
+ filler.setExists(false);
+ assertEquals("def123", func.getString());
+ assertTrue(func.exists());
+ }
+
+ @Test
+ public void singleValueObjectTest() {
+ TestAnalyticsValue val = new TestAnalyticsValue();
+ TestAnalyticsValue filler = new TestAnalyticsValue();
+
+ AnalyticsValueStream uncasted = FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filler});
+ assertTrue(uncasted instanceof AnalyticsValue);
+ AnalyticsValue func = (AnalyticsValue) uncasted;
+
+ // Value doesn't exist
+ val.setExists(false);
+ filler.setExists(false);
+ func.getObject();
+ assertFalse(func.exists());
+
+ // Value exists
+ val.setValue("abc123").setExists(true);
+ filler.setValue(new Date(123)).setExists(true);
+ assertEquals("abc123", func.getObject());
+ assertTrue(func.exists());
+
+ val.setExists(false);
+ filler.setValue(Boolean.TRUE).setExists(true);
+ assertEquals(Boolean.TRUE, func.getObject());
+ assertTrue(func.exists());
+
+ val.setValue(234L).setExists(true);
+ filler.setExists(false);
+ assertEquals(234L, func.getObject());
+ assertTrue(func.exists());
+ }
+
+ @Test
+ public void multiValueBooleanTest() {
+ TestBooleanValueStream val = new TestBooleanValueStream();
+ TestBooleanValueStream filler = new TestBooleanValueStream();
+
+ AnalyticsValueStream uncasted = FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filler});
+ assertTrue(uncasted instanceof BooleanValueStream);
+ BooleanValueStream func = (BooleanValueStream) uncasted;
+
+ // No values
+ val.setValues();
+ filler.setValues();
+ func.streamBooleans( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // Values exist
+ val.setValues(true);
+ filler.setValues(true, false);
+ Iterator<Boolean> values1 = Arrays.asList(true).iterator();
+ func.streamBooleans( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next(), value);
+ });
+ assertFalse(values1.hasNext());
+
+ val.setValues();
+ filler.setValues(true, false);
+ Iterator<Boolean> values2 = Arrays.asList(true, false).iterator();
+ func.streamBooleans( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next(), value);
+ });
+ assertFalse(values2.hasNext());
+
+ val.setValues(true, false, true);
+ filler.setValues();
+ Iterator<Boolean> values3 = Arrays.asList(true, false, true).iterator();
+ func.streamBooleans( value -> {
+ assertTrue(values3.hasNext());
+ assertEquals(values3.next(), value);
+ });
+ assertFalse(values3.hasNext());
+ }
+
+ @Test
+ public void multiValueIntTest() {
+ TestIntValueStream val = new TestIntValueStream();
+ TestIntValueStream filler = new TestIntValueStream();
+
+ AnalyticsValueStream uncasted = FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filler});
+ assertTrue(uncasted instanceof IntValueStream);
+ IntValueStream func = (IntValueStream) uncasted;
+
+ // No values
+ val.setValues();
+ filler.setValues();
+ func.streamInts( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // Values exist
+ val.setValues(323, -9423);
+ filler.setValues(-1234, 5433, -73843, 1245144);
+ Iterator<Integer> values1 = Arrays.asList(323, -9423).iterator();
+ func.streamInts( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next().intValue(), value);
+ });
+ assertFalse(values1.hasNext());
+
+ val.setValues();
+ filler.setValues(312423, -12343, -23, 2);
+ Iterator<Integer> values2 = Arrays.asList(312423, -12343, -23, 2).iterator();
+ func.streamInts( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next().intValue(), value);
+ });
+ assertFalse(values2.hasNext());
+
+ val.setValues(1423, -413543);
+ filler.setValues();
+ Iterator<Integer> values3 = Arrays.asList(1423, -413543).iterator();
+ func.streamInts( value -> {
+ assertTrue(values3.hasNext());
+ assertEquals(values3.next().intValue(), value);
+ });
+ assertFalse(values3.hasNext());
+ }
+
+ @Test
+ public void multiValueLongTest() {
+ TestLongValueStream val = new TestLongValueStream();
+ TestLongValueStream filler = new TestLongValueStream();
+
+ AnalyticsValueStream uncasted = FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filler});
+ assertTrue(uncasted instanceof LongValueStream);
+ LongValueStream func = (LongValueStream) uncasted;
+
+ // No values
+ val.setValues();
+ filler.setValues();
+ func.streamLongs( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // Values exist
+ val.setValues(323L, -9423L);
+ filler.setValues(-1234L, -5433L, -73843L, 1245144L);
+ Iterator<Long> values1 = Arrays.asList(323L, -9423L).iterator();
+ func.streamLongs( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next().longValue(), value);
+ });
+ assertFalse(values1.hasNext());
+
+ val.setValues();
+ filler.setValues(-312423L, 12343L, 23L, 2L);
+ Iterator<Long> values2 = Arrays.asList(-312423L, 12343L, 23L, 2L).iterator();
+ func.streamLongs( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next().longValue(), value);
+ });
+ assertFalse(values2.hasNext());
+
+ val.setValues(1423L, -413543L);
+ filler.setValues();
+ Iterator<Long> values3 = Arrays.asList(1423L, -413543L).iterator();
+ func.streamLongs( value -> {
+ assertTrue(values3.hasNext());
+ assertEquals(values3.next().longValue(), value);
+ });
+ assertFalse(values3.hasNext());
+ }
+
+ @Test
+ public void multiValueFloatTest() {
+ TestFloatValueStream val = new TestFloatValueStream();
+ TestFloatValueStream filler = new TestFloatValueStream();
+
+ AnalyticsValueStream uncasted = FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filler});
+ assertTrue(uncasted instanceof FloatValueStream);
+ FloatValueStream func = (FloatValueStream) uncasted;
+
+ // No values
+ val.setValues();
+ filler.setValues();
+ func.streamFloats( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // Values exist
+ val.setValues(323.213F, -9423.5F);
+ filler.setValues(-1234.9478F, -5433.234F, -73843F, 1245144.2342F);
+ Iterator<Float> values1 = Arrays.asList(323.213F, -9423.5F).iterator();
+ func.streamFloats( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next(), value, .00000001);
+ });
+ assertFalse(values1.hasNext());
+
+ val.setValues();
+ filler.setValues(-312423.2F, 12343.234823F, 23.582F, 2.23F);
+ Iterator<Float> values2 = Arrays.asList(-312423.2F, 12343.234823F, 23.582F, 2.23F).iterator();
+ func.streamFloats( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next(), value, .00000001);
+ });
+ assertFalse(values2.hasNext());
+
+ val.setValues(1423.23039F, -413543F);
+ filler.setValues();
+ Iterator<Float> values3 = Arrays.asList(1423.23039F, -413543F).iterator();
+ func.streamFloats( value -> {
+ assertTrue(values3.hasNext());
+ assertEquals(values3.next(), value, .00000001);
+ });
+ assertFalse(values3.hasNext());
+ }
+
+ @Test
+ public void multiValueDoubleTest() {
+ TestDoubleValueStream val = new TestDoubleValueStream();
+ TestDoubleValueStream filler = new TestDoubleValueStream();
+
+ AnalyticsValueStream uncasted = FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filler});
+ assertTrue(uncasted instanceof DoubleValueStream);
+ DoubleValueStream func = (DoubleValueStream) uncasted;
+
+ // No values
+ val.setValues();
+ filler.setValues();
+ func.streamDoubles( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // Values exist
+ val.setValues(323.213, -9423.5);
+ filler.setValues(-1234.9478, -5433.234, -73843.0, 1245144.2342);
+ Iterator<Double> values1 = Arrays.asList(323.213, -9423.5).iterator();
+ func.streamDoubles( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next(), value, .00000001);
+ });
+ assertFalse(values1.hasNext());
+
+ val.setValues();
+ filler.setValues(-312423.2, 12343.234823, 23.582, 2.23);
+ Iterator<Double> values2 = Arrays.asList(-312423.2, 12343.234823, 23.582, 2.23).iterator();
+ func.streamDoubles( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next(), value, .00000001);
+ });
+ assertFalse(values2.hasNext());
+
+ val.setValues(1423.23039, -413543.0);
+ filler.setValues();
+ Iterator<Double> values3 = Arrays.asList(1423.23039, -413543.0).iterator();
+ func.streamDoubles( value -> {
+ assertTrue(values3.hasNext());
+ assertEquals(values3.next(), value, .00000001);
+ });
+ assertFalse(values3.hasNext());
+ }
+
+ @Test
+ public void multiValueDateTest() throws DateTimeParseException {
+ Date date1 = Date.from(Instant.parse("1810-12-02T10:30:15Z"));
+ Date date2 = Date.from(Instant.parse("1931-03-16T18:15:45Z"));
+ Date date3 = Date.from(Instant.parse("2023-11-01T20:30:15Z"));
+
+ TestDateValueStream val = new TestDateValueStream();
+ TestDateValueStream filler = new TestDateValueStream();
+
+ AnalyticsValueStream uncasted = FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filler});
+ assertTrue(uncasted instanceof DateValueStream);
+ DateValueStream func = (DateValueStream) uncasted;
+
+ // No values
+ val.setValues();
+ filler.setValues();
+ func.streamLongs( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // Values exist
+ val.setValues("1810-12-02T10:30:15Z");
+ filler.setValues("1931-03-16T18:15:45Z", "2023-11-01T20:30:15Z");
+ Iterator<Long> values1 = Arrays.asList(date1.getTime()).iterator();
+ func.streamLongs( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next().longValue(), value);
+ });
+ assertFalse(values1.hasNext());
+
+ val.setValues();
+ filler.setValues("1931-03-16T18:15:45Z", "1810-12-02T10:30:15Z");
+ Iterator<Long> values2 = Arrays.asList(date2.getTime(), date1.getTime()).iterator();
+ func.streamLongs( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next().longValue(), value);
+ });
+ assertFalse(values2.hasNext());
+
+ val.setValues("2023-11-01T20:30:15Z", "1931-03-16T18:15:45Z");
+ filler.setValues();
+ Iterator<Long> values3 = Arrays.asList(date3.getTime(), date2.getTime()).iterator();
+ func.streamLongs( value -> {
+ assertTrue(values3.hasNext());
+ assertEquals(values3.next().longValue(), value);
+ });
+ assertFalse(values3.hasNext());
+ }
+
+ @Test
+ public void multiValueStringTest() {
+ TestStringValueStream val = new TestStringValueStream();
+ TestStringValueStream filler = new TestStringValueStream();
+
+ AnalyticsValueStream uncasted = FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filler});
+ assertTrue(uncasted instanceof StringValueStream);
+ StringValueStream func = (StringValueStream) uncasted;
+
+ // No values
+ val.setValues();
+ filler.setValues();
+ func.streamStrings( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // Values exist
+ val.setValues("abc");
+ filler.setValues("123", "def");
+ Iterator<String> values1 = Arrays.asList("abc").iterator();
+ func.streamStrings( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next(), value);
+ });
+ assertFalse(values1.hasNext());
+
+ val.setValues();
+ filler.setValues("def", "12341");
+ Iterator<String> values2 = Arrays.asList("def", "12341").iterator();
+ func.streamStrings( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next(), value);
+ });
+ assertFalse(values2.hasNext());
+
+ val.setValues("string1", "another string", "the final and last string");
+ filler.setValues();
+ Iterator<String> values3 = Arrays.asList("string1", "another string", "the final and last string").iterator();
+ func.streamStrings( value -> {
+ assertTrue(values3.hasNext());
+ assertEquals(values3.next(), value);
+ });
+ assertFalse(values3.hasNext());
+ }
+
+ @Test
+ public void multiValueObjectTest() {
+ TestAnalyticsValueStream val = new TestAnalyticsValueStream();
+ TestAnalyticsValueStream filler = new TestAnalyticsValueStream();
+
+ AnalyticsValueStream func = FillMissingFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filler});
+
+ // No values
+ val.setValues();
+ filler.setValues();
+ func.streamObjects( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // Values exist
+ val.setValues("asdfs");
+ filler.setValues(new Date(12312), 213123L);
+ Iterator<Object> values1 = Arrays.<Object>asList("asdfs").iterator();
+ func.streamObjects( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next(), value);
+ });
+ assertFalse(values1.hasNext());
+
+ val.setValues();
+ filler.setValues(3234.42d, "replacement");
+ Iterator<Object> values2 = Arrays.<Object>asList(3234.42d, "replacement").iterator();
+ func.streamObjects( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next(), value);
+ });
+ assertFalse(values2.hasNext());
+
+ val.setValues(new Date(3), "3", 3F);
+ filler.setValues();
+ Iterator<Object> values3 = Arrays.<Object>asList(new Date(3), "3", 3F).iterator();
+ func.streamObjects( value -> {
+ assertTrue(values3.hasNext());
+ assertEquals(values3.next(), value);
+ });
+ assertFalse(values3.hasNext());
+ }
+}
http://git-wip-us.apache.org/repos/asf/lucene-solr/blob/f0366b94/solr/contrib/analytics/src/test/java/org/apache/solr/analytics/function/mapping/FilterFunctionTest.java
----------------------------------------------------------------------
diff --git a/solr/contrib/analytics/src/test/java/org/apache/solr/analytics/function/mapping/FilterFunctionTest.java b/solr/contrib/analytics/src/test/java/org/apache/solr/analytics/function/mapping/FilterFunctionTest.java
new file mode 100644
index 0000000..87dd3e7
--- /dev/null
+++ b/solr/contrib/analytics/src/test/java/org/apache/solr/analytics/function/mapping/FilterFunctionTest.java
@@ -0,0 +1,1035 @@
+/*
+ * 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.solr.analytics.function.mapping;
+
+import java.time.Instant;
+import java.time.format.DateTimeParseException;
+import java.util.Arrays;
+import java.util.Date;
+import java.util.Iterator;
+
+import org.apache.solr.SolrTestCaseJ4;
+import org.apache.solr.analytics.value.AnalyticsValue;
+import org.apache.solr.analytics.value.AnalyticsValueStream;
+import org.apache.solr.analytics.value.BooleanValue;
+import org.apache.solr.analytics.value.BooleanValueStream;
+import org.apache.solr.analytics.value.DateValue;
+import org.apache.solr.analytics.value.DateValueStream;
+import org.apache.solr.analytics.value.DoubleValue;
+import org.apache.solr.analytics.value.DoubleValueStream;
+import org.apache.solr.analytics.value.FloatValue;
+import org.apache.solr.analytics.value.FloatValueStream;
+import org.apache.solr.analytics.value.IntValue;
+import org.apache.solr.analytics.value.IntValueStream;
+import org.apache.solr.analytics.value.LongValue;
+import org.apache.solr.analytics.value.LongValueStream;
+import org.apache.solr.analytics.value.StringValue;
+import org.apache.solr.analytics.value.StringValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestAnalyticsValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestAnalyticsValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestBooleanValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestBooleanValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestDateValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestDateValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestDoubleValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestDoubleValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestFloatValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestFloatValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestIntValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestIntValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestLongValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestLongValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestStringValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestStringValueStream;
+import org.junit.Test;
+
+public class FilterFunctionTest extends SolrTestCaseJ4 {
+
+ @Test
+ public void singleValueBooleanTest() {
+ TestBooleanValue val = new TestBooleanValue();
+ TestBooleanValue filter = new TestBooleanValue();
+
+ AnalyticsValueStream uncasted = FilterFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filter});
+ assertTrue(uncasted instanceof BooleanValue);
+ BooleanValue func = (BooleanValue) uncasted;
+
+ // Value doesn't exist
+ val.setExists(false);
+ filter.setExists(false);
+ func.getBoolean();
+ assertFalse(func.exists());
+
+ val.setValue(true).setExists(true);
+ filter.setExists(false);
+ func.getBoolean();
+ assertFalse(func.exists());
+
+ val.setExists(false);
+ filter.setValue(true).setExists(true);
+ func.getBoolean();
+ assertFalse(func.exists());
+
+ // Value exists
+ val.setValue(true).setExists(true);
+ filter.setValue(true).setExists(true);
+ assertEquals(true, func.getBoolean());
+ assertTrue(func.exists());
+
+ val.setValue(true).setExists(true);
+ filter.setValue(false).setExists(true);
+ func.getBoolean();
+ assertFalse(func.exists());
+
+ val.setValue(false).setExists(true);
+ filter.setValue(true).setExists(true);
+ assertEquals(false, func.getBoolean());
+ assertTrue(func.exists());
+
+ val.setValue(false).setExists(true);
+ filter.setValue(false).setExists(true);
+ func.getBoolean();
+ assertFalse(func.exists());
+ }
+
+ @Test
+ public void singleValueIntTest() {
+ TestIntValue val = new TestIntValue();
+ TestBooleanValue filter = new TestBooleanValue();
+
+ AnalyticsValueStream uncasted = FilterFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filter});
+ assertTrue(uncasted instanceof IntValue);
+ IntValue func = (IntValue) uncasted;
+
+ // Value doesn't exist
+ val.setExists(false);
+ filter.setExists(false);
+ func.getInt();
+ assertFalse(func.exists());
+
+ val.setValue(32).setExists(true);
+ filter.setExists(false);
+ func.getInt();
+ assertFalse(func.exists());
+
+ val.setExists(false);
+ filter.setValue(true).setExists(true);
+ func.getInt();
+ assertFalse(func.exists());
+
+ // Value exists
+ val.setValue(21).setExists(true);
+ filter.setValue(true).setExists(true);
+ assertEquals(21, func.getInt());
+ assertTrue(func.exists());
+
+ val.setValue(-100).setExists(true);
+ filter.setValue(false).setExists(true);
+ func.getInt();
+ assertFalse(func.exists());
+
+ val.setValue(50).setExists(true);
+ filter.setValue(true).setExists(true);
+ assertEquals(50, func.getInt());
+ assertTrue(func.exists());
+
+ val.setValue(5432).setExists(true);
+ filter.setValue(false).setExists(true);
+ func.getInt();
+ assertFalse(func.exists());
+ }
+
+ @Test
+ public void singleValueLongTest() {
+ TestLongValue val = new TestLongValue();
+ TestBooleanValue filter = new TestBooleanValue();
+
+ AnalyticsValueStream uncasted = FilterFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filter});
+ assertTrue(uncasted instanceof LongValue);
+ LongValue func = (LongValue) uncasted;
+
+ // Value doesn't exist
+ val.setExists(false);
+ filter.setExists(false);
+ func.getLong();
+ assertFalse(func.exists());
+
+ val.setValue(32L).setExists(true);
+ filter.setExists(false);
+ func.getLong();
+ assertFalse(func.exists());
+
+ val.setExists(false);
+ filter.setValue(true).setExists(true);
+ func.getLong();
+ assertFalse(func.exists());
+
+ // Value exists
+ val.setValue(21L).setExists(true);
+ filter.setValue(true).setExists(true);
+ assertEquals(21L, func.getLong());
+ assertTrue(func.exists());
+
+ val.setValue(-100L).setExists(true);
+ filter.setValue(false).setExists(true);
+ func.getLong();
+ assertFalse(func.exists());
+
+ val.setValue(50L).setExists(true);
+ filter.setValue(true).setExists(true);
+ assertEquals(50L, func.getLong());
+ assertTrue(func.exists());
+
+ val.setValue(3245L).setExists(true);
+ filter.setValue(false).setExists(true);
+ func.getLong();
+ assertFalse(func.exists());
+ }
+
+ @Test
+ public void singleValueFloatTest() {
+ TestFloatValue val = new TestFloatValue();
+ TestBooleanValue filter = new TestBooleanValue();
+
+ AnalyticsValueStream uncasted = FilterFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filter});
+ assertTrue(uncasted instanceof FloatValue);
+ FloatValue func = (FloatValue) uncasted;
+
+ // Value doesn't exist
+ val.setExists(false);
+ filter.setExists(false);
+ func.getFloat();
+ assertFalse(func.exists());
+
+ val.setValue(32.123F).setExists(true);
+ filter.setExists(false);
+ func.getFloat();
+ assertFalse(func.exists());
+
+ val.setExists(false);
+ filter.setValue(true).setExists(true);
+ func.getFloat();
+ assertFalse(func.exists());
+
+ // Value exists
+ val.setValue(21.453F).setExists(true);
+ filter.setValue(true).setExists(true);
+ assertEquals(21.453F, func.getFloat(), .000001);
+ assertTrue(func.exists());
+
+ val.setValue(-100.123F).setExists(true);
+ filter.setValue(false).setExists(true);
+ func.getFloat();
+ assertFalse(func.exists());
+
+ val.setValue(50.5F).setExists(true);
+ filter.setValue(true).setExists(true);
+ assertEquals(50.5F, func.getFloat(), .000001);
+ assertTrue(func.exists());
+
+ val.setValue(25643.23F).setExists(true);
+ filter.setValue(false).setExists(true);
+ func.getFloat();
+ assertFalse(func.exists());
+ }
+
+ @Test
+ public void singleValueDoubleTest() {
+ TestDoubleValue val = new TestDoubleValue();
+ TestBooleanValue filter = new TestBooleanValue();
+
+ AnalyticsValueStream uncasted = FilterFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filter});
+ assertTrue(uncasted instanceof DoubleValue);
+ DoubleValue func = (DoubleValue) uncasted;
+
+ // Value doesn't exist
+ val.setExists(false);
+ filter.setExists(false);
+ func.getDouble();
+ assertFalse(func.exists());
+
+ val.setValue(32.123).setExists(true);
+ filter.setExists(false);
+ func.getDouble();
+ assertFalse(func.exists());
+
+ val.setExists(false);
+ filter.setValue(true).setExists(true);
+ func.getDouble();
+ assertFalse(func.exists());
+
+ // Value exists
+ val.setValue(21.453).setExists(true);
+ filter.setValue(true).setExists(true);
+ assertEquals(21.453, func.getDouble(), .000001);
+ assertTrue(func.exists());
+
+ val.setValue(-100.123).setExists(true);
+ filter.setValue(false).setExists(true);
+ func.getDouble();
+ assertFalse(func.exists());
+
+ val.setValue(50.5).setExists(true);
+ filter.setValue(true).setExists(true);
+ assertEquals(50.5, func.getDouble(), .000001);
+ assertTrue(func.exists());
+
+ val.setValue(25643.23).setExists(true);
+ filter.setValue(false).setExists(true);
+ func.getDouble();
+ assertFalse(func.exists());
+ }
+
+ @Test
+ public void singleValueDateTest() throws DateTimeParseException {
+ Date date1 = Date.from(Instant.parse("1810-12-02T10:30:15Z"));
+ Date date3 = Date.from(Instant.parse("2023-11-01T20:30:15Z"));
+
+ TestDateValue val = new TestDateValue();
+ TestBooleanValue filter = new TestBooleanValue();
+
+ AnalyticsValueStream uncasted = FilterFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filter});
+ assertTrue(uncasted instanceof DateValue);
+ DateValue func = (DateValue) uncasted;
+
+ // Value doesn't exist
+ val.setExists(false);
+ filter.setExists(false);
+ func.getLong();
+ assertFalse(func.exists());
+
+ val.setValue("1810-12-02T10:30:15Z").setExists(true);
+ filter.setExists(false);
+ func.getLong();
+ assertFalse(func.exists());
+
+ val.setExists(false);
+ filter.setValue(true).setExists(true);
+ func.getLong();
+ assertFalse(func.exists());
+
+ // Value exists
+ val.setValue("1810-12-02T10:30:15Z").setExists(true);
+ filter.setValue(true).setExists(true);
+ assertEquals(date1.getTime(), func.getLong());
+ assertTrue(func.exists());
+
+ val.setValue("1931-03-16T18:15:45Z").setExists(true);
+ filter.setValue(false).setExists(true);
+ func.getLong();
+ assertFalse(func.exists());
+
+ val.setValue("2023-11-01T20:30:15Z").setExists(true);
+ filter.setValue(true).setExists(true);
+ assertEquals(date3.getTime(), func.getLong());
+ assertTrue(func.exists());
+
+ val.setValue("1810-12-02T10:30:15Z").setExists(true);
+ filter.setValue(false).setExists(true);
+ func.getLong();
+ assertFalse(func.exists());
+ }
+
+ @Test
+ public void singleValueStringTest() {
+ TestStringValue val = new TestStringValue();
+ TestBooleanValue filter = new TestBooleanValue();
+
+ AnalyticsValueStream uncasted = FilterFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filter});
+ assertTrue(uncasted instanceof StringValue);
+ StringValue func = (StringValue) uncasted;
+
+ // Value doesn't exist
+ val.setExists(false);
+ filter.setExists(false);
+ func.getString();
+ assertFalse(func.exists());
+
+ val.setValue("abc123").setExists(true);
+ filter.setExists(false);
+ func.getString();
+ assertFalse(func.exists());
+
+ val.setExists(false);
+ filter.setValue(true).setExists(true);
+ func.getString();
+ assertFalse(func.exists());
+
+ // Value exists
+ val.setValue("abc").setExists(true);
+ filter.setValue(true).setExists(true);
+ assertEquals("abc", func.getString());
+ assertTrue(func.exists());
+
+ val.setValue("def").setExists(true);
+ filter.setValue(false).setExists(true);
+ func.getString();
+ assertFalse(func.exists());
+
+ val.setValue("123").setExists(true);
+ filter.setValue(true).setExists(true);
+ assertEquals("123", func.getString());
+ assertTrue(func.exists());
+
+ val.setValue("456").setExists(true);
+ filter.setValue(false).setExists(true);
+ func.getString();
+ assertFalse(func.exists());
+ }
+
+ @Test
+ public void singleValueObjectTest() {
+ TestAnalyticsValue val = new TestAnalyticsValue();
+ TestBooleanValue filter = new TestBooleanValue();
+
+ AnalyticsValueStream uncasted = FilterFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filter});
+ assertTrue(uncasted instanceof AnalyticsValue);
+ AnalyticsValue func = (AnalyticsValue) uncasted;
+
+ // Value doesn't exist
+ val.setExists(false);
+ filter.setExists(false);
+ func.getObject();
+ assertFalse(func.exists());
+
+ val.setValue("1810-12-02T10:30:15Z").setExists(true);
+ filter.setExists(false);
+ func.getObject();
+ assertFalse(func.exists());
+
+ val.setExists(false);
+ filter.setValue(true).setExists(true);
+ func.getObject();
+ assertFalse(func.exists());
+
+ // Value exists
+ val.setValue(3L).setExists(true);
+ filter.setValue(true).setExists(true);
+ assertEquals(3L, func.getObject());
+ assertTrue(func.exists());
+
+ val.setValue(new Date(2)).setExists(true);
+ filter.setValue(false).setExists(true);
+ func.getObject();
+ assertFalse(func.exists());
+
+ val.setValue("3").setExists(true);
+ filter.setValue(true).setExists(true);
+ assertEquals("3", func.getObject());
+ assertTrue(func.exists());
+
+ val.setValue(new TestAnalyticsValue()).setExists(true);
+ filter.setValue(false).setExists(true);
+ func.getObject();
+ assertFalse(func.exists());
+ }
+
+ @Test
+ public void multiValueBooleanTest() {
+ TestBooleanValueStream val = new TestBooleanValueStream();
+ TestBooleanValue filter = new TestBooleanValue();
+
+ AnalyticsValueStream uncasted = FilterFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filter});
+ assertTrue(uncasted instanceof BooleanValueStream);
+ BooleanValueStream func = (BooleanValueStream) uncasted;
+
+ // No values
+ val.setValues();
+ filter.setExists(false);
+ func.streamBooleans( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ val.setValues(false, true, false);
+ filter.setExists(false);
+ func.streamBooleans( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ val.setValues();
+ filter.setValue(true).setExists(true);
+ func.streamBooleans( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // One value
+ val.setValues(true);
+
+ filter.setValue(true).setExists(true);
+ Iterator<Boolean> values1 = Arrays.asList(true).iterator();
+ func.streamBooleans( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next(), value);
+ });
+ assertFalse(values1.hasNext());
+
+ filter.setValue(false).setExists(true);
+ func.streamBooleans( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ filter.setValue(true).setExists(true);
+ Iterator<Boolean> values2 = Arrays.asList(true).iterator();
+ func.streamBooleans( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next(), value);
+ });
+ assertFalse(values2.hasNext());
+
+ // Multiple values
+ val.setValues(false, false, true, true);
+
+ filter.setValue(false).setExists(true);
+ func.streamBooleans( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ filter.setValue(true).setExists(true);
+ Iterator<Boolean> values3 = Arrays.asList(false, false, true, true).iterator();
+ func.streamBooleans( value -> {
+ assertTrue(values3.hasNext());
+ assertEquals(values3.next(), value);
+ });
+ assertFalse(values3.hasNext());
+
+ filter.setValue(false).setExists(true);
+ func.streamBooleans( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+ }
+
+ @Test
+ public void multiValueIntTest() {
+ TestIntValueStream val = new TestIntValueStream();
+ TestBooleanValue filter = new TestBooleanValue();
+
+ AnalyticsValueStream uncasted = FilterFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filter});
+ assertTrue(uncasted instanceof IntValueStream);
+ IntValueStream func = (IntValueStream) uncasted;
+
+ // No values
+ val.setValues();
+ filter.setExists(false);
+ func.streamInts( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ val.setValues(1, 3, 5);
+ filter.setExists(false);
+ func.streamInts( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ val.setValues();
+ filter.setValue(true).setExists(true);
+ func.streamInts( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // One value
+ val.setValues(-4);
+
+ filter.setValue(true).setExists(true);
+ Iterator<Integer> values1 = Arrays.asList(-4).iterator();
+ func.streamInts( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next().intValue(), value);
+ });
+ assertFalse(values1.hasNext());
+
+ filter.setValue(false).setExists(true);
+ func.streamInts( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ filter.setValue(true).setExists(true);
+ Iterator<Integer> values2 = Arrays.asList(-4).iterator();
+ func.streamInts( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next().intValue(), value);
+ });
+ assertFalse(values2.hasNext());
+
+ // Multiple values
+ val.setValues(4, -10, 50, -74);
+
+ filter.setValue(false).setExists(true);
+ func.streamInts( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ filter.setValue(true).setExists(true);
+ Iterator<Integer> values3 = Arrays.asList(4, -10, 50, -74).iterator();
+ func.streamInts( value -> {
+ assertTrue(values3.hasNext());
+ assertEquals(values3.next().intValue(), value);
+ });
+ assertFalse(values3.hasNext());
+
+ filter.setValue(false).setExists(true);
+ func.streamInts( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+ }
+
+ @Test
+ public void multiValueLongTest() {
+ TestLongValueStream val = new TestLongValueStream();
+ TestBooleanValue filter = new TestBooleanValue();
+
+ AnalyticsValueStream uncasted = FilterFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filter});
+ assertTrue(uncasted instanceof LongValueStream);
+ LongValueStream func = (LongValueStream) uncasted;
+
+ // No values
+ val.setValues();
+ filter.setExists(false);
+ func.streamLongs( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ val.setValues(1L, 3L, 5L);
+ filter.setExists(false);
+ func.streamLongs( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ val.setValues();
+ filter.setValue(true).setExists(true);
+ func.streamLongs( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // One value
+ val.setValues(-4L);
+
+ filter.setValue(true).setExists(true);
+ Iterator<Long> values1 = Arrays.asList(-4L).iterator();
+ func.streamLongs( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next().longValue(), value);
+ });
+ assertFalse(values1.hasNext());
+
+ filter.setValue(false).setExists(true);
+ func.streamLongs( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ filter.setValue(true).setExists(true);
+ Iterator<Long> values2 = Arrays.asList(-4L).iterator();
+ func.streamLongs( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next().longValue(), value);
+ });
+ assertFalse(values2.hasNext());
+
+ // Multiple values
+ val.setValues(4L, -10L, 50L, -74L);
+
+ filter.setValue(false).setExists(true);
+ func.streamLongs( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ filter.setValue(true).setExists(true);
+ Iterator<Long> values3 = Arrays.asList(4L, -10L, 50L, -74L).iterator();
+ func.streamLongs( value -> {
+ assertTrue(values3.hasNext());
+ assertEquals(values3.next().longValue(), value);
+ });
+ assertFalse(values3.hasNext());
+
+ filter.setValue(false).setExists(true);
+ func.streamLongs( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+ }
+
+ @Test
+ public void multiValueFloatTest() {
+ TestFloatValueStream val = new TestFloatValueStream();
+ TestBooleanValue filter = new TestBooleanValue();
+
+ AnalyticsValueStream uncasted = FilterFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filter});
+ assertTrue(uncasted instanceof FloatValueStream);
+ FloatValueStream func = (FloatValueStream) uncasted;
+
+ // No values
+ val.setValues();
+ filter.setExists(false);
+ func.streamFloats( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ val.setValues(50.343F, -74.9874F, 2342332342.32F);
+ filter.setExists(false);
+ func.streamFloats( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ val.setValues();
+ filter.setValue(true).setExists(true);
+ func.streamFloats( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // One value
+ val.setValues(-4.23423F);
+
+ filter.setValue(true).setExists(true);
+ Iterator<Float> values1 = Arrays.asList(-4.23423F).iterator();
+ func.streamFloats( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next(), value, .0000001);
+ });
+ assertFalse(values1.hasNext());
+
+ filter.setValue(false).setExists(true);
+ func.streamFloats( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ filter.setValue(true).setExists(true);
+ Iterator<Float> values2 = Arrays.asList(-4.23423F).iterator();
+ func.streamFloats( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next(), value, .0000001);
+ });
+ assertFalse(values2.hasNext());
+
+ // Multiple values
+ val.setValues(4.3F, -10.134F, 50.343F, -74.9874F);
+
+ filter.setValue(false).setExists(true);
+ func.streamFloats( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ filter.setValue(true).setExists(true);
+ Iterator<Float> values3 = Arrays.asList(4.3F, -10.134F, 50.343F, -74.9874F).iterator();
+ func.streamFloats( value -> {
+ assertTrue(values3.hasNext());
+ assertEquals(values3.next(), value, .0000001);
+ });
+ assertFalse(values3.hasNext());
+
+ filter.setValue(false).setExists(true);
+ func.streamFloats( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+ }
+
+ @Test
+ public void multiValueDoubleTest() {
+ TestDoubleValueStream val = new TestDoubleValueStream();
+ TestBooleanValue filter = new TestBooleanValue();
+
+ AnalyticsValueStream uncasted = FilterFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filter});
+ assertTrue(uncasted instanceof DoubleValueStream);
+ DoubleValueStream func = (DoubleValueStream) uncasted;
+
+ // No values
+ val.setValues();
+ filter.setExists(false);
+ func.streamDoubles( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ val.setValues(50.343, -74.9874, 2342332342.32);
+ filter.setExists(false);
+ func.streamDoubles( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ val.setValues();
+ filter.setValue(true).setExists(true);
+ func.streamDoubles( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // One value
+ val.setValues(-4.23423);
+
+ filter.setValue(true).setExists(true);
+ Iterator<Double> values1 = Arrays.asList(-4.23423).iterator();
+ func.streamDoubles( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next(), value, .0000001);
+ });
+ assertFalse(values1.hasNext());
+
+ filter.setValue(false).setExists(true);
+ func.streamDoubles( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ filter.setValue(true).setExists(true);
+ Iterator<Double> values2 = Arrays.asList(-4.23423).iterator();
+ func.streamDoubles( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next(), value, .0000001);
+ });
+ assertFalse(values2.hasNext());
+
+ // Multiple values
+ val.setValues(4.3, -10.134, 50.343, -74.9874);
+
+ filter.setValue(false).setExists(true);
+ func.streamDoubles( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ filter.setValue(true).setExists(true);
+ Iterator<Double> values3 = Arrays.asList(4.3, -10.134, 50.343, -74.9874).iterator();
+ func.streamDoubles( value -> {
+ assertTrue(values3.hasNext());
+ assertEquals(values3.next(), value, .0000001);
+ });
+ assertFalse(values3.hasNext());
+
+ filter.setValue(false).setExists(true);
+ func.streamDoubles( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+ }
+
+ @Test
+ public void multiValueDateTest() throws DateTimeParseException {
+ Date date1 = Date.from(Instant.parse("1810-12-02T10:30:15Z"));
+ Date date2 = Date.from(Instant.parse("1931-03-16T18:15:45Z"));
+ Date date3 = Date.from(Instant.parse("2023-11-01T20:30:15Z"));
+
+ TestDateValueStream val = new TestDateValueStream();
+ TestBooleanValue filter = new TestBooleanValue();
+
+ AnalyticsValueStream uncasted = FilterFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filter});
+ assertTrue(uncasted instanceof DateValueStream);
+ DateValueStream func = (DateValueStream) uncasted;
+
+ // No values
+ val.setValues();
+ filter.setExists(false);
+ func.streamLongs( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ val.setValues("1810-12-02T10:30:15Z", "1850-12-02T20:30:15Z");
+ filter.setExists(false);
+ func.streamLongs( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ val.setValues();
+ filter.setValue(true).setExists(true);
+ func.streamLongs( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // One value
+ val.setValues("1810-12-02T10:30:15Z");
+
+ filter.setValue(true).setExists(true);
+ Iterator<Long> values1 = Arrays.asList(date1.getTime()).iterator();
+ func.streamLongs( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next().longValue(), value);
+ });
+ assertFalse(values1.hasNext());
+
+ filter.setValue(false).setExists(true);
+ func.streamLongs( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ filter.setValue(true).setExists(true);
+ Iterator<Long> values2 = Arrays.asList(date1.getTime()).iterator();
+ func.streamLongs( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next().longValue(), value);
+ });
+ assertFalse(values2.hasNext());
+
+ // Multiple values
+ val.setValues("1810-12-02T10:30:15Z", "1931-03-16T18:15:45Z", "2023-11-01T20:30:15Z");
+
+ filter.setValue(false).setExists(true);
+ func.streamLongs( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ filter.setValue(true).setExists(true);
+ Iterator<Long> values3 = Arrays.asList(date1.getTime(), date2.getTime(), date3.getTime()).iterator();
+ func.streamLongs( value -> {
+ assertTrue(values3.hasNext());
+ assertEquals(values3.next().longValue(), value);
+ });
+ assertFalse(values3.hasNext());
+
+ filter.setValue(false).setExists(true);
+ func.streamLongs( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+ }
+
+ @Test
+ public void multiValueStringTest() {
+ TestStringValueStream val = new TestStringValueStream();
+ TestBooleanValue filter = new TestBooleanValue();
+
+ AnalyticsValueStream uncasted = FilterFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filter});
+ assertTrue(uncasted instanceof StringValueStream);
+ StringValueStream func = (StringValueStream) uncasted;
+
+ // No values
+ val.setValues();
+ filter.setExists(false);
+ func.streamStrings( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ val.setValues("abc", "123", "def", "456");
+ filter.setExists(false);
+ func.streamStrings( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ val.setValues();
+ filter.setValue(true).setExists(true);
+ func.streamStrings( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // One value
+ val.setValues("abcdef");
+
+ filter.setValue(true).setExists(true);
+ Iterator<String> values1 = Arrays.asList("abcdef").iterator();
+ func.streamStrings( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next(), value);
+ });
+ assertFalse(values1.hasNext());
+
+ filter.setValue(false).setExists(true);
+ func.streamStrings( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ filter.setValue(true).setExists(true);
+ Iterator<String> values2 = Arrays.asList("abcdef").iterator();
+ func.streamStrings( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next(), value);
+ });
+ assertFalse(values2.hasNext());
+
+ // Multiple values
+ val.setValues("abc", "123", "def", "456");
+
+ filter.setValue(false).setExists(true);
+ func.streamStrings( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ filter.setValue(true).setExists(true);
+ Iterator<String> values3 = Arrays.asList("abc", "123", "def", "456").iterator();
+ func.streamStrings( value -> {
+ assertTrue(values3.hasNext());
+ assertEquals(values3.next(), value);
+ });
+ assertFalse(values3.hasNext());
+
+ filter.setValue(false).setExists(true);
+ func.streamStrings( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+ }
+
+ @Test
+ public void multiValueObjectTest() {
+ TestAnalyticsValueStream val = new TestAnalyticsValueStream();
+ TestBooleanValue filter = new TestBooleanValue();
+
+ AnalyticsValueStream func = FilterFunction.creatorFunction.apply(new AnalyticsValueStream[] {val, filter});
+
+ // No values
+ val.setValues();
+ filter.setExists(false);
+ func.streamObjects( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ val.setValues(3, "3", new Date(3));
+ filter.setExists(false);
+ func.streamObjects( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ val.setValues();
+ filter.setValue(true).setExists(true);
+ func.streamObjects( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // One value
+ Object obj = new TestAnalyticsValueStream();
+ val.setValues(obj);
+
+ filter.setValue(true).setExists(true);
+ Iterator<Object> values1 = Arrays.<Object>asList(obj).iterator();
+ func.streamObjects( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next(), value);
+ });
+ assertFalse(values1.hasNext());
+
+ filter.setValue(false).setExists(true);
+ func.streamObjects( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ filter.setValue(true).setExists(true);
+ Iterator<Object> values2 = Arrays.<Object>asList(obj).iterator();
+ func.streamObjects( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next(), value);
+ });
+ assertFalse(values2.hasNext());
+
+ // Multiple values
+ val.setValues(3, "3", new Date(3));
+
+ filter.setValue(false).setExists(true);
+ func.streamObjects( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ filter.setValue(true).setExists(true);
+ Iterator<Object> values3 = Arrays.<Object>asList(3, "3", new Date(3)).iterator();
+ func.streamObjects( value -> {
+ assertTrue(values3.hasNext());
+ assertEquals(values3.next(), value);
+ });
+ assertFalse(values3.hasNext());
+
+ filter.setValue(false).setExists(true);
+ func.streamObjects( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+ }
+}
http://git-wip-us.apache.org/repos/asf/lucene-solr/blob/f0366b94/solr/contrib/analytics/src/test/java/org/apache/solr/analytics/function/mapping/FloorFunctionTest.java
----------------------------------------------------------------------
diff --git a/solr/contrib/analytics/src/test/java/org/apache/solr/analytics/function/mapping/FloorFunctionTest.java b/solr/contrib/analytics/src/test/java/org/apache/solr/analytics/function/mapping/FloorFunctionTest.java
new file mode 100644
index 0000000..ba94fb6
--- /dev/null
+++ b/solr/contrib/analytics/src/test/java/org/apache/solr/analytics/function/mapping/FloorFunctionTest.java
@@ -0,0 +1,180 @@
+/*
+ * 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.solr.analytics.function.mapping;
+
+import java.util.Arrays;
+import java.util.Iterator;
+
+import org.apache.solr.SolrTestCaseJ4;
+import org.apache.solr.analytics.function.mapping.DecimalNumericConversionFunction.FloorFunction;
+import org.apache.solr.analytics.value.AnalyticsValueStream;
+import org.apache.solr.analytics.value.IntValue;
+import org.apache.solr.analytics.value.IntValueStream;
+import org.apache.solr.analytics.value.LongValue;
+import org.apache.solr.analytics.value.LongValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestDoubleValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestDoubleValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestFloatValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestFloatValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestIntValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestIntValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestLongValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestLongValueStream;
+import org.junit.Test;
+
+public class FloorFunctionTest extends SolrTestCaseJ4 {
+
+ @Test
+ public void singleValueFloatParameterTest() {
+ TestFloatValue val = new TestFloatValue();
+
+ AnalyticsValueStream uncasted = FloorFunction.creatorFunction.apply(new AnalyticsValueStream[] {val});
+ assertTrue(uncasted instanceof IntValue);
+ IntValue func = (IntValue) uncasted;
+
+ // Value doesn't exist
+ val.setExists(false);
+ func.getInt();
+ assertFalse(func.exists());
+
+ // Value exists
+ val.setValue(21.56F).setExists(true);
+ assertEquals(21, func.getInt());
+ assertTrue(func.exists());
+
+ val.setValue(-100.3F).setExists(true);
+ assertEquals(-101, func.getInt());
+ assertTrue(func.exists());
+ }
+
+ @Test
+ public void singleValueDoubleParameterTest() {
+ TestDoubleValue val = new TestDoubleValue();
+
+ AnalyticsValueStream uncasted = FloorFunction.creatorFunction.apply(new AnalyticsValueStream[] {val});
+ assertTrue(uncasted instanceof LongValue);
+ LongValue func = (LongValue) uncasted;
+
+ // Value doesn't exist
+ val.setExists(false);
+ func.getLong();
+ assertFalse(func.exists());
+
+ // Value exists
+ val.setValue(21.56).setExists(true);
+ assertEquals(21L, func.getLong());
+ assertTrue(func.exists());
+
+ val.setValue(-100.3).setExists(true);
+ assertEquals(-101L, func.getLong());
+ assertTrue(func.exists());
+ }
+
+ @Test
+ public void multiValueFloatParameterTest() {
+ TestFloatValueStream val = new TestFloatValueStream();
+
+ AnalyticsValueStream uncasted = FloorFunction.creatorFunction.apply(new AnalyticsValueStream[] {val});
+ assertTrue(uncasted instanceof IntValueStream);
+ IntValueStream func = (IntValueStream) uncasted;
+
+ // No values
+ val.setValues();
+ func.streamInts( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // One value
+ val.setValues(-4F);
+ Iterator<Integer> values1 = Arrays.asList(-4).iterator();
+ func.streamInts( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next().intValue(), value);
+ });
+ assertFalse(values1.hasNext());
+
+ // Multiple values
+ val.setValues(4F, -10.9999F, 50.00001F, 74.99999F);
+ Iterator<Integer> values2 = Arrays.asList(4, -11, 50, 74).iterator();
+ func.streamInts( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next().intValue(), value);
+ });
+ assertFalse(values2.hasNext());
+ }
+
+ @Test
+ public void multiValueDoubleParameterTest() {
+ TestDoubleValueStream val = new TestDoubleValueStream();
+
+ AnalyticsValueStream uncasted = FloorFunction.creatorFunction.apply(new AnalyticsValueStream[] {val});
+ assertTrue(uncasted instanceof LongValueStream);
+ LongValueStream func = (LongValueStream) uncasted;
+
+ // No values
+ val.setValues();
+ func.streamLongs( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // One value
+ val.setValues(-4);
+ Iterator<Long> values1 = Arrays.asList(-4L).iterator();
+ func.streamLongs( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next().longValue(), value);
+ });
+ assertFalse(values1.hasNext());
+
+ // Multiple values
+ val.setValues(4, -10.9999, 50.000001, 74.99999);
+ Iterator<Long> values2 = Arrays.asList(4L, -11L, 50L, 74L).iterator();
+ func.streamLongs( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next().longValue(), value);
+ });
+ assertFalse(values2.hasNext());
+ }
+
+ @Test
+ public void nonDecimalParameterTest() {
+ AnalyticsValueStream result;
+ AnalyticsValueStream param;
+
+ param = new TestIntValue();
+ result = FloorFunction.creatorFunction.apply(new AnalyticsValueStream[] {param});
+ assertTrue(result instanceof IntValue);
+ assertEquals(param, result);
+
+ param = new TestIntValueStream();
+ result = FloorFunction.creatorFunction.apply(new AnalyticsValueStream[] {param});
+ assertTrue(result instanceof IntValueStream);
+ assertEquals(param, result);
+
+ param = new TestLongValue();
+ result = FloorFunction.creatorFunction.apply(new AnalyticsValueStream[] {param});
+ assertTrue(result instanceof LongValue);
+ assertEquals(param, result);
+
+ param = new TestLongValueStream();
+ result = FloorFunction.creatorFunction.apply(new AnalyticsValueStream[] {param});
+ assertTrue(result instanceof LongValueStream);
+ assertEquals(param, result);
+ }
+
+
+}
http://git-wip-us.apache.org/repos/asf/lucene-solr/blob/f0366b94/solr/contrib/analytics/src/test/java/org/apache/solr/analytics/function/mapping/GTEFunctionTest.java
----------------------------------------------------------------------
diff --git a/solr/contrib/analytics/src/test/java/org/apache/solr/analytics/function/mapping/GTEFunctionTest.java b/solr/contrib/analytics/src/test/java/org/apache/solr/analytics/function/mapping/GTEFunctionTest.java
new file mode 100644
index 0000000..b6883c5
--- /dev/null
+++ b/solr/contrib/analytics/src/test/java/org/apache/solr/analytics/function/mapping/GTEFunctionTest.java
@@ -0,0 +1,290 @@
+/*
+ * 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.solr.analytics.function.mapping;
+
+import java.util.Arrays;
+import java.util.Iterator;
+
+import org.apache.solr.SolrTestCaseJ4;
+import org.apache.solr.analytics.function.mapping.ComparisonFunction.GTEFunction;
+import org.apache.solr.analytics.value.AnalyticsValueStream;
+import org.apache.solr.analytics.value.BooleanValue;
+import org.apache.solr.analytics.value.BooleanValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestDateValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestDateValueStream;
+import org.apache.solr.analytics.value.FillableTestValue.TestDoubleValue;
+import org.apache.solr.analytics.value.FillableTestValue.TestDoubleValueStream;
+import org.junit.Test;
+
+public class GTEFunctionTest extends SolrTestCaseJ4 {
+
+ @Test
+ public void singleValueNumericParameterTest() {
+ TestDoubleValue base = new TestDoubleValue();
+ TestDoubleValue comp = new TestDoubleValue();
+
+ AnalyticsValueStream uncasted = GTEFunction.creatorFunction.apply(new AnalyticsValueStream[] {base, comp});
+ assertTrue(uncasted instanceof BooleanValue);
+ BooleanValue func = (BooleanValue) uncasted;
+
+ // Value doesn't exist
+ base.setExists(false);
+ comp.setExists(false);
+ func.getBoolean();
+ assertFalse(func.exists());
+
+ base.setExists(false);
+ comp.setValue(3.3).setExists(true);
+ func.getBoolean();
+ assertFalse(func.exists());
+
+ base.setValue(3.3).setExists(true);
+ comp.setExists(false);
+ func.getBoolean();
+ assertFalse(func.exists());
+
+ // Value exists
+ base.setValue(21.56).setExists(true);
+ comp.setValue(21.56).setExists(true);
+ assertEquals(true, func.getBoolean());
+ assertTrue(func.exists());
+
+ base.setValue(21.56).setExists(true);
+ comp.setValue(21.57).setExists(true);
+ assertEquals(false, func.getBoolean());
+ assertTrue(func.exists());
+
+ base.setValue(21.56).setExists(true);
+ comp.setValue(-21.57).setExists(true);
+ assertEquals(true, func.getBoolean());
+ assertTrue(func.exists());
+ }
+
+ @Test
+ public void singleValueDateParameterTest() {
+ TestDateValue base = new TestDateValue();
+ TestDateValue comp = new TestDateValue();
+
+ AnalyticsValueStream uncasted = GTEFunction.creatorFunction.apply(new AnalyticsValueStream[] {base, comp});
+ assertTrue(uncasted instanceof BooleanValue);
+ BooleanValue func = (BooleanValue) uncasted;
+
+ // Value doesn't exist
+ base.setExists(false);
+ comp.setExists(false);
+ func.getBoolean();
+ assertFalse(func.exists());
+
+ base.setExists(false);
+ comp.setValue("1800-01-02T10:20:30Z").setExists(true);
+ func.getBoolean();
+ assertFalse(func.exists());
+
+ base.setValue("1800-01-02T10:20:30Z").setExists(true);
+ comp.setExists(false);
+ func.getBoolean();
+ assertFalse(func.exists());
+
+ // Value exists
+ base.setValue("1800-01-02T10:20:30Z").setExists(true);
+ comp.setValue("1800-01-02T10:20:30Z").setExists(true);
+ assertEquals(true, func.getBoolean());
+ assertTrue(func.exists());
+
+ base.setValue("1800-01-02T10:20:30Z").setExists(true);
+ comp.setValue("1800-01-02T10:20:31Z").setExists(true);
+ assertEquals(false, func.getBoolean());
+ assertTrue(func.exists());
+
+ base.setValue("1800-01-02T10:20:30Z").setExists(true);
+ comp.setValue("1000-01-02T10:20:31Z").setExists(true);
+ assertEquals(true, func.getBoolean());
+ assertTrue(func.exists());
+ }
+
+ @Test
+ public void oneSingleOneMultiValueNumericParameterTest() {
+ TestDoubleValue base = new TestDoubleValue();
+ TestDoubleValueStream comp = new TestDoubleValueStream();
+
+ AnalyticsValueStream uncasted = GTEFunction.creatorFunction.apply(new AnalyticsValueStream[] {base, comp});
+ assertTrue(uncasted instanceof BooleanValueStream);
+ BooleanValueStream func = (BooleanValueStream) uncasted;
+
+ // No values
+ base.setValue(-4.2).setExists(true);
+ comp.setValues();
+ func.streamBooleans( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ base.setExists(false);
+ comp.setValues(-4.2);
+ func.streamBooleans( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // One value
+ base.setValue(-4.2).setExists(true);
+ comp.setValues(-4);
+ Iterator<Boolean> values1 = Arrays.asList(false).iterator();
+ func.streamBooleans( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next(), value);
+ });
+ assertFalse(values1.hasNext());
+
+ // Multiple values
+ base.setValue(4).setExists(true);
+ comp.setValues(4, -10, 2345, -74, 4.0001);
+ Iterator<Boolean> values2 = Arrays.asList(true, true, false, true, false).iterator();
+ func.streamBooleans( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next(), value);
+ });
+ assertFalse(values2.hasNext());
+ }
+
+ @Test
+ public void oneMultiOneSingleValueNumericParameterTest() {
+ TestDoubleValueStream base = new TestDoubleValueStream();
+ TestDoubleValue comp = new TestDoubleValue();
+
+ AnalyticsValueStream uncasted = GTEFunction.creatorFunction.apply(new AnalyticsValueStream[] {base, comp});
+ assertTrue(uncasted instanceof BooleanValueStream);
+ BooleanValueStream func = (BooleanValueStream) uncasted;
+
+ // No values
+ base.setValues();
+ comp.setValue(-4.2).setExists(true);
+ func.streamBooleans( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ base.setValues(-4.2);
+ comp.setExists(false);
+ func.streamBooleans( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // One value
+ base.setValues(-4.2);
+ comp.setValue(-4).setExists(true);
+ Iterator<Boolean> values1 = Arrays.asList(false).iterator();
+ func.streamBooleans( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next(), value);
+ });
+ assertFalse(values1.hasNext());
+
+ // Multiple values
+ base.setValues(4, -10, 2345, -74, 4.0001);
+ comp.setValue(4).setExists(true);
+ Iterator<Boolean> values2 = Arrays.asList(true, false, true, false, true).iterator();
+ func.streamBooleans( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next(), value);
+ });
+ assertFalse(values2.hasNext());
+ }
+
+ @Test
+ public void oneSingleOneMultiValueDateParameterTest() {
+ TestDateValue base = new TestDateValue();
+ TestDateValueStream comp = new TestDateValueStream();
+
+ AnalyticsValueStream uncasted = GTEFunction.creatorFunction.apply(new AnalyticsValueStream[] {base, comp});
+ assertTrue(uncasted instanceof BooleanValueStream);
+ BooleanValueStream func = (BooleanValueStream) uncasted;
+
+ // No values
+ base.setValue("1800-01-02T10:20:30Z").setExists(true);
+ comp.setValues();
+ func.streamBooleans( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ base.setExists(false);
+ comp.setValues("1800-01-02T10:20:30Z");
+ func.streamBooleans( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // One value
+ base.setValue("1803-01-02T10:20:30Z").setExists(true);
+ comp.setValues("1800-01-02T10:20:30Z");
+ Iterator<Boolean> values1 = Arrays.asList(true).iterator();
+ func.streamBooleans( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next(), value);
+ });
+ assertFalse(values1.hasNext());
+
+ // Multiple values
+ base.setValue("1800-01-02T10:20:30Z").setExists(true);
+ comp.setValues("1800-03-02T10:20:30Z", "1799-01-01T10:20:29Z", "1800-01-02T10:20:31Z", "1800-01-02T10:20:30Z", "1800-01-02T10:20:29Z");
+ Iterator<Boolean> values2 = Arrays.asList(false, true, false, true, true).iterator();
+ func.streamBooleans( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next(), value);
+ });
+ assertFalse(values2.hasNext());
+ }
+
+ @Test
+ public void oneMultiOneSingleValueDateParameterTest() {
+ TestDateValueStream base = new TestDateValueStream();
+ TestDateValue comp = new TestDateValue();
+
+ AnalyticsValueStream uncasted = GTEFunction.creatorFunction.apply(new AnalyticsValueStream[] {base, comp});
+ assertTrue(uncasted instanceof BooleanValueStream);
+ BooleanValueStream func = (BooleanValueStream) uncasted;
+
+ // No values
+ base.setValues();
+ comp.setValue("1800-01-02T10:20:30Z").setExists(true);
+ func.streamBooleans( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ base.setValues("1800-01-02T10:20:30Z");
+ comp.setExists(false);
+ func.streamBooleans( value -> {
+ assertTrue("There should be no values to stream", false);
+ });
+
+ // One value
+ base.setValues("1800-01-02T10:20:30Z");
+ comp.setValue("1803-01-02T10:20:30Z").setExists(true);
+ Iterator<Boolean> values1 = Arrays.asList(false).iterator();
+ func.streamBooleans( value -> {
+ assertTrue(values1.hasNext());
+ assertEquals(values1.next(), value);
+ });
+ assertFalse(values1.hasNext());
+
+ // Multiple values
+ base.setValues("1800-03-02T10:20:30Z", "1799-01-01T10:20:29Z", "1800-01-02T10:20:31Z", "1800-01-02T10:20:30Z", "1800-01-02T10:20:29Z");
+ comp.setValue("1800-01-02T10:20:30Z").setExists(true);
+ Iterator<Boolean> values2 = Arrays.asList(true, false, true, true, false).iterator();
+ func.streamBooleans( value -> {
+ assertTrue(values2.hasNext());
+ assertEquals(values2.next(), value);
+ });
+ assertFalse(values2.hasNext());
+ }
+}