def test_run_raised(self, aborted, select_plugin, pending): plugin = Mock() sn = 1234 pending.return_value.get.return_value = Document(sn=sn) select_plugin.side_effect = ValueError aborted.side_effect = [False, True] # test scheduler = Scheduler(plugin) scheduler.run() # validation pending.return_value.commit.assert_called_once_with(sn)
def test_run(self, builtin, pending, task, select_plugin, tx, aborted): _builtin = Mock(name='builtin') plugin = Mock(name='plugin') task_list = [ Mock(name='task-1'), Mock(name='task-2'), ] tx_list = [ Mock(name='tx-1'), Mock(name='tx-2'), ] request_list = [ Document(sn=1), Document(sn=2), ] task.side_effect = task_list tx.side_effect = tx_list aborted.side_effect = [False, False, True] pending.return_value.get.side_effect = request_list builtin.return_value = _builtin builtin.return_value.provides.side_effect = [True, False] select_plugin.side_effect = [_builtin, plugin] # test scheduler = Scheduler(plugin) scheduler.run() # validation _builtin.pool.run.assert_called_once_with(task_list[0]) plugin.pool.run.assert_called_once_with(task_list[1]) self.assertEqual( select_plugin.call_args_list, [ ((request_list[0],), {}), ((request_list[1],), {}), ]) self.assertEqual( tx.call_args_list, [ ((_builtin, pending.return_value, request_list[0]), {}), ((plugin, pending.return_value, request_list[1]), {}) ]) self.assertEqual( task.call_args_list, [ ((tx_list[0],), {}), ((tx_list[1],), {}), ])
def test_run(self, builtin, pending, task, select_plugin, tx, aborted): builtin.return_value = Mock(name='builtin') plugin = Mock(name='plugin') task_list = [ Mock(name='task-1'), Mock(name='task-2'), ] tx_list = [ Mock(name='tx-1'), Mock(name='tx-2'), ] request_list = [ Document(sn=1), Document(sn=2), ] task.side_effect = task_list tx.side_effect = tx_list aborted.side_effect = [False, False, True] pending.return_value.get.side_effect = request_list builtin.return_value.provides.side_effect = [True, False] select_plugin.side_effect = [builtin.return_value, plugin] # test scheduler = Scheduler(plugin) scheduler.run() # validation builtin.return_value.pool.run.assert_called_once_with(task_list[0]) plugin.pool.run.assert_called_once_with(task_list[1]) self.assertEqual( select_plugin.call_args_list, [ ((request_list[0],), {}), ((request_list[1],), {}), ]) self.assertEqual( tx.call_args_list, [ ((builtin.return_value, pending.return_value, request_list[0]), {}), ((plugin, pending.return_value, request_list[1]), {}) ]) self.assertEqual( task.call_args_list, [ ((tx_list[0],), {}), ((tx_list[1],), {}), ])