public Map<SystemStreamPartition, List<IncomingMessageEnvelope>> poll(Set<SystemStreamPartition> systemStreamPartitions, long timeout) throws InterruptedException {
Map<SystemStreamPartition, List<IncomingMessageEnvelope>> map = new LinkedHashMap<SystemStreamPartition, List<IncomingMessageEnvelope>>();
assertEquals(1, systemStreamPartitions.size());
SystemStreamPartition systemStreamPartition = systemStreamPartitions.iterator().next();
assertEquals(expectedSystemStreamPartition, systemStreamPartition);
if (pollCount++ == 0) {
List<IncomingMessageEnvelope> list = new ArrayList<IncomingMessageEnvelope>();
SetConfig setConfig1 = new SetConfig("test", "job.name", "my-job-name");
SetConfig setConfig2 = new SetConfig("test", "job.id", "1234");
Delete delete = new Delete("test", "job.name", SetConfig.TYPE);
list.add(new IncomingMessageEnvelope(systemStreamPartition, null, serialize(setConfig1.getKeyArray()), serialize(setConfig1.getMessageMap())));
list.add(new IncomingMessageEnvelope(systemStreamPartition, null, serialize(setConfig2.getKeyArray()), serialize(setConfig2.getMessageMap())));
list.add(new IncomingMessageEnvelope(systemStreamPartition, null, serialize(delete.getKeyArray()), delete.getMessageMap()));
map.put(systemStreamPartition, list);
}
return map;
}