@Test
public void testBandwidthOnPartitions()
{

  when(partitionContextMock.getParallelPartitionCount()).thenReturn(0); // no partitions
  Collection<Partition<BandwidthLimitingOperator>> partitions = Lists.newArrayList();
  DefaultPartition<BandwidthLimitingOperator> defaultPartition = new DefaultPartition<BandwidthLimitingOperator>(operatorMock);
  partitions.add(defaultPartition);
  underTest.definePartitions(partitions, partitionContextMock);
  verify(bandwidthManagerMock).setBandwidth(10L);
}