tests/pytests/integration/modules/state/test_state.py - master - githubmirror/salt

tests/pytests/integration/modules/state/test_state.py
161 строка5 KB
Erik Johnson
Add umask as a global state argument
01 ноя 2022, 23:18
01 ноя 2022, 23:18765a5f0
import hashlib
import os
import zipfile

import pytest
from saltfactories.utils.functional import MultiStateResult

import salt.utils.files
import salt.utils.stringutils
from tests.support.runtests import RUNTIME_VARS


def test_get_file_from_env_in_top_match(salt_cli, salt_sub_minion):
    tgt = os.path.join(RUNTIME_VARS.TMP, "prod-cheese-file")
    try:
        ret = salt_cli.run("state.highstate", minion_tgt=salt_sub_minion.id)
        assert ret.returncode == 0
        assert os.path.isfile(tgt)
        with salt.utils.files.fopen(tgt, "r") as cheese:
            data = salt.utils.stringutils.to_unicode(cheese.read())
        assert "Gromit" in data
        assert "Comte" in data
    finally:
        if os.path.exists(tgt):
            os.unlink(tgt)


def test_issue_56131(salt_minion, base_env_state_tree_root_dir, tmp_path):
    """
    archive.extracted fails if setting an unless clause and pip is not installed.
    """
    zipfile_path = base_env_state_tree_root_dir / "issue-56131.zip"
    with zipfile.ZipFile(
        str(zipfile_path), "w", compression=zipfile.ZIP_DEFLATED
    ) as myzip:
        myzip.writestr("issue-56131.txt", "issue-56131")
    zipfile_hash = hashlib.sha256(zipfile_path.read_bytes()).hexdigest()
    sls_contents = """
    {}:
      archive.extracted:
        - source: salt://issue-56131.zip
        - source_hash: sha256={}
        - archive_format: zip
        - enforce_toplevel: False
        - unless:
          - echo hello; exit 1
    """.format(
        tmp_path, zipfile_hash
    )
    pip_path_dir = tmp_path / "fakepip"
    pip_path_dir.mkdir()
    pip_path = pip_path_dir / "pip.py"
    pip_path.write_text('raise ImportError("No module named pip")')

    environ = os.environ.copy()
    pythonpath = os.environ.get("PYTHONPATH")
    if pythonpath is None:
        pythonpath = str(pip_path_dir)
    else:
        pythonpath = os.pathsep.join([str(pip_path_dir)] + pythonpath.split(os.pathsep))
    environ["PYTHONPATH"] = pythonpath
    salt_call_cli = salt_minion.salt_call_cli(environ=environ)
    extract_path = tmp_path / "issue-56131.txt"
    try:
        assert extract_path.exists() is False
        with pytest.helpers.temp_file(
            "issue-56131.sls", sls_contents, base_env_state_tree_root_dir
        ):
            ret = salt_call_cli.run("state.sls", "issue-56131")
            assert ret.returncode == 0
            for state_return in MultiStateResult(ret.data):
                assert state_return.result is True
            assert extract_path.exists()
    finally:
        zipfile_path.unlink()


@pytest.mark.skip_unless_on_windows(
    reason="Tested in tests/pytests/functional/modules/state/test_state.py"
)
def test_pydsl(salt_call_cli, base_env_state_tree_root_dir, tmp_path):
    """
    Test the basics of the pydsl
    """
    testfile = tmp_path / "testfile"
    sls_contents = """
    #!pydsl

    state("{}").file("touch")
    """.format(
        testfile
    )
    with pytest.helpers.temp_file(
        "pydsl.sls", sls_contents, base_env_state_tree_root_dir
    ):
        ret = salt_call_cli.run("state.sls", "pydsl")
        assert ret.returncode == 0
        assert ret.data
        for state_return in MultiStateResult(ret.data):
            assert state_return.result is True
        assert testfile.exists()


@pytest.mark.skip_unless_on_windows(
    reason="Tested in tests/pytests/functional/modules/state/test_state.py"
)
def test_state_sls_unicode_characters(salt_call_cli, base_env_state_tree_root_dir):
    """
    test state.sls when state file contains non-ascii characters
    """
    sls_contents = """
    echo1:
      cmd.run:
        - name: "echo 'This is Æ test!'"
    """
    with pytest.helpers.temp_file(
        "issue-46672.sls", sls_contents, base_env_state_tree_root_dir
    ):
        ret = salt_call_cli.run("state.sls", "issue-46672")
        assert ret.returncode == 0
        assert ret.data
        expected = "cmd_|-echo1_|-echo 'This is Æ test!'_|-run"
        assert expected in ret.data


@pytest.mark.skip_on_windows(reason="umask is a no-op on Windows")
@pytest.mark.parametrize("umask", (22, "022"))
def test_umask_022(salt_call_cli, umask):
    """
    Should produce a file with mode 644
    """
    with pytest.helpers.temp_file() as name:
        name.unlink()
        salt_call_cli.run("state.single", fun="file.touch", name=str(name), umask=umask)
        assert oct(name.stat().st_mode)[-3:] == "644"


@pytest.mark.skip_on_windows(reason="umask is a no-op on Windows")
@pytest.mark.parametrize("umask", (27, "027"))
def test_umask_027(salt_call_cli, umask):
    """
    Should produce a file with mode 640
    """
    with pytest.helpers.temp_file() as name:
        name.unlink()
        salt_call_cli.run("state.single", fun="file.touch", name=str(name), umask=umask)
        assert oct(name.stat().st_mode)[-3:] == "640"


@pytest.mark.skip_on_windows(reason="umask is a no-op on Windows")
@pytest.mark.parametrize("umask", (999, "999", "foo"))
def test_umask_invalid(salt_call_cli, umask):
    """
    Test invalid umask values
    """
    with pytest.helpers.temp_file() as name:
        name.unlink()
        ret = salt_call_cli.run(
            "state.single", fun="file.touch", name=str(name), umask=umask
        )
        assert ret.data == [f"Invalid umask: {umask}"]