Merge pull request #3043 from copperchin/use-tmp-in-testutils

getpelican · Oct 26, 2022 · a51d75c · a51d75c
2 parents 1f6f4a3 + 27f2c67
commit a51d75c
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/pelican/tests/test_utils.py b/pelican/tests/test_utils.py
@@ -20,6 +20,14 @@
 class TestUtils(LoggedTestCase):
     _new_attribute = 'new_value'
 
+    def setUp(self):
+        super().setUp()
+        self.temp_output = mkdtemp(prefix='pelicantests.')
+
+    def tearDown(self):
+        super().tearDown()
+        shutil.rmtree(self.temp_output)
+
     @utils.deprecated_attribute(
         old='_old_attribute', new='_new_attribute',
         since=(3, 1, 0), remove=(4, 1, 3))
@@ -468,7 +476,7 @@ def create_file(name, content):
 
     def test_clean_output_dir(self):
         retention = ()
-        test_directory = os.path.join(os.path.dirname(__file__),
+        test_directory = os.path.join(self.temp_output,
                                       'clean_output')
         content = os.path.join(os.path.dirname(__file__), 'content')
         shutil.copytree(content, test_directory)
@@ -479,14 +487,14 @@ def test_clean_output_dir(self):
 
     def test_clean_output_dir_not_there(self):
         retention = ()
-        test_directory = os.path.join(os.path.dirname(__file__),
+        test_directory = os.path.join(self.temp_output,
                                       'does_not_exist')
         utils.clean_output_dir(test_directory, retention)
         self.assertFalse(os.path.exists(test_directory))
 
     def test_clean_output_dir_is_file(self):
         retention = ()
-        test_directory = os.path.join(os.path.dirname(__file__),
+        test_directory = os.path.join(self.temp_output,
                                       'this_is_a_file')
         f = open(test_directory, 'w')
         f.write('')