tests/pytests/unit/utils/test_slack.py - master - githubmirror/salt

tests/pytests/unit/utils/test_slack.py
61 строка2 KB
jeanluc
Remove more unnecessary references
18 апр 2024, 01:22
18 апр 2024, 01:229063d0c
"""
Test case for the slack utils module
"""

import logging

import pytest

import salt.utils.slack as slack
from tests.support.mock import MagicMock, patch

log = logging.getLogger(__name__)


@pytest.fixture
def configure_loader_modules():
    yield {slack: {}}


def test_query():
    """
    Test case for the query function in the slack utils module
    """
    function = "message"
    api_key = "xoxp-xxxxxxxxxx-xxxxxxxxxx-xxxxxxxxxx-xxxxxx"
    args = None
    method = "POST"
    header_dict = {"Content-Type": "application/x-www-form-urlen coded"}
    data = "channel=%23general&username=Slack+User&as_user=Slack+User&text=%60%60%60id%3A+minion%0D%0Afunction%3A+test.ping%0D%0Afunction+args%3A+%5B%5D%0D%0Ajid%3A+20201017004822956482%0D%0Areturn%3A+true%0A%0D%0A%60%60%60"
    opts = None

    mock_result = {
        "body": '{"ok": false, "error": "token_revoked"}',
        "status": 200,
        "dict": {"ok": False, "error": "token_revoked"},
    }
    mock = MagicMock(return_value=mock_result)
    with patch("salt.utils.http.query", mock):
        expected = {"message": "token_revoked", "res": False}
        ret = slack.query(function, api_key, args, method, header_dict, data, opts)
        assert ret == expected

    mock_result = {
        "status": 0,
        "error": (
            "[SSL: CERTIFICATE_VERIFY_FAILED] certificate verify failed: unable to get"
            " local issuer certificate (_ssl.c:1076)"
        ),
    }
    mock = MagicMock(return_value=mock_result)
    with patch("salt.utils.http.query", mock):
        expected = {"message": "invalid_auth", "res": False}
        ret = slack.query(function, api_key, args, method, header_dict, data, opts)
        assert ret == expected

    mock_result = {"status": 0, "dict": {}}
    mock = MagicMock(return_value=mock_result)
    with patch("salt.utils.http.query", mock):
        expected = {"message": "Unknown response", "res": False}
        ret = slack.query(function, api_key, args, method, header_dict, data, opts)
        assert ret == expected