diff --git a/tests/common/utils/test_file_util.py b/tests/common/utils/test_file_util.py index f1514e5d0..930f20b3f 100644 --- 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 index 66c525dff..8a4465b4e 100644 --- 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)