def test_render_and_file(self):
err_ini = utils.get_render('configs/config.ini',
dict(config=dict(test=[])))
for line in err_ini.split('\n'):
self.assertEqual(line[0], '#')
self.assertTrue('# Invalid config.' in err_ini, err_ini)
config_dict = dict(config=dict(main=dict(test_key="test_value")))
ini = utils.get_render('configs/config.ini', config_dict)
with utils.tmp_file_context(ini, delete=False) as file:
file_name = file.name
file.write('\n')
with open(file_name, 'r') as tmp_file:
self.assertEqual(tmp_file.read(), test_config)
try:
self.assertFalse(utils.os.path.exists(file_name))
except AssertionError: # nocv
utils.os.remove(file_name)
try:
with utils.tmp_file(ini) as file:
file_name = file.name
file.write('\n')
with open(file_name, 'r') as tmp_file:
self.assertEqual(tmp_file.read(), test_config)
raise Exception('Normal')
except AssertionError: # nocv
raise
except Exception:
pass
with utils.tmp_file_context() as file:
with open(file.name, 'w') as output:
with utils.redirect_stdany(output):
print("Test")
with open(file.name, 'r') as output:
self.assertEqual(output.read(), "Test\n")
def get_generated_vars(self):
tmp = None
obj_vars = self.get_vars()
if "ansible_ssh_private_key_file" in obj_vars:
tmp = tmp_file()
tmp.write(obj_vars["ansible_ssh_private_key_file"])
obj_vars["ansible_ssh_private_key_file"] = tmp.name
return obj_vars, tmp
def get_generated_vars(self, tmp_dir='/tmp') -> Tuple[Dict, List]:
files = []
obj_vars = self.get_vars()
if "ansible_ssh_private_key_file" in obj_vars:
tmp = tmp_file(dir=tmp_dir)
tmp.write(obj_vars["ansible_ssh_private_key_file"])
obj_vars["ansible_ssh_private_key_file"] = tmp.name
files.append(tmp)
return dict(obj_vars), files
def get_generated_vars(self):
files = []
obj_vars = self.get_vars()
if "ansible_ssh_private_key_file" in obj_vars:
tmp = tmp_file()
tmp.write(obj_vars["ansible_ssh_private_key_file"])
obj_vars["ansible_ssh_private_key_file"] = tmp.name
files.append(tmp)
return dict(obj_vars), files
def __generate_arg_file(self, value: Text) -> Tuple[Text, List[tmp_file]]:
file = tmp_file(value, dir=self.cwd)
return file.name, [file]
def file(self) -> Union[tmp_file, Text]:
self._file = self._file or tmp_file(self.raw, dir=self.tmpdir)
return self._file
def __generate_arg_file(self, value):
file = tmp_file(value)
return file.name, [file]
def file(self):
self._file = self._file or tmp_file(self.raw)
return self._file
def test_magic_enter_exit(self):
tmp = tmp_file(mode="r")
with tmp as test_tmp_file:
self.assertEqual(test_tmp_file, tmp)
tmp = tmp_file(mode="r")
self.assertEqual(tmp.__exit__(ValueError, 22, "Traceback"), False)
