def test_map_task_failure(self): self.addEngine(1) t1 = task.MapTask(lambda x: 1/0,(10,)) d = self.tc.run(t1) d.addCallback(self.tc.get_task_result, block=True) d.addErrback(lambda f: self.assertRaises(ZeroDivisionError, f.raiseException)) return d
def test_map_task(self): self.addEngine(1) t1 = task.MapTask(lambda x: 2*x,(10,)) d = self.tc.run(t1) d.addCallback(self.tc.get_task_result, block=True) d.addCallback(lambda r: self.assertEquals(r,20)) t2 = task.MapTask(lambda : 20) d.addCallback(lambda _: self.tc.run(t2)) d.addCallback(self.tc.get_task_result, block=True) d.addCallback(lambda r: self.assertEquals(r,20)) t3 = task.MapTask(lambda x: x,(),{'x':20}) d.addCallback(lambda _: self.tc.run(t3)) d.addCallback(self.tc.get_task_result, block=True) d.addCallback(lambda r: self.assertEquals(r,20)) return d
def test_clear(self): self.addEngine(1) t1 = task.MapTask(lambda x: 2*x,(10,)) d = self.tc.run(t1) d.addCallback(lambda _: self.tc.get_task_result(0, block=True)) d.addCallback(lambda r: self.assertEquals(r,20)) d.addCallback(lambda _: self.tc.clear()) d.addCallback(lambda _: self.tc.get_task_result(0, block=True)) d.addErrback(lambda f: self.assertRaises(IndexError, f.raiseException)) return d