Assert.assertTrue(connectionZK.getRecoverableZooKeeper().getState().
equals(States.CLOSED));
// Check that the client recovered
ZooKeeperWatcher newConnectionZK = connection.getZooKeeperWatcher();
States state = newConnectionZK.getRecoverableZooKeeper().getState();
LOG.info("state=" + state);
Assert.assertTrue(state.equals(States.CONNECTED) || state.equals(States.CONNECTING));
}