fix

Author: JohananOppongAmoateng

tests/panels/test_profiling.py

@@ -3,6 +3,7 @@
 import sys
 import tempfile
 import unittest
+from unittest import mock
 
 from django.contrib.auth.models import User
 from django.core import signing
@@ -84,8 +85,7 @@ def test_generate_stats_no_profiler(self):
         self.assertIsNone(self.panel.generate_stats(self.request, response))
 
     def test_generate_stats_signed_path(self):
-        tmpdir = tempfile.mkdtemp()
-        try:
+        with tempfile.TemporaryDirectory() as tmpdir:
             with self.settings(DEBUG_TOOLBAR_CONFIG={"PROFILER_PROFILE_ROOT": tmpdir}):
                 response = self.panel.process_request(self.request)
                 self.panel.generate_stats(self.request, response)
@@ -94,8 +94,6 @@ def test_generate_stats_signed_path(self):
                 # Check that it's a valid signature
                 filename = signing.loads(path)
                 self.assertTrue(filename.endswith(".prof"))
-        finally:
-            shutil.rmtree(tmpdir)
 
     def test_generate_stats_no_root(self):
         response = self.panel.process_request(self.request)
@@ -114,11 +112,10 @@ def test_generate_stats_no_root_func(self):
         self.panel.generate_stats(self.request, response)
         self.assertNotIn("func_list", self.panel.get_stats())
 
-    @unittest.mock.patch("cProfile.Profile.dump_stats")
+    @mock.patch("cProfile.Profile.dump_stats")
     def test_generate_stats_oserror(self, mock_dump_stats):
         mock_dump_stats.side_effect = OSError
-        tmpdir = tempfile.mkdtemp()
-        try:
+        with tempfile.TemporaryDirectory() as tmpdir:
             with self.settings(DEBUG_TOOLBAR_CONFIG={"PROFILER_PROFILE_ROOT": tmpdir}):
                 response = self.panel.process_request(self.request)
                 with self.assertLogs(
@@ -128,8 +125,6 @@ def test_generate_stats_oserror(self, mock_dump_stats):
                 self.assertIn("Failed to dump profiling stats", cm.output[0])
                 # Ensure prof_file_path is not set/updated if dump fails
                 self.assertFalse(hasattr(self.panel, "prof_file_path"))
-        finally:
-            shutil.rmtree(tmpdir)
 
 
 @override_settings(