/** * Creates a new {@link StreamBuilder} */ public StreamBuilder() { graph = new DefaultDirectedGraph<>(new StreamsEdgeFactory()); }
@Before public void setUp() throws Exception { mockTopologyContext = Mockito.mock(TopologyContext.class); mockOutputCollector = Mockito.mock(OutputCollector.class); mockTuple1 = Mockito.mock(Tuple.class); mockTuple2 = Mockito.mock(Tuple.class); mockTuple3 = Mockito.mock(Tuple.class); setUpMockTuples(mockTuple1, mockTuple2, mockTuple3); punctuation = Mockito.mock(Tuple.class); setUpPunctuation(punctuation); mockStreamToProcessors = Mockito.mock(Multimap.class); graph = new DefaultDirectedGraph(new StreamsEdgeFactory()); }
private void setUpWindowedProcessorBolt(Processor<?> processor, Window<?, ?> window) { ProcessorNode node = new ProcessorNode(processor, "outputstream", new Fields("value")); node.setWindowed(true); Mockito.when(mockStreamToProcessors.get(Mockito.anyString())).thenReturn(Collections.singletonList(node)); Mockito.when(mockStreamToProcessors.keySet()).thenReturn(Collections.singleton("inputstream")); graph = new DefaultDirectedGraph<>(new StreamsEdgeFactory()); graph.addVertex(node); bolt = new WindowedProcessorBolt("bolt1", graph, Collections.singletonList(node), window); bolt.setStreamToInitialProcessors(mockStreamToProcessors); bolt.prepare(new HashMap<>(), mockTopologyContext, mockOutputCollector); }
private void setUpStatefulProcessorBolt(Processor<?> processor) { ProcessorNode node = new ProcessorNode(processor, "outputstream", new Fields("value")); node.setEmitsPair(true); Mockito.when(mockStreamToProcessors.get(Mockito.anyString())).thenReturn(Collections.singletonList(node)); graph = new DefaultDirectedGraph(new StreamsEdgeFactory()); graph.addVertex(node); bolt = new StatefulProcessorBolt<>("bolt1", graph, Collections.singletonList(node)); bolt.setStreamToInitialProcessors(mockStreamToProcessors); bolt.prepare(new HashMap<>(), mockTopologyContext, mockOutputCollector); bolt.initState(mockKeyValueState); }