Search in sources :

Example 6 with CustomTopologyContext

use of com.yahoo.bullet.storm.testing.CustomTopologyContext in project bullet-storm by yahoo.

the class DRPCPubSubTest method setup.

@BeforeMethod
public void setup() throws Exception {
    config = new DRPCConfig("src/test/resources/test_drpc_config.yaml");
    // 1 task for the component named "foo" with task index 0
    CustomTopologyContext context = new CustomTopologyContext(Collections.singletonList(1), "foo", 0);
    config.set(DRPCConfig.STORM_CONTEXT, context);
    Map stormConfig = new Config("src/test/resources/test_storm_config.yaml").getAll(Optional.empty());
    config.set(DRPCConfig.STORM_CONFIG, stormConfig);
}
Also used : Config(com.yahoo.bullet.common.Config) CustomTopologyContext(com.yahoo.bullet.storm.testing.CustomTopologyContext) Map(java.util.Map) BeforeMethod(org.testng.annotations.BeforeMethod)

Aggregations

CustomTopologyContext (com.yahoo.bullet.storm.testing.CustomTopologyContext)6 CustomCollector (com.yahoo.bullet.storm.testing.CustomCollector)4 Map (java.util.Map)3 BeforeMethod (org.testng.annotations.BeforeMethod)3 Test (org.testng.annotations.Test)3 Config (com.yahoo.bullet.common.Config)2 CountDistinct (com.yahoo.bullet.aggregations.CountDistinct)1 CountDistinctTest (com.yahoo.bullet.aggregations.CountDistinctTest)1 Distribution (com.yahoo.bullet.aggregations.Distribution)1 DistributionTest (com.yahoo.bullet.aggregations.DistributionTest)1 TopK (com.yahoo.bullet.aggregations.TopK)1 TopKTest (com.yahoo.bullet.aggregations.TopKTest)1 GroupData (com.yahoo.bullet.aggregations.grouping.GroupData)1 GroupOperation (com.yahoo.bullet.aggregations.grouping.GroupOperation)1 COUNT (com.yahoo.bullet.aggregations.grouping.GroupOperation.GroupOperationType.COUNT)1 COUNT_FIELD (com.yahoo.bullet.aggregations.sketches.QuantileSketch.COUNT_FIELD)1 END_EXCLUSIVE (com.yahoo.bullet.aggregations.sketches.QuantileSketch.END_EXCLUSIVE)1 NEGATIVE_INFINITY_START (com.yahoo.bullet.aggregations.sketches.QuantileSketch.NEGATIVE_INFINITY_START)1 POSITIVE_INFINITY_END (com.yahoo.bullet.aggregations.sketches.QuantileSketch.POSITIVE_INFINITY_END)1 PROBABILITY_FIELD (com.yahoo.bullet.aggregations.sketches.QuantileSketch.PROBABILITY_FIELD)1