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
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.search.grouping.result;
import com.yahoo.search.grouping.Continuation;
import com.yahoo.search.result.HitGroup;
import com.yahoo.search.result.Relevance;
import com.yahoo.text.Utf8;
import com.yahoo.text.XMLWriter;
import org.junit.jupiter.api.Test;
import java.io.IOException;
import java.io.StringWriter;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.fail;
/**
* @author Simon Thoresen Hult
*/
public class HitRendererTestCase {
@Test
void requireThatGroupListsRenderAsExpected() {
assertRender(new GroupList("foo"), "<grouplist label=\"foo\"></grouplist>\n");
assertRender(new GroupList("b\u00e6z"), "<grouplist label=\"b\u00e6z\"></grouplist>\n");
GroupList lst = new GroupList("foo");
lst.continuations().put("bar.key", new MyContinuation("bar.val"));
lst.continuations().put("baz.key", new MyContinuation("baz.val"));
assertRender(lst, "<grouplist label=\"foo\">\n" +
"<continuation id=\"bar.key\">bar.val</continuation>\n" +
"<continuation id=\"baz.key\">baz.val</continuation>\n" +
"</grouplist>\n");
}
@Test
void requireThatGroupIdsRenderAsExpected() {
assertRender(newGroup(new DoubleId(6.9)),
"<group relevance=\"1.0\">\n" +
"<id type=\"double\">6.9</id>\n" +
"</group>\n");
assertRender(newGroup(new LongId(69L)),
"<group relevance=\"1.0\">\n" +
"<id type=\"long\">69</id>\n" +
"</group>\n");
assertRender(newGroup(new BoolId(true)),
"<group relevance=\"1.0\">\n" +
"<id type=\"bool\">true</id>\n" +
"</group>\n");
assertRender(newGroup(new BoolId(false)),
"<group relevance=\"1.0\">\n" +
"<id type=\"bool\">false</id>\n" +
"</group>\n");
assertRender(newGroup(new NullId()),
"<group relevance=\"1.0\">\n" +
"<id type=\"null\"/>\n" +
"</group>\n");
assertRender(newGroup(new RawId(Utf8.toBytes("foo"))),
"<group relevance=\"1.0\">\n" +
"<id type=\"raw\">Zm9v</id>\n" +
"</group>\n");
assertRender(newGroup(new StringId("foo")),
"<group relevance=\"1.0\">\n" +
"<id type=\"string\">foo</id>\n" +
"</group>\n");
assertRender(newGroup(new StringId("b\u00e6z")),
"<group relevance=\"1.0\">\n" +
"<id type=\"string\">b\u00e6z</id>\n" +
"</group>\n");
assertRender(newGroup(new DoubleBucketId(6.9, 9.6)),
"<group relevance=\"1.0\">\n" +
"<id type=\"double_bucket\">\n<from>6.9</from>\n<to>9.6</to>\n</id>\n" +
"</group>\n");
assertRender(newGroup(new LongBucketId(6L, 9L)),
"<group relevance=\"1.0\">\n" +
"<id type=\"long_bucket\">\n<from>6</from>\n<to>9</to>\n</id>\n" +
"</group>\n");
assertRender(newGroup(new StringBucketId("bar", "baz")),
"<group relevance=\"1.0\">\n" +
"<id type=\"string_bucket\">\n<from>bar</from>\n<to>baz</to>\n</id>\n" +
"</group>\n");
assertRender(newGroup(new StringBucketId("b\u00e6r", "b\u00e6z")),
"<group relevance=\"1.0\">\n" +
"<id type=\"string_bucket\">\n<from>b\u00e6r</from>\n<to>b\u00e6z</to>\n</id>\n" +
"</group>\n");
assertRender(newGroup(new RawBucketId(Utf8.toBytes("bar"), Utf8.toBytes("baz"))),
"<group relevance=\"1.0\">\n" +
"<id type=\"raw_bucket\">\n<from>YmFy</from>\n<to>YmF6</to>\n</id>\n" +
"</group>\n");
}
@Test
void requireThatGroupsRenderAsExpected() {
Group group = newGroup(new StringId("foo"));
group.setField("foo", "bar");
group.setField("baz", "cox");
assertRender(group, "<group relevance=\"1.0\">\n" +
"<id type=\"string\">foo</id>\n" +
"<output label=\"foo\">bar</output>\n" +
"<output label=\"baz\">cox</output>\n" +
"</group>\n");
group = newGroup(new StringId("foo"));
group.setField("foo", "b\u00e6r");
group.setField("b\u00e5z", "cox");
assertRender(group, "<group relevance=\"1.0\">\n" +
"<id type=\"string\">foo</id>\n" +
"<output label=\"foo\">b\u00e6r</output>\n" +
"<output label=\"b\u00e5z\">cox</output>\n" +
"</group>\n");
}
@Test
void requireThatRootGroupsRenderAsExpected() {
RootGroup group = new RootGroup(0, new MyContinuation("69"));
group.setField("foo", "bar");
group.setField("baz", "cox");
assertRender(group, "<group relevance=\"1.0\">\n" +
"<id type=\"root\"/>\n" +
"<continuation id=\"this\">69</continuation>\n" +
"<output label=\"foo\">bar</output>\n" +
"<output label=\"baz\">cox</output>\n" +
"</group>\n");
group = new RootGroup(0, new MyContinuation("96"));
group.setField("foo", "b\u00e6r");
group.setField("b\u00e5z", "cox");
assertRender(group, "<group relevance=\"1.0\">\n" +
"<id type=\"root\"/>\n" +
"<continuation id=\"this\">96</continuation>\n" +
"<output label=\"foo\">b\u00e6r</output>\n" +
"<output label=\"b\u00e5z\">cox</output>\n" +
"</group>\n");
}
@Test
void requireThatHitListsRenderAsExpected() {
assertRender(new HitList("foo"), "<hitlist label=\"foo\"></hitlist>\n");
assertRender(new HitList("b\u00e6z"), "<hitlist label=\"b\u00e6z\"></hitlist>\n");
HitList lst = new HitList("foo");
lst.continuations().put("bar.key", new MyContinuation("bar.val"));
lst.continuations().put("baz.key", new MyContinuation("baz.val"));
assertRender(lst, "<hitlist label=\"foo\">\n" +
"<continuation id=\"bar.key\">bar.val</continuation>\n" +
"<continuation id=\"baz.key\">baz.val</continuation>\n" +
"</hitlist>\n");
}
private static Group newGroup(GroupId id) {
return new Group(id, new Relevance(1));
}
private static void assertRender(HitGroup hit, String expectedXml) {
StringWriter str = new StringWriter();
XMLWriter out = new XMLWriter(str, 0, -1);
try {
HitRenderer.renderHeader(hit, out);
while (out.openTags().size() > 0) {
out.closeTag();
}
} catch (IOException e) {
fail();
}
assertEquals(expectedXml, str.toString());
}
private static class MyContinuation extends Continuation {
final String str;
MyContinuation(String str) {
this.str = str;
}
@Override
public Continuation copy() {
return null;
}
@Override
public String toString() {
return str;
}
}
}
|
