Search in sources :

Example 1 with RawScheme

use of org.apache.storm.spout.RawScheme in project kafka-spout by HolmesNL.

the class KafkaSpoutBufferBehaviourTest method setup.

@Before
public void setup() {
    final Scheme scheme = spy(new RawScheme());
    // main test subject
    _subject = new KafkaSpout(scheme);
    // assign the topic to be used for stream retrieval
    _subject._topic = "test-topic";
    // use a buffer size higher than the expected amount of messages available
    _subject._bufSize = 4;
    // assign the default FailHandler
    _subject._failHandler = ConfigUtils.DEFAULT_FAIL_HANDLER;
    // mocked consumer to avoid actually contacting zookeeper
    _consumer = mock(ConsumerConnector.class);
    // ... but make sure it will return a valid (empty) stream
    when(_consumer.createMessageStreams(any(Map.class))).thenReturn(EMPTY_STREAM);
    // assign the consumer to the test subject
    _subject._consumer = _consumer;
    // provide a mocked collector to be able to check for emitted values
    _subject._collector = mock(SpoutOutputCollector.class);
}
Also used : RawScheme(org.apache.storm.spout.RawScheme) Scheme(org.apache.storm.spout.Scheme) RawScheme(org.apache.storm.spout.RawScheme) SpoutOutputCollector(org.apache.storm.spout.SpoutOutputCollector) ConsumerConnector(kafka.javaapi.consumer.ConsumerConnector) HashMap(java.util.HashMap) Map(java.util.Map) SortedMap(java.util.SortedMap) Before(org.junit.Before)

Aggregations

HashMap (java.util.HashMap)1 Map (java.util.Map)1 SortedMap (java.util.SortedMap)1 ConsumerConnector (kafka.javaapi.consumer.ConsumerConnector)1 RawScheme (org.apache.storm.spout.RawScheme)1 Scheme (org.apache.storm.spout.Scheme)1 SpoutOutputCollector (org.apache.storm.spout.SpoutOutputCollector)1 Before (org.junit.Before)1