class TestWorkItem(TestCase): calls = list() def closure(self): self.calls.append(True) return True def setUp(self): self.f = MockFuture() self.wi = WorkItem(self.f, self.closure, (), {}) def test_success(self): self.wi.run() self.assertEquals(self.calls, [True]) self.assertTrue(self.f.result) def test_cancelled(self): self.f.cancelled = True self.wi.run() self.assertEquals(self.calls, []) def test_exception(self): self.wi.fn = lambda: 1/0 self.wi.run() self.assertTrue(isinstance(self.f.exception, ZeroDivisionError))
class TestWorkItem(TestCase): calls = list() def closure(self): self.calls.append(True) return True def setUp(self): self.f = MockFuture() self.wi = WorkItem(self.f, self.closure, (), {}) def test_success(self): self.wi.run() self.assertEquals(self.calls, [True]) self.assertTrue(self.f.result) def test_cancelled(self): self.f.cancelled = True self.wi.run() self.assertEquals(self.calls, []) def test_exception(self): self.wi.fn = lambda: 1 / 0 self.wi.run() self.assertTrue(isinstance(self.f.exception, ZeroDivisionError))