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)