@Test
public void rebalances() throws Exception {

  Set<Integer> assignment = singleton(0);
  Rebalance rebalance = new Rebalance(assignment);
  Disposable subscribe = Flux.from(underTest.rebalances()).limitRequest(1).subscribe();
  underTest.getRebalances().onNext(rebalance);
  subscribe.dispose();
}