Use salt.utils.fopen

This commit is contained in:
Daniel A. Wozniak 2018-04-13 13:14:19 -07:00
parent aa965310f1
commit 3c565d7e54
No known key found for this signature in database
GPG key ID: 166B9D2C06C82D61

View file

@ -68,10 +68,10 @@ def _setup_test_dir(src_dir, test_dir):
os.makedirs(test_dir)
salt.utils.files.recursive_copy(src_dir, test_dir)
filename = os.path.join(test_dir, 'non_ascii')
with open(filename, 'wb') as fp:
with salt.utils.fopen(filename, 'wb') as fp:
fp.write(b'Assun\xc3\xa7\xc3\xa3o' + os.linesep)
filename = os.path.join(test_dir, 'hello_simple')
with open(filename, 'wb') as fp:
with salt.utils.fopen(filename, 'wb') as fp:
fp.write(b'world' + os.linesep)
filename = os.path.join(test_dir, 'hello_import')
lines = [
@ -79,7 +79,7 @@ def _setup_test_dir(src_dir, test_dir):
r"{% from 'macro' import mymacro -%}",
r"{{ mymacro('Hey') ~ mymacro(a|default('a'), b|default('b')) }}",
]
with open(filename, 'wb') as fp:
with salt.utils.fopen(filename, 'wb') as fp:
for line in lines:
fp.write(line.encode('utf-8') + os.linesep)