Github user srdo commented on a diff in the pull request:
https://github.com/apache/storm/pull/2334#discussion_r140026143
--- Diff: storm-server/src/test/java/org/apache/storm/TickTupleTest.java ---
@@ -18,85 +18,103 @@
package org.apache.storm;
+import java.util.Map;
+import java.util.concurrent.LinkedBlockingQueue;
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.atomic.AtomicReference;
+import org.apache.storm.ILocalCluster.ILocalTopology;
import org.apache.storm.generated.StormTopology;
import org.apache.storm.spout.SpoutOutputCollector;
import org.apache.storm.task.OutputCollector;
import org.apache.storm.task.TopologyContext;
-import org.apache.storm.topology.IRichSpout;
import org.apache.storm.topology.OutputFieldsDeclarer;
import org.apache.storm.topology.TopologyBuilder;
import org.apache.storm.topology.base.BaseRichBolt;
import org.apache.storm.topology.base.BaseRichSpout;
import org.apache.storm.tuple.Fields;
import org.apache.storm.tuple.Tuple;
-import org.apache.storm.utils.Utils;
-import org.junit.Assert;
+import org.apache.storm.utils.Time;
+import org.apache.storm.utils.TupleUtils;
import org.junit.Test;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
-import java.util.Map;
+import static org.junit.Assert.*;
public class TickTupleTest {
+ private final static Logger LOG =
LoggerFactory.getLogger(TickTupleTest.class);
+ private static LinkedBlockingQueue<Long> tickTupleTimes = new
LinkedBlockingQueue<>();
+ private static AtomicReference<Tuple> nonTickTuple = new
AtomicReference<>(null);
@Test
public void testTickTupleWorksWithSystemBolt() throws Exception {
- ILocalCluster cluster = null;
- try {
- cluster = new
LocalCluster.Builder().withSimulatedTime().withNimbusDaemon(true).build();
+ try (ILocalCluster cluster = new
LocalCluster.Builder().withSimulatedTime().withNimbusDaemon(true).build()){
StormTopology topology = createNoOpTopology();
Config topoConf = new Config();
- topoConf.putAll(Utils.readDefaultConfig());
- topoConf.put("storm.cluster.mode", "local");
topoConf.put(Config.TOPOLOGY_TICK_TUPLE_FREQ_SECS, 1);
- cluster.submitTopology("test", topoConf, topology);
- cluster.advanceClusterTime(2);
- Assert.assertTrue("Test is passed", true);
- } finally {
- cluster.close();
+ try (ILocalTopology topo = cluster.submitTopology("test",
topoConf, topology)) {
+ //Give the cluster some time to come up
+ cluster.advanceClusterTime(40);
+ tickTupleTimes.clear();
+ cluster.advanceClusterTime(1);
+ assertEquals(41000, tickTupleTimes.poll(100,
TimeUnit.MILLISECONDS).longValue());
+ cluster.advanceClusterTime(1);
+ assertEquals(42000, tickTupleTimes.poll(100,
TimeUnit.MILLISECONDS).longValue());
+ cluster.advanceClusterTime(1);
+ assertEquals(43000, tickTupleTimes.poll(100,
TimeUnit.MILLISECONDS).longValue());
+ cluster.advanceClusterTime(1);
+ assertEquals(44000, tickTupleTimes.poll(100,
TimeUnit.MILLISECONDS).longValue());
+ cluster.advanceClusterTime(1);
+ assertEquals(45000, tickTupleTimes.poll(100,
TimeUnit.MILLISECONDS).longValue());
+ }
+ assertNull("The bolt got a tuple that is not a tick tuple " +
nonTickTuple.get(), nonTickTuple.get());
}
-
}
- private IRichSpout makeNoOpSpout() {
- return new BaseRichSpout() {
- @Override
- public void declareOutputFields(OutputFieldsDeclarer declarer)
{
- declarer.declare(new Fields("tuple"));
- }
-
- @Override
- public void open(Map<String, Object> conf, TopologyContext
context, SpoutOutputCollector collector) {
- }
+ private static class NoopSpout extends BaseRichSpout {
+ @Override
+ public void declareOutputFields(OutputFieldsDeclarer declarer) {
+ declarer.declare(new Fields("tuple"));
+ }
- @Override
- public void nextTuple() {
- }
+ @Override
+ public void open(Map<String, Object> conf, TopologyContext
context, SpoutOutputCollector collector) {
+ }
- private void writeObject(java.io.ObjectOutputStream stream) {
- }
- };
+ @Override
+ public void nextTuple() {
+ }
}
- private BaseRichBolt makeNoOpBolt() {
- return new BaseRichBolt() {
- @Override
- public void prepare(Map<String, Object> conf, TopologyContext
topologyContext, OutputCollector outputCollector) {}
- @Override
- public void execute(Tuple tuple) {}
+ private static class NoopBlot extends BaseRichBolt {
--- End diff --
Nit: Blot -> Bolt
---
