/
AbstractStormSpoutTests.java
88 lines (73 loc) · 3.28 KB
/
AbstractStormSpoutTests.java
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
/*
* Licensed to Elasticsearch under one or more contributor
* license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright
* ownership. Elasticsearch licenses this file to you under
* the Apache License, Version 2.0 (the "License"); you may
* not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.elasticsearch.integration.storm;
import java.io.IOException;
import java.util.Arrays;
import java.util.Collection;
import java.util.LinkedHashMap;
import java.util.Map;
import org.elasticsearch.hadoop.util.TestSettings;
import org.elasticsearch.storm.cfg.StormConfigurationOptions;
import org.junit.After;
import org.junit.Before;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;
import org.junit.runners.Parameterized.Parameters;
import com.google.common.collect.ImmutableMap;
import static org.elasticsearch.integration.storm.AbstractStormSuite.COMPONENT_HAS_COMPLETED;
@RunWith(Parameterized.class)
public abstract class AbstractStormSpoutTests {
protected Map conf;
protected String index;
public AbstractStormSpoutTests(Map conf, String index) {
this.conf = conf;
this.index = index;
new TestSettings();
conf.putAll(TestSettings.TESTING_PROPS);
}
@Before
public void setup() {
// -1 bolt, -1 test
COMPONENT_HAS_COMPLETED = new Counter(2);
CapturingBolt.CAPTURED.clear();
}
@After
public void destroy() {
COMPONENT_HAS_COMPLETED.decrement();
}
@Parameters
public static Collection<Object[]> configs() throws IOException {
// no ack
Map noAck = new LinkedHashMap(ImmutableMap.of(StormConfigurationOptions.ES_STORM_SPOUT_RELIABLE, Boolean.FALSE.toString()));
// read ack
Map ack = new LinkedHashMap(ImmutableMap.of(StormConfigurationOptions.ES_STORM_SPOUT_RELIABLE, Boolean.TRUE.toString()));
// read ack bounded queue
Map ackWithSize = new LinkedHashMap(ImmutableMap.of(StormConfigurationOptions.ES_STORM_SPOUT_RELIABLE, Boolean.TRUE.toString(), StormConfigurationOptions.ES_STORM_SPOUT_RELIABLE_QUEUE_SIZE, "1"));
// read ack bounded queue with no retries
Map ackWithSizeNoRetries = new LinkedHashMap(ImmutableMap.of(StormConfigurationOptions.ES_STORM_SPOUT_RELIABLE, Boolean.TRUE.toString(),
StormConfigurationOptions.ES_STORM_SPOUT_RELIABLE_QUEUE_SIZE, "1",
StormConfigurationOptions.ES_STORM_SPOUT_RELIABLE_RETRIES_PER_TUPLE, "1",
StormConfigurationOptions.ES_STORM_SPOUT_FIELDS, "message"));
return Arrays.asList(new Object[][] {
{ noAck, "storm-spout" },
{ ack, "storm-spout-reliable" },
//{ ackWithSize, "storm-spout-reliable-size" },
{ ackWithSizeNoRetries, "storm-spout-reliable-size-no-retries" } });
}
}