@Test
public void testBandwidthOnIncresedPartitions()
{

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