def testReceiveIter(self): test_channel = Channel() def sender(): for i in range(10): test_channel.send(i) t = Tasklet.new(sender)() x = [] for i in test_channel.receive_n(10): x.append(i) self.assertEquals(range(10), x) t = Tasklet.new(sender)() try: Tasklet.set_current_timeout(1.0) x = [] for i in test_channel: x.append(i) except TimeoutError: pass finally: Tasklet.set_current_timeout(TIMEOUT_NEVER) self.assertEquals(range(10), x)
def testReceiveIter(self): test_channel = Channel() def sender(): for i in range(10): test_channel.send(i) t = Tasklet.new(sender)() x = [] for i in test_channel.receive_n(10): x.append(i) self.assertEquals(range(10), x) t = Tasklet.new(sender)() try: Tasklet.set_current_timeout(1.0) x = [] for i in test_channel: x.append(i) except TimeoutError: pass finally: Tasklet.set_current_timeout(TIMEOUT_NEVER) self.assertEquals(range(10), x)
def testPushPop4(self): self.assertEquals(TIMEOUT_NEVER, Timeout.current()) Tasklet.set_current_timeout(10.0) self.assertAlmostEqual(10.0, Timeout.current(), places=1) Timeout.push(TIMEOUT_CURRENT) self.assertAlmostEqual(10.0, Timeout.current(), places=1) Timeout.pop() self.assertAlmostEqual(10.0, Timeout.current(), places=1) Tasklet.set_current_timeout(TIMEOUT_NEVER)
def testPushPop4(self): self.assertEquals(TIMEOUT_NEVER, Timeout.current()) Tasklet.set_current_timeout(10.0) self.assertAlmostEqual(10.0, Timeout.current(), places = 1) Timeout.push(TIMEOUT_CURRENT) self.assertAlmostEqual(10.0, Timeout.current(), places = 1) Timeout.pop() self.assertAlmostEqual(10.0, Timeout.current(), places = 1) Tasklet.set_current_timeout(TIMEOUT_NEVER)
def testTimeout(self): self.assertEquals(TIMEOUT_NEVER, Tasklet.get_current_timeout()) Tasklet.set_current_timeout(10) self.assertAlmostEqual(10.0, Tasklet.get_current_timeout(), places = 1) Tasklet.sleep(1.0) self.assertAlmostEqual(9.0, Tasklet.get_current_timeout(), places = 1) Tasklet.sleep(1.0) self.assertAlmostEqual(8.0, Tasklet.get_current_timeout(), places = 1) current = Tasklet.current() current.timeout = 10.0 self.assertAlmostEqual(10.0, current.timeout, places = 1) Tasklet.sleep(1.0) self.assertAlmostEqual(9.0, current.timeout, places = 1) current.timeout = TIMEOUT_NEVER self.assertEquals(TIMEOUT_NEVER, current.timeout)
def testTimeout(self): self.assertEquals(TIMEOUT_NEVER, Tasklet.get_current_timeout()) Tasklet.set_current_timeout(10) self.assertAlmostEqual(10.0, Tasklet.get_current_timeout(), places = 1) Tasklet.sleep(1.0) self.assertAlmostEqual(9.0, Tasklet.get_current_timeout(), places = 1) Tasklet.sleep(1.0) self.assertAlmostEqual(8.0, Tasklet.get_current_timeout(), places = 1) current = Tasklet.current() current.timeout = 10.0 self.assertAlmostEqual(10.0, current.timeout, places = 1) Tasklet.sleep(1.0) self.assertAlmostEqual(9.0, current.timeout, places = 1) current.timeout = TIMEOUT_NEVER self.assertEquals(TIMEOUT_NEVER, current.timeout)
def run(self): Tasklet.set_current_timeout(2.0, False) super(MemcacheTaskletPool, self).run()