mirror of
https://github.com/saltstack/salt.git
synced 2025-04-16 09:40:20 +00:00
104 lines
2.9 KiB
Python
104 lines
2.9 KiB
Python
import pytest
|
|
|
|
import salt.utils.path
|
|
|
|
pytestmark = [
|
|
pytest.mark.core_test,
|
|
pytest.mark.windows_whitelisted,
|
|
]
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def cmd(modules):
|
|
return modules.cmd
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def exitcode_script(state_tree):
|
|
exit_code = 12345
|
|
script_contents = f"""
|
|
Write-Host "Expected exit code: {exit_code}"
|
|
exit {exit_code}
|
|
"""
|
|
with pytest.helpers.temp_file("exit_code.ps1", script_contents, state_tree):
|
|
yield exit_code
|
|
|
|
|
|
@pytest.fixture(params=["powershell", "pwsh"])
|
|
def shell(request):
|
|
"""
|
|
This will run the test on powershell and powershell core (pwsh). If
|
|
powershell core is not installed that test run will be skipped
|
|
"""
|
|
if request.param == "pwsh" and salt.utils.path.which("pwsh") is None:
|
|
pytest.skip("Powershell 7 Not Present")
|
|
return request.param
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def account():
|
|
with pytest.helpers.create_account() as _account:
|
|
yield _account
|
|
|
|
|
|
@pytest.fixture
|
|
def issue_56195(state_tree):
|
|
contents = """
|
|
[CmdLetBinding()]
|
|
Param(
|
|
[SecureString] $SecureString
|
|
)
|
|
$Credential = New-Object System.Net.NetworkCredential("DummyId", $SecureString)
|
|
$Credential.Password
|
|
"""
|
|
with pytest.helpers.temp_file("test.ps1", contents, state_tree / "issue-56195"):
|
|
yield
|
|
|
|
|
|
@pytest.mark.skip_unless_on_windows(reason="Minion is not Windows")
|
|
def test_windows_script_args_powershell(cmd, shell, issue_56195):
|
|
"""
|
|
Ensure that powershell processes an inline script with args where the args
|
|
contain powershell that needs to be rendered
|
|
"""
|
|
password = "i like cheese"
|
|
args = (
|
|
"-SecureString (ConvertTo-SecureString -String '{}' -AsPlainText -Force)"
|
|
" -ErrorAction Stop".format(password)
|
|
)
|
|
script = "salt://issue-56195/test.ps1"
|
|
|
|
ret = cmd.script(source=script, args=args, shell=shell, saltenv="base")
|
|
|
|
assert ret["stdout"] == password
|
|
|
|
|
|
@pytest.mark.skip_unless_on_windows(reason="Minion is not Windows")
|
|
def test_windows_script_args_powershell_runas(cmd, shell, account, issue_56195):
|
|
"""
|
|
Ensure that powershell processes an inline script with args where the args
|
|
contain powershell that needs to be rendered
|
|
"""
|
|
password = "i like cheese"
|
|
args = (
|
|
"-SecureString (ConvertTo-SecureString -String '{}' -AsPlainText -Force)"
|
|
" -ErrorAction Stop".format(password)
|
|
)
|
|
script = "salt://issue-56195/test.ps1"
|
|
|
|
ret = cmd.script(
|
|
source=script,
|
|
args=args,
|
|
shell=shell,
|
|
saltenv="base",
|
|
runas=account.username,
|
|
password=account.password,
|
|
)
|
|
|
|
assert ret["stdout"] == password
|
|
|
|
|
|
@pytest.mark.skip_unless_on_windows(reason="Minion is not Windows")
|
|
def test_windows_script_exitcode(cmd, shell, exitcode_script):
|
|
ret = cmd.script("salt://exit_code.ps1", shell=shell, saltenv="base")
|
|
assert ret["retcode"] == exitcode_script
|