def test_run_nonexistent(self, shutil_mock):
"""Run the CleanUpDownloadLocationTask class for nonexistent location."""
dnf_manager = DNFManager()
dnf_manager.set_download_location("/my/nonexistent/path")
task = CleanUpDownloadLocationTask(dnf_manager)
task.run()
shutil_mock.rmtree.assert_not_called()
def test_run(self):
"""Run the CleanUpDownloadLocationTask class."""
dnf_manager = DNFManager()
with tempfile.TemporaryDirectory() as path:
# Mock the download location.
dnf_manager.set_download_location(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 = CleanUpDownloadLocationTask(dnf_manager)
task.run()
# 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"))