Remove usages of tempfile.mktemp (Azure#3270)

* Remove usages of tempfile.mktemp * unused import --------- Co-authored-by: narrieta@microsoft <narrieta>
nagworld9 · Dec 3, 2024 · dd2deb7 · dd2deb7
1 parent 3e2a70d
commit dd2deb7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/tests/common/utils/test_file_util.py b/tests/common/utils/test_file_util.py
@@ -80,7 +80,7 @@ def test_append_file(self):
         os.remove(test_file)
 
     def test_findre_in_file(self):
-        fp = tempfile.mktemp()
+        fp = os.path.join(self.tmp_dir, "test_findre_in_file")
         with open(fp, 'w') as f:
             f.write(
 '''
@@ -107,7 +107,7 @@ def test_findre_in_file(self):
             fileutil.findre_in_file(fp, "^Do not match.*"))
 
     def test_findstr_in_file(self):
-        fp = tempfile.mktemp()
+        fp = os.path.join(self.tmp_dir, "test_findstr_in_file")
         with open(fp, 'w') as f:
             f.write(
 '''

diff --git a/tests/pa/test_provision.py b/tests/pa/test_provision.py
@@ -17,7 +17,6 @@
 
 import os
 import re
-import tempfile
 import unittest
 
 import azurelinuxagent.common.conf as conf
@@ -55,8 +54,7 @@ def test_provision(self, mock_util, distro_name, distro_version, distro_full_nam
 
     def test_customdata(self):
         base64data = 'Q3VzdG9tRGF0YQ=='
-        data = DefaultOSUtil().decode_customdata(base64data)
-        fileutil.write_file(tempfile.mktemp(), data)
+        DefaultOSUtil().decode_customdata(base64data)
 
     @patch('azurelinuxagent.common.conf.get_provision_enabled',
         return_value=False)