valitydev/salt
14
0
Fork 0
mirror of https://github.com/valitydev/salt.git synced 2024-11-07 08:58:59 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use with to open temp file

Browse Source
This commit is contained in:
twangboy 2017-09-14 08:42:19 -06:00
parent 352fe69e35
commit 056f3bb4c0
No known key found for this signature in database
GPG Key ID: 93FF3BDEB278C9EB
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/unit/modules/test_file.py
View File

@ -111,8 +111,8 @@ class FileReplaceTestCase(TestCase, LoaderModuleMockMixin):
os.remove(tfile.name)
# A newline should not be added in empty files
tfile = tempfile.NamedTemporaryFile('w+b', delete=False)
tfile.close()
with tempfile.NamedTemporaryFile('w+b', delete=False) as tfile:
pass
filemod.replace(tfile.name, **args)
expected = args['repl'] + os.linesep
with salt.utils.fopen(tfile.name) as tfile2:
@ -343,8 +343,8 @@ class FileBlockReplaceTestCase(TestCase, LoaderModuleMockMixin):
os.remove(tfile.name)
# A newline should not be added in empty files
tfile = tempfile.NamedTemporaryFile(mode='w+b', delete=False)
tfile.close()
with tempfile.NamedTemporaryFile(mode='w+b', delete=False) as tfile:
pass
filemod.blockreplace(tfile.name, **args)
with salt.utils.fopen(tfile.name) as tfile2:
self.assertEqual(tfile2.read(), block)