saltstack/salt
1
0
Fork 0
mirror of https://github.com/saltstack/salt.git synced 2025-04-17 10:10:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Add pytest helper to create temporary pillar files

Browse source
This commit is contained in:
Pedro Algarvio 2020-04-08 19:28:37 +01:00 • committed by Daniel Wozniak
parent 14e606b39c
commit 3f77f6eb28
1 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
tests/conftest.py
View file

@ -708,6 +708,22 @@ def temp_state_file(name, contents, saltenv="base", strip_first_newline=True):
)
@pytest.helpers.register
def temp_pillar_file(name, contents, saltenv="base", strip_first_newline=True):
if saltenv == "base":
directory = RUNTIME_VARS.TMP_PILLAR_TREE
elif saltenv == "prod":
directory = RUNTIME_VARS.TMP_PRODENV_PILLAR_TREE
else:
raise RuntimeError(
'"saltenv" can only be "base" or "prod", not "{}"'.format(saltenv)
)
return temp_file(
name, contents, directory=directory, strip_first_newline=strip_first_newline
)
# <---- Pytest Helpers -----------------------------------------------------------------------------------------------