def test_timed_request_timeout(self): class Parser(): def __init__(self, nc, t): self.nc = nc self.t = t def read(self, data=''): self.nc._process_pong() nc = Client() nc._ps = Parser(nc, self) yield nc.connect(io_loop=self.io_loop) with self.assertRaises(tornado.gen.TimeoutError): yield nc.timed_request("hello", "world", 500)
def test_custom_ping_interval(self): class Parser(): def __init__(self, nc, t): self.nc = nc self.t = t def read(self, data=''): self.t.assertEqual(1, len(self.nc._pongs)) self.nc._process_pong() self.t.assertEqual(0, len(self.nc._pongs)) nc = Client() nc._ps = Parser(nc, self) yield nc.connect(io_loop=self.io_loop, ping_interval=0.01) yield tornado.gen.sleep(1) self.assertEqual(3, nc._pings_outstanding) self.assertFalse(nc.is_connected()) self.assertTrue(nc.is_reconnecting())
def test_flush_timeout(self): class Parser(): def __init__(self, nc, t): self.nc = nc self.t = t def read(self, data=''): self.t.assertEqual(1, self.nc._pings_outstanding) self.t.assertEqual(1, len(self.nc._pongs)) self.nc._process_pong() self.t.assertEqual(0, len(self.nc._pongs)) nc = Client() nc._ps = Parser(nc, self) yield nc.connect(io_loop=self.io_loop) with self.assertRaises(tornado.gen.TimeoutError): yield nc.flush(1) self.assertEqual(1, len(nc._pongs)) nc.flush(1) self.assertEqual(2, nc._pings_outstanding) self.assertEqual(2, len(nc._pongs))