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
|
// 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.search.searchchain.model.federation.FederationOptions;
import org.junit.jupiter.api.Test;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import static com.yahoo.search.federation.sourceref.SearchChainResolverTestCase.emptySourceToProviderMap;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertTrue;
/**
* 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 final String schema1 = "document1";
private static final String schema2 = "document2";
private static final String schema3 = "document3";
private static final SourceRefResolver sourceRefResolver = createSourceRefResolver();
private static SourceRefResolver createSourceRefResolver() {
SearchChainResolver.Builder builder = new SearchChainResolver.Builder();
builder.addSearchChain(ComponentId.fromString(cluster1), new FederationOptions().setUseByDefault(true), List.of());
builder.addSearchChain(ComponentId.fromString(cluster2), new FederationOptions().setUseByDefault(true), List.of());
return new SourceRefResolver(builder.build(), Map.of(schema1, List.of(cluster1, cluster2),
schema2, List.of(cluster1),
schema3, List.of(cluster3)));
}
@Test
void lookup_search_chain() throws Exception {
List<ResolveResult> searchChains = resolve(cluster1);
assertEquals(1, searchChains.size());
assertTrue(searchChainIds(searchChains).contains(cluster1));
}
@Test
void lookup_search_chains_for_document1() throws Exception {
List<ResolveResult> 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() {
List<ResolveResult> result = resolve("document3");
assertEquals("Failed to resolve cluster search chain 'cluster3' " +
"when using source ref 'document3' as a document name.",
result.get(0).errorMsg());
}
@Test
void error_when_no_document_or_search_chain() {
List<ResolveResult> results = resolve("document4");
assertEquals("Could not resolve source ref 'document4'.", results.get(0).errorMsg());
}
private List<String> searchChainIds(Collection<ResolveResult> searchChains) {
return searchChains.stream().map(r -> r.invocationSpec().searchChainId.stringValue()).toList();
}
private List<ResolveResult> resolve(String documentName) {
return sourceRefResolver.resolve(ComponentSpecification.fromString(documentName), emptySourceToProviderMap());
}
}
|
