def test_heartbeat_server_disconnected(self):
conn = Connection('localhost', io_loop=self.io_loop)
def clean_up():
conn.status == status.CLOSED
self.stop()
conn.on_disconnect = clean_up
def launch_heartbeat():
conn.heartbeat = 1
HeartbeatMonitor(conn).start()
conn.connect(launch_heartbeat)
self.io_loop.add_timeout(time.time() + 3, clean_up)
self.wait()
def test_heartbeat_client_disconnected(self):
conn = Connection('localhost', heartbeat=1, io_loop=self.io_loop)
conn.process_heartbeat = lambda hb: None
conn.on_disconnect = self.stop
conn.connect(lambda: None)
self.wait()
