From 2e7201fe3767e0e7fbc88afcd1590d7f51620e95 Mon Sep 17 00:00:00 2001
From: Mike McLean <mikem@redhat.com>
Date: Jun 27 2017 17:00:36 +0000
Subject: provide a temporary workdir for restart task unit tests


Followup to https://pagure.io/koji/pull-request/472
The test cases added in this PR left temporary files
in the checkout dir.

---

diff --git a/tests/test_lib_py2only/test_restart_tasks.py b/tests/test_lib_py2only/test_restart_tasks.py
index 7cf56a0..bfe7b67 100644
--- a/tests/test_lib_py2only/test_restart_tasks.py
+++ b/tests/test_lib_py2only/test_restart_tasks.py
@@ -1,4 +1,6 @@
 import mock
+import shutil
+import tempfile
 import unittest
 
 import koji.tasks
@@ -10,9 +12,12 @@ class TestRestartTask(unittest.TestCase):
         self.session = mock.MagicMock()
         self.options = mock.MagicMock()
         self.manager = mock.MagicMock()
+        self.workdir = tempfile.mkdtemp()
+        self.options.workdir = self.workdir
         self.safe_rmtree = mock.patch('koji.tasks.safe_rmtree').start()
 
     def tearDown(self):
+        shutil.rmtree(self.workdir)
         mock.patch.stopall()
 
     def get_handler(self, *args, **kwargs):
@@ -47,9 +52,12 @@ class TestRestartVerifyTask(unittest.TestCase):
         self.session = mock.MagicMock()
         self.options = mock.MagicMock()
         self.manager = mock.MagicMock()
+        self.workdir = tempfile.mkdtemp()
+        self.options.workdir = self.workdir
         self.safe_rmtree = mock.patch('koji.tasks.safe_rmtree').start()
 
     def tearDown(self):
+        shutil.rmtree(self.workdir)
         mock.patch.stopall()
 
     def get_handler(self, *args, **kwargs):
@@ -133,9 +141,12 @@ class TestRestartHostsTask(unittest.TestCase):
         self.session = mock.MagicMock()
         self.options = mock.MagicMock()
         self.manager = mock.MagicMock()
+        self.workdir = tempfile.mkdtemp()
+        self.options.workdir = self.workdir
         self.safe_rmtree = mock.patch('koji.tasks.safe_rmtree').start()
 
     def tearDown(self):
+        shutil.rmtree(self.workdir)
         mock.patch.stopall()
 
     def get_handler(self, *args, **kwargs):