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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.search.federation.sourceref;
import com.yahoo.component.ComponentId;
import com.yahoo.component.ComponentSpecification;
import com.yahoo.processing.request.properties.PropertyMap;
import com.yahoo.processing.request.Properties;
import com.yahoo.search.searchchain.model.federation.FederationOptions;
import org.junit.jupiter.api.Test;
import java.util.Collections;
import java.util.Iterator;
import java.util.SortedSet;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertNull;
/**
* @author Tony Vaagenes
*/
public class SearchChainResolverTestCase {
private static final FederationOptions federationOptions =
new FederationOptions().setTimeoutInMilliseconds(3000).setOptional(true);
private static final ComponentId searchChainId = ComponentId.fromString("search-chain");
private static final ComponentId providerId = ComponentId.fromString("provider");
private static final ComponentId provider2Id = ComponentId.fromString("provider2");
private static final ComponentId sourceId = ComponentId.fromString("source");
private static final ComponentId sourceChainInProviderId =
ComponentId.fromString("source-chain").nestInNamespace(providerId);
private static final ComponentId sourceChainInProvider2Id =
ComponentId.fromString("source-chain").nestInNamespace(provider2Id);
private static final SearchChainResolver searchChainResolver;
static {
SearchChainResolver.Builder builder = new SearchChainResolver.Builder();
builder.addSearchChain(searchChainId, federationOptions.setUseByDefault(true), Collections.<String>emptyList());
builder.addSearchChain(providerId, federationOptions.setUseByDefault(false), Collections.<String>emptyList());
builder.addSourceForProvider(sourceId, providerId, sourceChainInProviderId, true,
federationOptions.setUseByDefault(true), Collections.<String>emptyList());
builder.addSourceForProvider(sourceId, provider2Id, sourceChainInProvider2Id, false,
federationOptions.setUseByDefault(false), Collections.<String>emptyList());
searchChainResolver = builder.build();
}
@Test
void check_default_search_chains() {
assertEquals(2, searchChainResolver.defaultTargets().size());
Iterator<Target> iterator = searchChainResolver.defaultTargets().iterator();
assertEquals(searchChainId.toString(), iterator.next().searchRefDescription());
assertEquals(sourceChainInProviderId.toString(), iterator.next().searchRefDescription());
}
@Test
void require_error_message_for_invalid_source() {
var result = resolve("no-such-source");
assertEquals("Could not resolve source ref 'no-such-source'.", result.errorMsg());
}
@Test
void lookup_search_chain() {
SearchChainInvocationSpec res = resolve(searchChainId.getName()).invocationSpec();
assertEquals(searchChainId, res.searchChainId);
}
//TODO: TVT: @Test()
public void lookup_provider() {
SearchChainInvocationSpec res = resolve(providerId.getName()).invocationSpec();
assertEquals(providerId, res.provider);
assertNull(res.source);
assertEquals(providerId, res.searchChainId);
}
@Test
void lookup_source() {
SearchChainInvocationSpec res = resolve(sourceId.getName()).invocationSpec();
assertIsSourceInProvider(res);
}
@Test
void lookup_source_search_chain_directly() {
SearchChainInvocationSpec res = resolve(sourceChainInProviderId.stringValue()).invocationSpec();
assertIsSourceInProvider(res);
}
private void assertIsSourceInProvider(SearchChainInvocationSpec res) {
assertEquals(providerId, res.provider);
assertEquals(sourceId, res.source);
assertEquals(sourceChainInProviderId, res.searchChainId);
}
@Test
void lookup_source_for_provider2() {
SearchChainInvocationSpec res = resolve(sourceId.getName(), provider2Id.getName()).invocationSpec();
assertEquals(provider2Id, res.provider);
assertEquals(sourceId, res.source);
assertEquals(sourceChainInProvider2Id, res.searchChainId);
}
@Test
void lists_source_ref_description_for_top_level_targets() {
SortedSet<Target> topLevelTargets = searchChainResolver.allTopLevelTargets();
assertEquals(3, topLevelTargets.size());
Iterator<Target> i = topLevelTargets.iterator();
assertSearchRefDescriptionIs(i.next(), providerId.toString());
assertSearchRefDescriptionIs(i.next(), searchChainId.toString());
assertSearchRefDescriptionIs(i.next(), "source[provider = provider, provider2]");
}
private void assertSearchRefDescriptionIs(Target target, String expected) {
assertEquals(expected, target.searchRefDescription());
}
static Properties emptySourceToProviderMap() {
return new PropertyMap();
}
private ResolveResult resolve(String sourceSpecification) {
return resolve(sourceSpecification, emptySourceToProviderMap());
}
private ResolveResult resolve(String sourceSpecification, String providerSpecification) {
Properties sourceToProviderMap = emptySourceToProviderMap();
sourceToProviderMap.set("source." + sourceSpecification + ".provider", providerSpecification);
return resolve(sourceSpecification, sourceToProviderMap);
}
private ResolveResult resolve(String sourceSpecification, Properties sourceToProviderMap) {
ResolveResult res = searchChainResolver.resolve(
ComponentSpecification.fromString(sourceSpecification), sourceToProviderMap);
if (res.invocationSpec() != null) {
assertEquals(federationOptions, res.invocationSpec().federationOptions);
} else {
assertNotNull(res.errorMsg());
}
return res;
}
}
|
