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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
// Copyright 2017 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.model.container.search.searchchain;
import com.yahoo.component.ComponentId;
import com.yahoo.container.core.ChainsConfig;
import com.yahoo.prelude.cluster.ClusterSearcher;
import com.yahoo.search.config.ClusterConfig;
import com.yahoo.search.federation.ProviderConfig;
import com.yahoo.vespa.defaults.Defaults;
import org.junit.Before;
import org.junit.Test;
import org.w3c.dom.Element;
import java.util.List;
import static org.hamcrest.CoreMatchers.instanceOf;
import static org.hamcrest.CoreMatchers.is;
import static org.junit.Assert.*;
/**
* Test of search chains config
* <p>TODO: examine the actual values in the configs.</p>
* @author Tony Vaagenes
*/
public class SearchChainsTest extends SearchChainsTestBase {
private ChainsConfig chainsConfig;
private ProviderConfig providerConfig;
private ClusterConfig clusterConfig;
@Before
public void subscribe() {
ChainsConfig.Builder chainsBuilder = new ChainsConfig.Builder();
chainsBuilder = (ChainsConfig.Builder)root.getConfig(chainsBuilder, "searchchains");
chainsConfig = new ChainsConfig(chainsBuilder);
ProviderConfig.Builder providerBuilder = new ProviderConfig.Builder();
providerBuilder = (ProviderConfig.Builder)root.getConfig(providerBuilder, "searchchains/chain/provider:1/component/com.yahoo.search.federation.vespa.VespaSearcher");
providerConfig = new ProviderConfig(providerBuilder);
ClusterConfig.Builder clusterBuilder = new ClusterConfig.Builder();
clusterBuilder = (ClusterConfig.Builder)root.getConfig(clusterBuilder, "searchchains/chain/cluster2/component/" + ClusterSearcher.class.getName());
clusterConfig = new ClusterConfig(clusterBuilder);
}
@Override
Element servicesXml() {
return parse(
"<searchchains>",
" <searcher id='searcher:1' classId='classId1' />",
" <provider id='provider:1' type='vespa' inherits='parentChain1 parentChain2' excludes='ExcludedSearcher1 ExcludedSearcher2'",
" cacheweight='2.3'>",
" <federationoptions optional='true' timeout='2.3 s' />",
" <nodes>",
" <node host='sourcehost1' port='12'/>",
" <node host='sourcehost2' port='34'/>",
" </nodes>",
" <source id='source:1' inherits='parentChain3 parentChain4' excludes='ExcludedSearcher3 ExcludedSearcher4'>",
" <federationoptions timeout='12 ms' />",
" </source>",
" <source id='source:2' />",
" </provider>",
" <provider id='provider:2' type='local' cluster='cluster1' />",
" <provider id='provider:3' />",
" <provider id='vespa-provider' type='vespa' >",
" <nodes>",
" <node host='localhost' port='" + Defaults.getDefaults().vespaWebServicePort() + "' />",
" </nodes>",
" <config name='search.federation.provider'>",
" <queryType>PROGRAMMATIC</queryType>",
" </config>",
" </provider>",
" <searchchain id='default:99'>",
" <federation id='federation:98' provides='provide_federation' before='p1 p2' after='s1 s2'>",
" <source id='source:1'>",
" <federationoptions optional='false' />",
" </source>",
" </federation>",
" </searchchain>",
" <searchchain id='parentChain1' />",
" <searchchain id='parentChain2' />",
" <searchchain id='parentChain3' />",
" <searchchain id='parentChain4' />",
"</searchchains>");
}
private SearchChains getSearchChains() {
return (SearchChains) root.getChildren().get("searchchains");
}
@Test
public void require_user_config_for_vespa_searcher_works() {
assertEquals(root.getConfig(ProviderConfig.class, "searchchains/chain/vespa-provider/component/com.yahoo.search.federation.vespa.VespaSearcher").
queryType(), ProviderConfig.QueryType.PROGRAMMATIC);
}
@Test
public void require_that_source_chain_spec_id_is_namespaced_in_provider_id() {
Source source = (Source) getSearchChains().allChains().getComponent("source:1@provider:1");
assertThat(source.getChainSpecification().componentId.getNamespace(), is(ComponentId.fromString("provider:1")));
}
@Test
public void validateHttpProviderConfig() {
assertNotNull(providerConfig);
}
@Test
public void validateLocalProviderConfig() {
assertEquals(2, clusterConfig.clusterId());
assertEquals("cluster2", clusterConfig.clusterName());
}
public static boolean verifyChainExists(List<ChainsConfig.Chains> chains, String componentId) {
for (ChainsConfig.Chains c : chains) {
if (c.id().equals(componentId)) return true;
}
return false;
}
}
|
