def test_run(self, pick_location):
"""Run the PrepareDownloadLocationTask class."""
dnf_manager = Mock()
with tempfile.TemporaryDirectory() as path:
# Mock the download location.
pick_location.return_value = path
# Create files in the download location.
os.mknod(os.path.join(path, "f1"))
os.mknod(os.path.join(path, "f2"))
os.mknod(os.path.join(path, "f3"))
task = PrepareDownloadLocationTask(dnf_manager)
assert task.run() == path
# The manager should apply the location.
dnf_manager.set_download_location.assert_called_once_with(path)
# The files should be deleted.
assert not os.path.exists(os.path.join(path, "f1"))
assert not os.path.exists(os.path.join(path, "f2"))
assert not os.path.exists(os.path.join(path, "f3"))
def install(self):
progress_message(N_('Starting package installation process'))
# Get the packages configuration and selection data.
configuration = self.get_packages_configuration()
selection = self.get_packages_selection()
# Add the rpm macros to the global transaction environment
task = SetRPMMacrosTask(configuration)
task.run()
try:
# Resolve packages.
task = ResolvePackagesTask(self._dnf_manager, selection)
task.run()
except NonCriticalInstallationError as e:
# FIXME: This is a temporary workaround.
# Allow users to handle the error. If they don't want
# to continue with the installation, raise a different
# exception to make sure that we will not run the error
# handler again.
if error_handler.cb(e) == ERROR_RAISE:
raise InstallationError(str(e)) from e
# Set up the download location.
task = PrepareDownloadLocationTask(self._dnf_manager)
task.run()
# Download the packages.
task = DownloadPackagesTask(self._dnf_manager)
task.progress_changed_signal.connect(self._progress_cb)
task.run()
# Install the packages.
task = InstallPackagesTask(self._dnf_manager)
task.progress_changed_signal.connect(self._progress_cb)
task.run()
# Clean up the download location.
task = CleanUpDownloadLocationTask(self._dnf_manager)
task.run()