def setUp(self):
cli = mock.MagicMock()
self.cmd = download.DownloadCommand(cli)
self.cmd.cli.base = dnf.cli.cli.BaseCli()
self.cmd.cli.base.add_remote_rpms = mock.MagicMock()
self.cmd.cli.base.download_packages = mock.Mock()
# point the Sack and Subject to out stubs
# b/c these are used in the _get_query methods
self.orig_sack = self.cmd.cli.base.sack
self.cmd.cli.base._sack = SackStub()
self.orig_subject = dnf.subject.Subject
dnf.subject.Subject = SubjectStub
self.cmd.opts = mock.Mock()
self.cmd.opts.resolve = False
self.cmd.opts.arch = []
repo = RepoStub('foo')
repo.enable()
self.cmd.base.repos.add(repo)
repo = RepoStub('foo-source')
repo.disable()
self.cmd.base.repos.add(repo)
repo = RepoStub('bar')
repo.enable()
self.cmd.base.repos.add(repo)
repo = RepoStub('foobar-source')
repo.disable()
self.cmd.base.repos.add(repo)
repo = RepoStub('foo-debuginfo')
repo.disable()
self.cmd.base.repos.add(repo)
def setUp(self):
def stub_fn(pkg_spec):
if '.src.rpm' in pkg_spec:
return Query.filter(sourcerpm=pkg_spec)
else:
q = Query.latest()
return [pkg for pkg in q if pkg_spec == pkg.name]
cli = mock.MagicMock()
self.cmd = download.DownloadCommand(cli)
self.cmd.cli.base.repos = dnf.repodict.RepoDict()
self.cmd._get_query = stub_fn
self.cmd._get_query_source = stub_fn
self.cmd.opts = mock.Mock()
self.cmd.opts.resolve = False
repo = RepoStub('foo')
repo.enable()
self.cmd.base.repos.add(repo)
repo = RepoStub('foo-source')
repo.disable()
self.cmd.base.repos.add(repo)
repo = RepoStub('bar')
repo.enable()
self.cmd.base.repos.add(repo)
repo = RepoStub('foobar-source')
repo.disable()
self.cmd.base.repos.add(repo)