def test_2stage(self): t = Pipeline.n(self.doubler.n(), self.doubler.n()) t.start(2) self._handle_messages() t.refresh() self.assertEqual(M.Message.m.find().count(), 0) self.assertEqual(M.TaskState.m.find().count(), 1) self.assertEqual(t.result.get(), 8)
def test_2stage(self): t = Pipeline.n( self.doubler.n(), self.doubler.n()) t.start(2) self._handle_messages() t.refresh() self.assertEqual(M.Message.m.find().count(), 0) self.assertEqual(M.TaskState.m.find().count(), 1) self.assertEqual(t.result.get(), 8)
def test_2stage_err(self): t = Pipeline.n(self.doubler.n(), self.doubler.n()) t.start(None) self._handle_messages() t.refresh() self.assertEqual(M.Message.m.find().count(), 0) self.assertEqual(M.TaskState.m.find().count(), 1) with self.assertRaises(exc.TaskError) as err: t.result.get() self.assertEqual(err.exception.args[0], TypeError)
def test_2stage_err(self): t = Pipeline.n( self.doubler.n(), self.doubler.n()) t.start(None) self._handle_messages() t.refresh() self.assertEqual(M.Message.m.find().count(), 0) self.assertEqual(M.TaskState.m.find().count(), 1) with self.assertRaises(exc.TaskError) as err: t.result.get() self.assertEqual(err.exception.args[0], TypeError)