Cover ZooKeeper mock with tests