mirror of
https://github.com/saltstack/salt.git
synced 2025-04-16 09:40:20 +00:00
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
"""
|
|
tests.support.cli_scripts
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
Code to generate Salt CLI scripts for test runs
|
|
"""
|
|
|
|
import logging
|
|
import os
|
|
|
|
from saltfactories.utils.cli_scripts import generate_script
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def get_script_path(bin_dir, script_name):
|
|
"""
|
|
Return the path to a testing runtime script, generating one if it does not yet exist
|
|
"""
|
|
# Late import
|
|
from tests.support.runtests import RUNTIME_VARS
|
|
|
|
if not os.path.isdir(bin_dir):
|
|
os.makedirs(bin_dir)
|
|
|
|
cli_script_name = "cli_{}.py".format(script_name.replace("-", "_"))
|
|
script_path = os.path.join(bin_dir, cli_script_name)
|
|
|
|
if not os.path.isfile(script_path):
|
|
kwargs = {
|
|
"code_dir": str(RUNTIME_VARS.CODE_DIR),
|
|
"bin_dir": bin_dir,
|
|
"script_name": script_name,
|
|
"coverage_rc_path": os.environ.get("COVERAGE_PROCESS_START"),
|
|
"coverage_db_path": os.environ.get("COVERAGE_FILE"),
|
|
"inject_sitecustomize": "COVERAGE_PROCESS_START" in os.environ,
|
|
}
|
|
generate_script(**kwargs)
|
|
log.info("Returning script path %r", script_path)
|
|
return script_path
|
|
|
|
|
|
class ScriptPathMixin:
|
|
def get_script_path(self, script_name):
|
|
"""
|
|
Return the path to a testing runtime script
|
|
"""
|
|
# Late import
|
|
from tests.support.runtests import RUNTIME_VARS
|
|
|
|
return get_script_path(RUNTIME_VARS.TMP_SCRIPT_DIR, script_name)
|