/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF 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 com.jstorm.example.unittests.trident;
import backtype.storm.Config;
import backtype.storm.tuple.Fields;
import backtype.storm.tuple.Values;
import com.jstorm.example.unittests.utils.JStormUnitTestRunner;
import org.junit.Test;
import storm.trident.Stream;
import storm.trident.TridentTopology;
import storm.trident.operation.builtin.Debug;
import storm.trident.tuple.TridentTuple;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import static org.junit.Assert.*;
/**
* @author binyang.dby on 2016/7/9.
*
* This is the unit test version of TridentMinMaxOfDevicesTopology.
* The spout generate 2
*/
public class TridentMinMaxOfDevicesTest
{
public final static int SPOUT_BATCH_SIZE = 10;
@Test
public void testTridentMinMaxOfDevices()
{
Fields fields = new Fields("device-id", "count");
List<Values> content = new ArrayList<Values>();
for(int i=0; i<SPOUT_BATCH_SIZE; i++)
content.add(new Values(i+1));
ShuffleValuesBatchSpout spout = new ShuffleValuesBatchSpout(fields, content, content);
TridentTopology tridentTopology = new TridentTopology();
Stream stream = tridentTopology.newStream("device-gen-spout", spout)
.each(fields, new Debug("#### devices"));
stream.minBy("device-id").each(fields, new AssertMinDebug());
stream.maxBy("count").each(fields, new AssertMaxDebug());
Map config = new HashMap();
config.put(Config.TOPOLOGY_NAME, "TridentMinMaxOfDevicesTest");
//the test can pass if the 2 AssertDebug pass throughout the test
JStormUnitTestRunner.submitTopology(tridentTopology.build(), config, 120, null);
}
private static class AssertMinDebug extends Debug
{
@Override
public boolean isKeep(TridentTuple tuple) {
assertEquals(1, tuple.getInteger(0).intValue());
return true;
}
}
private static class AssertMaxDebug extends Debug
{
@Override
public boolean isKeep(TridentTuple tuple) {
assertEquals(SPOUT_BATCH_SIZE, tuple.getInteger(1).intValue());
return true;
}
}
}