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
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.documentapi;
import com.yahoo.document.fieldset.AllFields;
import com.yahoo.documentapi.messagebus.protocol.DocumentProtocol;
import org.junit.Test;
import static org.junit.Assert.*;
public class VisitorParametersTestCase {
private VisitorParameters createVisitorParameters() {
VisitorParameters params = new VisitorParameters("");
params.setDocumentSelection("id.user==5678");
params.setBucketSpace("narnia");
params.setFromTimestamp(9001);
params.setToTimestamp(10001);
params.setVisitorLibrary("CoolVisitor");
params.setLibraryParameter("groovy", "dudes");
params.setLibraryParameter("ninja", "turtles");
params.setMaxBucketsPerVisitor(55);
params.setPriority(DocumentProtocol.Priority.HIGHEST);
params.setRoute("extraterrestrial/highway");
params.setTimeoutMs(1337);
params.setMaxPending(111);
params.setFieldSet(AllFields.NAME);
params.setVisitRemoves(true);
params.setVisitInconsistentBuckets(true);
params.setTraceLevel(9);
params.setResumeFileName("foo.txt");
params.setResumeToken(new ProgressToken());
params.setRemoteDataHandler("mars_rover");
params.setControlHandler(new VisitorControlHandler());
params.setMaxTotalHits(777);
params.skipBucketsOnFatalErrors(true);
return params;
}
@Test
public void testCopyConstructor() {
VisitorParameters params = createVisitorParameters();
VisitorParameters copy = new VisitorParameters(params);
assertEquals("id.user==5678", copy.getDocumentSelection());
assertEquals(9001, copy.getFromTimestamp());
assertEquals(10001, copy.getToTimestamp());
assertEquals("CoolVisitor", copy.getVisitorLibrary());
assertEquals(2, copy.getLibraryParameters().size());
assertEquals("dudes", new String(copy.getLibraryParameters().get("groovy")));
assertEquals("turtles", new String(copy.getLibraryParameters().get("ninja")));
assertEquals(55, copy.getMaxBucketsPerVisitor());
assertEquals(DocumentProtocol.Priority.HIGHEST, copy.getPriority());
assertEquals("extraterrestrial/highway", copy.getRoute().toString());
assertEquals(1337, copy.getTimeoutMs());
assertEquals(111, copy.getMaxPending());
assertEquals(AllFields.NAME, copy.getFieldSet());
assertEquals(true, copy.getVisitRemoves());
assertEquals(true, copy.getVisitInconsistentBuckets());
assertEquals(9, copy.getTraceLevel());
assertEquals("foo.txt", copy.getResumeFileName());
assertEquals(params.getResumeToken(), copy.getResumeToken()); // instance compare
assertEquals("mars_rover", copy.getRemoteDataHandler());
assertEquals(params.getControlHandler(), copy.getControlHandler());
assertEquals(777, copy.getMaxTotalHits());
assertEquals(true, copy.skipBucketsOnFatalErrors());
// Test local data handler copy
VisitorParameters params2 = new VisitorParameters("");
params2.setLocalDataHandler(new SimpleVisitorDocumentQueue());
VisitorParameters copy2 = new VisitorParameters(params2);
assertEquals(params2.getLocalDataHandler(), copy2.getLocalDataHandler()); // instance compare
}
@Test
public void testToString() {
VisitorParameters params = createVisitorParameters();
assertEquals(
"VisitorParameters(\n" +
" Document selection: id.user==5678\n" +
" Bucket space: narnia\n" +
" Visitor library: CoolVisitor\n" +
" Max pending: 111\n" +
" Timeout (ms): 1337\n" +
" Time period: 9001 - 10001\n" +
" Visiting remove entries\n" +
" Visiting inconsistent buckets\n" +
" Visitor library parameters:\n" +
" groovy : dudes\n" +
" ninja : turtles\n" +
" Field set: [all]\n" +
" Route: extraterrestrial/highway\n" +
" Weight: 1.0\n" +
" Max total hits: 777\n" +
" Max buckets: 55\n" +
" Priority: HIGHEST\n" +
")",
params.toString());
}
}
|
