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))
