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
|
// 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.test;
import com.yahoo.component.ComponentId;
import com.yahoo.component.ComponentSpecification;
import com.yahoo.prelude.IndexFacts;
import com.yahoo.prelude.IndexModel;
import com.yahoo.search.federation.sourceref.SearchChainInvocationSpec;
import com.yahoo.search.federation.sourceref.SearchChainResolver;
import com.yahoo.search.federation.sourceref.SourceRefResolver;
import com.yahoo.search.federation.sourceref.UnresolvedSearchChainException;
import com.yahoo.search.searchchain.model.federation.FederationOptions;
import org.junit.jupiter.api.Test;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Set;
import java.util.TreeMap;
import static com.yahoo.search.federation.sourceref.test.SearchChainResolverTestCase.emptySourceToProviderMap;
import static org.junit.jupiter.api.Assertions.*;
/**
* Test for SourceRefResolver.
*
* @author Tony Vaagenes
*/
public class SourceRefResolverTestCase {
private static final String cluster1 = "cluster1";
private static final String cluster2 = "cluster2";
private static final String cluster3 = "cluster3";
private static IndexFacts indexFacts;
private static final SourceRefResolver sourceRefResolver = createSourceRefResolver();
static {
setupIndexFacts();
}
private static SourceRefResolver createSourceRefResolver() {
SearchChainResolver.Builder builder = new SearchChainResolver.Builder();
builder.addSearchChain(ComponentId.fromString(cluster1), new FederationOptions().setUseByDefault(true),
Collections.emptyList());
builder.addSearchChain(ComponentId.fromString(cluster2), new FederationOptions().setUseByDefault(true),
Collections.emptyList());
return new SourceRefResolver(builder.build());
}
private static void setupIndexFacts() {
TreeMap<String, List<String>> masterClusters = new TreeMap<>();
masterClusters.put(cluster1, List.of("document1", "document2"));
masterClusters.put(cluster2, List.of("document1"));
masterClusters.put(cluster3, List.of("document3"));
indexFacts = new IndexFacts(new IndexModel(masterClusters, Collections.emptyList()));
}
@Test
void check_test_assumptions() {
assertTrue(indexFacts.clustersHavingSearchDefinition("document1").containsAll(List.of("cluster1", "cluster2")));
}
@Test
void lookup_search_chain() throws Exception {
Set<SearchChainInvocationSpec> searchChains = resolve(cluster1);
assertEquals(1, searchChains.size());
assertTrue(searchChainIds(searchChains).contains(cluster1));
}
@Test
void lookup_search_chains_for_document1() throws Exception {
Set<SearchChainInvocationSpec> searchChains = resolve("document1");
assertEquals(2, searchChains.size());
assertTrue(searchChainIds(searchChains).containsAll(List.of(cluster1, cluster2)));
}
@Test
void error_when_document_gives_cluster_without_matching_search_chain() {
try {
resolve("document3");
fail("Expected exception");
} catch (UnresolvedSearchChainException e) {
assertEquals("Failed to resolve cluster search chain 'cluster3' " +
"when using source ref 'document3' as a document name.",
e.getMessage());
}
}
@Test
void error_when_no_document_or_search_chain() {
try {
resolve("document4");
fail("Expected exception");
} catch (UnresolvedSearchChainException e) {
assertEquals("Could not resolve source ref 'document4'.", e.getMessage());
}
}
private List<String> searchChainIds(Set<SearchChainInvocationSpec> searchChains) {
List<String> names = new ArrayList<>();
for (SearchChainInvocationSpec searchChain : searchChains) {
names.add(searchChain.searchChainId.stringValue());
}
return names;
}
private Set<SearchChainInvocationSpec> resolve(String documentName) throws UnresolvedSearchChainException {
return sourceRefResolver.resolve(ComponentSpecification.fromString(documentName), emptySourceToProviderMap(), indexFacts);
}
}
|
