Connection hangs forever when using connection context manager and there was a disconnect before __exit method is called

[zem00n]

I am using this library in a method and this method never returns if server/connection died for any reason. Currently I am not using Message Listener reconnect method(on_disconnect calls connect) as I didn't find any examples on how to gracefully disconnect without this reconnection.

stomp.connect.BaseConnection.exit

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.disconnect(self.disconnect_receipt_id)
        self.disconnect_listener.wait_on_receipt()
        self.disconnect_listener.wait_on_disconnected()

Should we check if connection is alive? And probably we don't want to wait forever for a disconnect.