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],), {}),
])
