def test_processTransaction(self, process, trans_callback, rpm_callback):
progress = Mock()
# test
yum = Yum(progress=progress)
yum.processTransaction()
# validation
process.assert_called_once_with(
yum, trans_callback.return_value, rpmDisplay=rpm_callback.return_value)
trans_callback.assert_called_once_with(yum.progress)
rpm_callback.assert_called_once_with(yum.progress)
yum.progress.set_status.assert_called_once_with(True)
def test_doPluginSetup(self, setup, parser):
parser.return_value.parse_args.return_value = (Mock(), Mock())
yum = Yum()
yum.plugins = Mock()
parser.return_value.parse_args.reset_mock()
setup.reset_mock()
# test
yum.doPluginSetup()
# validation
setup.assert_called_once_with(yum)
parser.return_value.parse_args.assert_once_called_with([])
yum.plugins.setCmdLine.assert_called_once_with(*parser.return_value.parse_args.return_value)
def test_init(self, parser, download):
parser.return_value.parse_args.return_value = (Mock(), Mock())
progress = Mock()
# test
yum = Yum(True, progress)
# validation
parser.assert_called_with()
parser.return_value.parse_args.assert_called_with([])
download.assert_called_once_with(progress)
yum.repos.setProgressBar.assert_called_once_with(download.return_value)
self.assertEqual(yum.progress, progress)
self.assertTrue(progress.push_step.called)
def test_processTransaction_raised(self, process, trans_callback, rpm_callback):
process.side_effect = ValueError
progress = Mock()
# test
yum = Yum(progress=progress)
self.assertRaises(ValueError, yum.processTransaction)
# validation
process.assert_called_once_with(
yum, trans_callback.return_value, rpmDisplay=rpm_callback.return_value)
trans_callback.assert_called_once_with(yum.progress)
rpm_callback.assert_called_once_with(yum.progress)
yum.progress.set_status.assert_called_once_with(False)
del yum
def test_registerCommand(self):
# test
yum = Yum()
yum.registerCommand('')
