particle-os/debian-koji
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

provide a temporary workdir for restart task unit tests

Browse source 
Followup to https://pagure.io/koji/pull-request/472
The test cases added in this PR left temporary files
in the checkout dir.
This commit is contained in:
Mike McLean 2017-06-21 14:33:00 -04:00
parent efd08fbcfd
commit 76352b4968
1 changed files with 11 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/test_lib_py2only/test_restart_tasks.py
View file

@ -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):