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
|
// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.search.dispatch;
import com.yahoo.compress.CompressionType;
import com.yahoo.prelude.fastsearch.FastHit;
import com.yahoo.search.Query;
import com.yahoo.search.Result;
import org.junit.Test;
import static org.junit.Assert.assertEquals;
import java.util.HashMap;
import java.util.Map;
/**
* Tests using a dispatcher to fill a result
*
* @author bratseth
*/
public class FillTestCase {
private MockClient client = new MockClient();
@Test
public void testFilling() {
Map<Integer, Client.NodeConnection> nodes = new HashMap<>();
nodes.put(0, client.createConnection("host0", 123));
nodes.put(1, client.createConnection("host1", 123));
nodes.put(2, client.createConnection("host2", 123));
Dispatcher dispatcher = new Dispatcher(nodes, client);
Query query = new Query();
Result result = new Result(query);
result.hits().add(createHit(0, 0));
result.hits().add(createHit(2, 1));
result.hits().add(createHit(1, 2));
result.hits().add(createHit(2, 3));
result.hits().add(createHit(0, 4));
client.setDocsumReponse("host0", 0, "summaryClass1", map("field1", "s.0.0", "field2", 0));
client.setDocsumReponse("host2", 1, "summaryClass1", map("field1", "s.2.1", "field2", 1));
client.setDocsumReponse("host1", 2, "summaryClass1", map("field1", "s.1.2", "field2", 2));
client.setDocsumReponse("host2", 3, "summaryClass1", map("field1", "s.2.3", "field2", 3));
client.setDocsumReponse("host0", 4, "summaryClass1", map("field1", "s.0.4", "field2", 4));
dispatcher.fill(result, "summaryClass1", CompressionType.valueOf("LZ4"));
assertEquals("s.0.0", result.hits().get("hit:0").getField("field1").toString());
assertEquals("s.2.1", result.hits().get("hit:1").getField("field1").toString());
assertEquals("s.1.2", result.hits().get("hit:2").getField("field1").toString());
assertEquals("s.2.3", result.hits().get("hit:3").getField("field1").toString());
assertEquals("s.0.4", result.hits().get("hit:4").getField("field1").toString());
assertEquals(0L, result.hits().get("hit:0").getField("field2"));
assertEquals(1L, result.hits().get("hit:1").getField("field2"));
assertEquals(2L, result.hits().get("hit:2").getField("field2"));
assertEquals(3L, result.hits().get("hit:3").getField("field2"));
assertEquals(4L, result.hits().get("hit:4").getField("field2"));
}
@Test
public void testErrorHandling() {
client.setMalfunctioning(true);
Map<Integer, Client.NodeConnection> nodes = new HashMap<>();
nodes.put(0, client.createConnection("host0", 123));
Dispatcher dispatcher = new Dispatcher(nodes, client);
Query query = new Query();
Result result = new Result(query);
result.hits().add(createHit(0, 0));
dispatcher.fill(result, "summaryClass1", CompressionType.valueOf("LZ4"));
assertEquals("Malfunctioning", result.hits().getError().getDetailedMessage());
}
private FastHit createHit(int sourceNodeId, int hitId) {
FastHit hit = new FastHit("hit:" + hitId, 1.0);
hit.setPartId(sourceNodeId, 0);
hit.setDistributionKey(sourceNodeId);
hit.setGlobalId(client.globalIdFrom(hitId));
return hit;
}
private Map<String, Object> map(String stringKey, String stringValue, String intKey, int intValue) {
Map<String, Object> map = new HashMap<>();
map.put(stringKey, stringValue);
map.put(intKey, intValue);
return map;
}
}
|