summaryrefslogtreecommitdiffstats
path: root/container-search/src/test/java/com/yahoo/search/grouping
diff options
context:
space:
mode:
authorJon Bratseth <bratseth@gmail.com>2022-01-03 15:20:03 +0100
committerJon Bratseth <bratseth@gmail.com>2022-01-03 15:20:03 +0100
commita9e52aa32101dc5ce39c5caa1d80c9cd7aa7684a (patch)
tree80bd9aa551ee9de7afaffee23130e7fb432488f9 /container-search/src/test/java/com/yahoo/search/grouping
parent1f3e5d1003a33d9a7076575eab8059c582611cdb (diff)
Avoid deprecated APIs
Diffstat (limited to 'container-search/src/test/java/com/yahoo/search/grouping')
-rw-r--r--container-search/src/test/java/com/yahoo/search/grouping/GroupingValidatorTestCase.java117
-rw-r--r--container-search/src/test/java/com/yahoo/search/grouping/UniqueGroupingSearcherTestCase.java9
-rw-r--r--container-search/src/test/java/com/yahoo/search/grouping/request/MathFunctionsTestCase.java100
3 files changed, 124 insertions, 102 deletions
diff --git a/container-search/src/test/java/com/yahoo/search/grouping/GroupingValidatorTestCase.java b/container-search/src/test/java/com/yahoo/search/grouping/GroupingValidatorTestCase.java
index 8f757a0dd49..4f473b29918 100644
--- a/container-search/src/test/java/com/yahoo/search/grouping/GroupingValidatorTestCase.java
+++ b/container-search/src/test/java/com/yahoo/search/grouping/GroupingValidatorTestCase.java
@@ -7,117 +7,142 @@ import com.yahoo.search.Query;
import com.yahoo.search.config.ClusterConfig;
import com.yahoo.search.grouping.request.GroupingOperation;
import com.yahoo.search.searchchain.Execution;
-import org.junit.Rule;
import org.junit.Test;
-import org.junit.rules.ExpectedException;
-import java.util.Arrays;
import java.util.Collection;
+import java.util.List;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.fail;
/**
* @author Simon Thoresen Hult
*/
public class GroupingValidatorTestCase {
- @Rule
- public ExpectedException thrown = ExpectedException.none();
-
@Test
public void requireThatAvailableAttributesDoNotThrow() {
- validateGrouping(Arrays.asList("foo", "bar"),
+ validateGrouping(List.of("foo", "bar"),
"all(group(foo) each(output(max(bar))))");;
}
@Test
public void requireThatUnavailableAttributesThrow() {
- thrown.expect(UnavailableAttributeException.class);
- thrown.expectMessage(createMessage("bar"));
- validateGrouping(Arrays.asList("foo"),
- "all(group(foo) each(output(max(bar))))");
+ try {
+ validateGrouping(List.of("foo"), "all(group(foo) each(output(max(bar))))");
+ fail("Excpected exception");
+ }
+ catch (UnavailableAttributeException e) {
+ assertEquals(createMessage("bar"), e.getMessage());
+ }
}
@Test
public void requireThatEnableFlagPreventsThrow() {
Query query = createQuery("all(group(foo) each(output(max(bar))))");
query.properties().set(GroupingValidator.PARAM_ENABLED, "false");
- validateGrouping(Arrays.asList("foo"), query);
+ validateGrouping(List.of("foo"), query);
}
@Test
public void available_primitive_map_attribute_does_not_throw() {
- validateGrouping(Arrays.asList("map.key", "map.value"),
+ validateGrouping(List.of("map.key", "map.value"),
"all(group(map{\"foo\"}) each(output(count())))");
}
@Test
public void unavailable_primitive_map_key_attribute_throws() {
- thrown.expect(UnavailableAttributeException.class);
- thrown.expectMessage(createMessage("map.key"));
- validateGrouping(Arrays.asList("null"),
- "all(group(map{\"foo\"}) each(output(count())))");
+ try {
+ validateGrouping(List.of("null"), "all(group(map{\"foo\"}) each(output(count())))");
+ fail("Expected exception");
+ }
+ catch (UnavailableAttributeException e) {
+ assertEquals(createMessage("map.key"), e.getMessage());
+ }
}
@Test
public void unavailable_primitive_map_value_attribute_throws() {
- thrown.expect(UnavailableAttributeException.class);
- thrown.expectMessage(createMessage("map.value"));
- validateGrouping(Arrays.asList("map.key"),
- "all(group(map{\"foo\"}) each(output(count())))");
+ try {
+ validateGrouping(List.of("map.key"), "all(group(map{\"foo\"}) each(output(count())))");
+ fail("Expected exception");
+ }
+ catch (UnavailableAttributeException e) {
+ assertEquals(createMessage("map.value"), e.getMessage());
+ }
}
@Test
public void available_struct_map_attribute_does_not_throw() {
- validateGrouping(Arrays.asList("map.key", "map.value.name"),
+ validateGrouping(List.of("map.key", "map.value.name"),
"all(group(map{\"foo\"}.name) each(output(count())))");
}
@Test
public void unavailable_struct_map_key_attribute_throws() {
- thrown.expect(UnavailableAttributeException.class);
- thrown.expectMessage(createMessage("map.key"));
- validateGrouping(Arrays.asList("null"),
- "all(group(map{\"foo\"}.name) each(output(count())))");
+ try {
+ validateGrouping(List.of("null"), "all(group(map{\"foo\"}.name) each(output(count())))");
+ fail("Expected exception");
+ }
+ catch (UnavailableAttributeException e) {
+ assertEquals(createMessage("map.key"), e.getMessage());
+ }
}
@Test
public void unavailable_struct_map_value_attribute_throws() {
- thrown.expect(UnavailableAttributeException.class);
- thrown.expectMessage(createMessage("map.value.name"));
- validateGrouping(Arrays.asList("map.key"),
- "all(group(map{\"foo\"}.name) each(output(count())))");
+ try {
+ validateGrouping(List.of("map.key"), "all(group(map{\"foo\"}.name) each(output(count())))");
+ fail("Expected exception");
+ }
+ catch (UnavailableAttributeException e) {
+ assertEquals(createMessage("map.value.name"), e.getMessage());
+ }
}
@Test
public void available_key_source_attribute_does_not_throw() {
- validateGrouping(Arrays.asList("map.key", "map.value", "key_source"),
+ validateGrouping(List.of("map.key", "map.value", "key_source"),
"all(group(map{attribute(key_source)}) each(output(count())))");
}
@Test
public void unavailable_key_source_attribute_throws() {
- thrown.expect(UnavailableAttributeException.class);
- thrown.expectMessage(createMessage("key_source"));
- validateGrouping(Arrays.asList("map.key", "map.value"),
- "all(group(map{attribute(key_source)}) each(output(count())))");
+ try {
+ validateGrouping(List.of("map.key", "map.value"),
+ "all(group(map{attribute(key_source)}) each(output(count())))");
+ fail("Expected exception");
+ }
+ catch (UnavailableAttributeException e) {
+ assertEquals(createMessage("key_source"), e.getMessage());
+ }
}
@Test
public void key_source_attribute_with_mismatching_data_type_throws() {
- thrown.expect(IllegalArgumentException.class);
- thrown.expectMessage("Grouping request references key source attribute 'key_source' with data type 'INT32' " +
- "that is different than data type 'STRING' of key attribute 'map.key'");
-
- validateGrouping(setupMismatchingKeySourceAttribute(false),
- "all(group(map{attribute(key_source)}) each(output(count())))");
+ try {
+ validateGrouping(setupMismatchingKeySourceAttribute(false),
+ "all(group(map{attribute(key_source)}) each(output(count())))");
+ fail("Expected exception");
+ }
+ catch (IllegalArgumentException e) {
+ assertEquals("Grouping request references key source attribute 'key_source' with data type 'INT32' " +
+ "that is different than data type 'STRING' of key attribute 'map.key'",
+ e.getMessage());
+ }
}
@Test
public void key_source_attribute_with_multi_value_collection_type_throws() {
- thrown.expect(IllegalArgumentException.class);
- thrown.expectMessage("Grouping request references key source attribute 'key_source' which is not of single value type");
-
- validateGrouping(setupMismatchingKeySourceAttribute(true),
- "all(group(map{attribute(key_source)}) each(output(count())))");
+ try {
+ validateGrouping(setupMismatchingKeySourceAttribute(true),
+ "all(group(map{attribute(key_source)}) each(output(count())))");
+ fail("Expected exception");
+ }
+ catch (IllegalArgumentException e) {
+ assertEquals("Grouping request references key source attribute 'key_source' which is not of single value type",
+ e.getMessage());
+ }
}
private static AttributesConfig setupMismatchingKeySourceAttribute(boolean matchingDataType) {
diff --git a/container-search/src/test/java/com/yahoo/search/grouping/UniqueGroupingSearcherTestCase.java b/container-search/src/test/java/com/yahoo/search/grouping/UniqueGroupingSearcherTestCase.java
index c5b608db4e2..3b95e44ad70 100644
--- a/container-search/src/test/java/com/yahoo/search/grouping/UniqueGroupingSearcherTestCase.java
+++ b/container-search/src/test/java/com/yahoo/search/grouping/UniqueGroupingSearcherTestCase.java
@@ -20,6 +20,7 @@ import org.junit.Test;
import static org.hamcrest.CoreMatchers.containsString;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertThat;
+import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
/**
@@ -49,11 +50,9 @@ public class UniqueGroupingSearcherTestCase {
fail("Above statement should throw");
} catch (IllegalArgumentException e) {
// As expected.
- assertThat(
- Exceptions.toMessageString(e),
- containsString(
- "Could not set 'ranking.sorting' to '-1': " +
- "Illegal attribute name '1' for sorting. Requires '[\\[]*[a-zA-Z_][\\.a-zA-Z0-9_-]*[\\]]*'"));
+ assertTrue(Exceptions.toMessageString(e).contains("Could not set 'ranking.sorting' to '-1': " +
+ "Illegal attribute name '1' for sorting. " +
+ "Requires '[\\[]*[a-zA-Z_][\\.a-zA-Z0-9_-]*[\\]]*'"));
}
}
diff --git a/container-search/src/test/java/com/yahoo/search/grouping/request/MathFunctionsTestCase.java b/container-search/src/test/java/com/yahoo/search/grouping/request/MathFunctionsTestCase.java
index 522e72d31b9..d22c2dee7b6 100644
--- a/container-search/src/test/java/com/yahoo/search/grouping/request/MathFunctionsTestCase.java
+++ b/container-search/src/test/java/com/yahoo/search/grouping/request/MathFunctionsTestCase.java
@@ -3,65 +3,63 @@ package com.yahoo.search.grouping.request;
import org.junit.Test;
-import static org.hamcrest.CoreMatchers.instanceOf;
-import static org.hamcrest.CoreMatchers.is;
-import static org.hamcrest.CoreMatchers.sameInstance;
-import static org.junit.Assert.assertThat;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertSame;
+import static org.junit.Assert.assertTrue;
/**
- * @author <a href="mailto:einarmr@yahoo-inc.com">Einar M R Rosenvinge</a>
- * @since 5.1.9
+ * @author Einar M R Rosenvinge
*/
public class MathFunctionsTestCase {
+
@Test
public void testMathFunctions() {
//if this fails, update the count AND add a test in each of the two blocks below
- assertThat(MathFunctions.Function.values().length, is(21));
+ assertEquals(21, MathFunctions.Function.values().length);
+ assertSame(MathFunctions.Function.create(0), MathFunctions.Function.EXP);
+ assertSame(MathFunctions.Function.create(1), MathFunctions.Function.POW);
+ assertSame(MathFunctions.Function.create(2), MathFunctions.Function.LOG);
+ assertSame(MathFunctions.Function.create(3), MathFunctions.Function.LOG1P);
+ assertSame(MathFunctions.Function.create(4), MathFunctions.Function.LOG10);
+ assertSame(MathFunctions.Function.create(5), MathFunctions.Function.SIN);
+ assertSame(MathFunctions.Function.create(6), MathFunctions.Function.ASIN);
+ assertSame(MathFunctions.Function.create(7), MathFunctions.Function.COS);
+ assertSame(MathFunctions.Function.create(8), MathFunctions.Function.ACOS);
+ assertSame(MathFunctions.Function.create(9), MathFunctions.Function.TAN);
+ assertSame(MathFunctions.Function.create(10), MathFunctions.Function.ATAN);
+ assertSame(MathFunctions.Function.create(11), MathFunctions.Function.SQRT);
+ assertSame(MathFunctions.Function.create(12), MathFunctions.Function.SINH);
+ assertSame(MathFunctions.Function.create(13), MathFunctions.Function.ASINH);
+ assertSame(MathFunctions.Function.create(14), MathFunctions.Function.COSH);
+ assertSame(MathFunctions.Function.create(15), MathFunctions.Function.ACOSH);
+ assertSame(MathFunctions.Function.create(16), MathFunctions.Function.TANH);
+ assertSame(MathFunctions.Function.create(17), MathFunctions.Function.ATANH);
+ assertSame(MathFunctions.Function.create(18), MathFunctions.Function.CBRT);
+ assertSame(MathFunctions.Function.create(19), MathFunctions.Function.HYPOT);
+ assertSame(MathFunctions.Function.create(20), MathFunctions.Function.FLOOR);
- assertThat(MathFunctions.Function.create(0), sameInstance(MathFunctions.Function.EXP));
- assertThat(MathFunctions.Function.create(1), sameInstance(MathFunctions.Function.POW));
- assertThat(MathFunctions.Function.create(2), sameInstance(MathFunctions.Function.LOG));
- assertThat(MathFunctions.Function.create(3), sameInstance(MathFunctions.Function.LOG1P));
- assertThat(MathFunctions.Function.create(4), sameInstance(MathFunctions.Function.LOG10));
- assertThat(MathFunctions.Function.create(5), sameInstance(MathFunctions.Function.SIN));
- assertThat(MathFunctions.Function.create(6), sameInstance(MathFunctions.Function.ASIN));
- assertThat(MathFunctions.Function.create(7), sameInstance(MathFunctions.Function.COS));
- assertThat(MathFunctions.Function.create(8), sameInstance(MathFunctions.Function.ACOS));
- assertThat(MathFunctions.Function.create(9), sameInstance(MathFunctions.Function.TAN));
- assertThat(MathFunctions.Function.create(10), sameInstance(MathFunctions.Function.ATAN));
- assertThat(MathFunctions.Function.create(11), sameInstance(MathFunctions.Function.SQRT));
- assertThat(MathFunctions.Function.create(12), sameInstance(MathFunctions.Function.SINH));
- assertThat(MathFunctions.Function.create(13), sameInstance(MathFunctions.Function.ASINH));
- assertThat(MathFunctions.Function.create(14), sameInstance(MathFunctions.Function.COSH));
- assertThat(MathFunctions.Function.create(15), sameInstance(MathFunctions.Function.ACOSH));
- assertThat(MathFunctions.Function.create(16), sameInstance(MathFunctions.Function.TANH));
- assertThat(MathFunctions.Function.create(17), sameInstance(MathFunctions.Function.ATANH));
- assertThat(MathFunctions.Function.create(18), sameInstance(MathFunctions.Function.CBRT));
- assertThat(MathFunctions.Function.create(19), sameInstance(MathFunctions.Function.HYPOT));
- assertThat(MathFunctions.Function.create(20), sameInstance(MathFunctions.Function.FLOOR));
-
-
- assertThat(MathFunctions.newInstance(MathFunctions.Function.EXP, null, null), instanceOf(MathExpFunction.class));
- assertThat(MathFunctions.newInstance(MathFunctions.Function.POW, null, null), instanceOf(MathPowFunction.class));
- assertThat(MathFunctions.newInstance(MathFunctions.Function.LOG, null, null), instanceOf(MathLogFunction.class));
- assertThat(MathFunctions.newInstance(MathFunctions.Function.LOG1P, null, null), instanceOf(MathLog1pFunction.class));
- assertThat(MathFunctions.newInstance(MathFunctions.Function.LOG10, null, null), instanceOf(MathLog10Function.class));
- assertThat(MathFunctions.newInstance(MathFunctions.Function.SIN, null, null), instanceOf(MathSinFunction.class));
- assertThat(MathFunctions.newInstance(MathFunctions.Function.ASIN, null, null), instanceOf(MathASinFunction.class));
- assertThat(MathFunctions.newInstance(MathFunctions.Function.COS, null, null), instanceOf(MathCosFunction.class));
- assertThat(MathFunctions.newInstance(MathFunctions.Function.ACOS, null, null), instanceOf(MathACosFunction.class));
- assertThat(MathFunctions.newInstance(MathFunctions.Function.TAN, null, null), instanceOf(MathTanFunction.class));
- assertThat(MathFunctions.newInstance(MathFunctions.Function.ATAN, null, null), instanceOf(MathATanFunction.class));
- assertThat(MathFunctions.newInstance(MathFunctions.Function.SQRT, null, null), instanceOf(MathSqrtFunction.class));
- assertThat(MathFunctions.newInstance(MathFunctions.Function.SINH, null, null), instanceOf(MathSinHFunction.class));
- assertThat(MathFunctions.newInstance(MathFunctions.Function.ASINH, null, null), instanceOf(MathASinHFunction.class));
- assertThat(MathFunctions.newInstance(MathFunctions.Function.COSH, null, null), instanceOf(MathCosHFunction.class));
- assertThat(MathFunctions.newInstance(MathFunctions.Function.ACOSH, null, null), instanceOf(MathACosHFunction.class));
- assertThat(MathFunctions.newInstance(MathFunctions.Function.TANH, null, null), instanceOf(MathTanHFunction.class));
- assertThat(MathFunctions.newInstance(MathFunctions.Function.ATANH, null, null), instanceOf(MathATanHFunction.class));
- assertThat(MathFunctions.newInstance(MathFunctions.Function.CBRT, null, null), instanceOf(MathCbrtFunction.class));
- assertThat(MathFunctions.newInstance(MathFunctions.Function.HYPOT, null, null), instanceOf(MathHypotFunction.class));
- assertThat(MathFunctions.newInstance(MathFunctions.Function.FLOOR, null, null), instanceOf(MathFloorFunction.class));
+ assertTrue(MathFunctions.newInstance(MathFunctions.Function.EXP, null, null) instanceof MathExpFunction);
+ assertTrue(MathFunctions.newInstance(MathFunctions.Function.POW, null, null) instanceof MathPowFunction);
+ assertTrue(MathFunctions.newInstance(MathFunctions.Function.LOG, null, null) instanceof MathLogFunction);
+ assertTrue(MathFunctions.newInstance(MathFunctions.Function.LOG1P, null, null) instanceof MathLog1pFunction);
+ assertTrue(MathFunctions.newInstance(MathFunctions.Function.LOG10, null, null) instanceof MathLog10Function);
+ assertTrue(MathFunctions.newInstance(MathFunctions.Function.SIN, null, null) instanceof MathSinFunction);
+ assertTrue(MathFunctions.newInstance(MathFunctions.Function.ASIN, null, null) instanceof MathASinFunction);
+ assertTrue(MathFunctions.newInstance(MathFunctions.Function.COS, null, null) instanceof MathCosFunction);
+ assertTrue(MathFunctions.newInstance(MathFunctions.Function.ACOS, null, null) instanceof MathACosFunction);
+ assertTrue(MathFunctions.newInstance(MathFunctions.Function.TAN, null, null) instanceof MathTanFunction);
+ assertTrue(MathFunctions.newInstance(MathFunctions.Function.ATAN, null, null) instanceof MathATanFunction);
+ assertTrue(MathFunctions.newInstance(MathFunctions.Function.SQRT, null, null) instanceof MathSqrtFunction);
+ assertTrue(MathFunctions.newInstance(MathFunctions.Function.SINH, null, null) instanceof MathSinHFunction);
+ assertTrue(MathFunctions.newInstance(MathFunctions.Function.ASINH, null, null) instanceof MathASinHFunction);
+ assertTrue(MathFunctions.newInstance(MathFunctions.Function.COSH, null, null) instanceof MathCosHFunction);
+ assertTrue(MathFunctions.newInstance(MathFunctions.Function.ACOSH, null, null) instanceof MathACosHFunction);
+ assertTrue(MathFunctions.newInstance(MathFunctions.Function.TANH, null, null) instanceof MathTanHFunction);
+ assertTrue(MathFunctions.newInstance(MathFunctions.Function.ATANH, null, null) instanceof MathATanHFunction);
+ assertTrue(MathFunctions.newInstance(MathFunctions.Function.CBRT, null, null) instanceof MathCbrtFunction);
+ assertTrue(MathFunctions.newInstance(MathFunctions.Function.HYPOT, null, null) instanceof MathHypotFunction);
+ assertTrue(MathFunctions.newInstance(MathFunctions.Function.FLOOR, null, null) instanceof MathFloorFunction);
}
+
}