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))
def setUp(self):
self.f = MockFuture()
self.wi = WorkItem(self.f, self.closure, (), {})
def setUp(self):
self.f = MockFuture()
self.wi = WorkItem(self.f, self.closure, (), {})
