def test_execute_PBConnectionLost(self): error = pb.PBConnectionLost() worker = Mock(spec=["workerId", "run"]) worker.run.side_effect = error task = Mock(spec=["call", "retryable", "priority"]) task.retryable = True handler = Mock() dfr = self.running.execute(worker, task) dfr.addErrback(handler) handler.assert_not_called() self.assertIsInstance(dfr, defer.Deferred) worker.run.assert_called_once_with(task.call) self.patches["_handle_start"].assert_called_once_with( task, worker.workerId, ) self.patches["_handle_retry"].assert_called_once_with(task, error) self.worklist.pushfront.assert_called_once_with( task.priority, task, ) self.workers.layoff.assert_called_once_with(worker) self.logger.error.assert_not_called() self.worklist.push.assert_not_called() self.patches["_handle_success"].assert_not_called() self.patches["_handle_failure"].assert_not_called() self.patches["_handle_error"].assert_not_called()
def makePBConnectionLostFailure(self): return failure.Failure(pb.PBConnectionLost("gone"))
def remote_print(message): worker1.master_persp.broker.transport.loseConnection() raise pb.PBConnectionLost("fake!")
def remote_print(what): raise pb.PBConnectionLost("fake!")