fixup! temp file handling part 3

bleachbit · Feb 9, 2017 · 772a35f · 772a35f
1 parent 77d91c8
commit 772a35f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/TestWorker.py b/tests/TestWorker.py
@@ -192,7 +192,7 @@ def action_test_helper(self, command, special_expected, errors_expected,
                            bytes_expected_posix, count_deleted_posix,
                            bytes_expected_nt, count_deleted_nt):
         ui = CLI.CliCallback()
-        (fd, filename) = tempfile.mkstemp(prefix='bleachbit-test-worker')
+        (fd, filename) = tempfile.mkstemp(prefix='bleachbit-test-worker', dir=self.tempdir)
         os.write(fd, '123')
         os.close(fd)
         self.assertExists(filename)
@@ -228,8 +228,7 @@ def test_DoesNotExist(self):
 
     def test_FunctionGenerator(self):
         """Test Worker using Action.FunctionGenerator"""
-        self.action_test_helper(
-            'function.generator', 1, 0, 4096 + 10, 1, 3 + 10, 1)
+        self.action_test_helper('function.generator', 1, 0, 4096 + 10, 1, 3 + 10, 1)
 
     def test_FunctionPath(self):
         """Test Worker using Action.FunctionPathAction"""