1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.search.grouping;
import com.yahoo.vespa.config.search.AttributesConfig;
import com.yahoo.container.QrSearchersConfig;
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.Test;
import java.util.Arrays;
import java.util.Collection;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail;
/**
* @author <a href="mailto:simon@yahoo-inc.com">Simon Thoresen</a>
*/
public class GroupingValidatorTestCase {
@Test
public void requireThatAvailableAttributesDoNotThrow() {
Query query = new Query();
GroupingRequest req = GroupingRequest.newInstance(query);
req.setRootOperation(GroupingOperation.fromString("all(group(foo) each(output(max(bar))))"));
validateGrouping("myCluster", Arrays.asList("foo", "bar"), query);
}
@Test
public void requireThatUnavailableAttributesThrow() {
Query query = new Query();
GroupingRequest req = GroupingRequest.newInstance(query);
req.setRootOperation(GroupingOperation.fromString("all(group(foo) each(output(max(bar))))"));
try {
validateGrouping("myCluster", Arrays.asList("foo"), query);
fail("Validator should throw exception because attribute 'bar' is unavailable.");
} catch (UnavailableAttributeException e) {
assertEquals("myCluster", e.getClusterName());
assertEquals("bar", e.getAttributeName());
}
}
@Test
public void requireThatEnableFlagPreventsThrow() {
Query query = new Query();
GroupingRequest req = GroupingRequest.newInstance(query);
req.setRootOperation(GroupingOperation.fromString("all(group(foo) each(output(max(bar))))"));
query.properties().set(GroupingValidator.PARAM_ENABLED, "false");
validateGrouping("myCluster", Arrays.asList("foo"), query);
}
private static void validateGrouping(String clusterName, Collection<String> attributeNames, Query query) {
QrSearchersConfig.Builder qrsConfig = new QrSearchersConfig.Builder().searchcluster(
new QrSearchersConfig.Searchcluster.Builder()
.indexingmode(QrSearchersConfig.Searchcluster.Indexingmode.Enum.REALTIME)
.name(clusterName));
ClusterConfig.Builder clusterConfig = new ClusterConfig.Builder().
clusterId(0).
clusterName("test");
AttributesConfig.Builder attributesConfig = new AttributesConfig.Builder();
for (String attributeName : attributeNames) {
attributesConfig.attribute(new AttributesConfig.Attribute.Builder()
.name(attributeName));
}
new Execution(
new GroupingValidator(new QrSearchersConfig(qrsConfig),
new ClusterConfig(clusterConfig),
new AttributesConfig(attributesConfig)),
Execution.Context.createContextStub()).search(query);
}
}
|
