def test_iter_userinstalled_badfromrepo(self):
"""Test iter_userinstalled with a package installed from a bad repository."""
base = dnf.Base()
base._sack = support.mock_sack('main')
base._yumdb = support.MockYumDB()
pkg, = base.sack.query().installed().filter(name='pepper')
base.yumdb.get_package(pkg).get = {'reason': 'user', 'from_repo': 'anakonda'}.get
iterator = base.iter_userinstalled()
self.assertRaises(StopIteration, next, iterator)
def test_iter_userinstalled(self):
"""Test iter_userinstalled with a package installed by the user."""
base = support.Base()
base._sack = support.mock_sack('main')
base._priv_yumdb = support.MockYumDB()
pkg, = base.sack.query().installed().filter(name='pepper')
base._yumdb.get_package(pkg).get = {'reason': 'user', 'from_repo': 'main'}.get
iterator = base.iter_userinstalled()
self.assertEqual(next(iterator), pkg)
self.assertRaises(StopIteration, next, iterator)
