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

moving test_sshknownhosts.py to pytest.

Browse source
This commit is contained in:
Gareth J. Greenaway 2022-10-02 11:44:33 -07:00
parent a3e5ba47d4
commit 513b43b6ce
2 changed files with 88 additions and 69 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

88
tests/pytests/unit/roster/test_sshknownhosts.py Normal file
View file

@ -0,0 +1,88 @@
import pytest
import salt.config
import salt.loader
import salt.roster.sshknownhosts as sshknownhosts
from tests.support.mock import patch
@pytest.fixture
def ALL_DICT():
yield {
"server1": {"host": "server1"},
"server2": {"host": "server2"},
"server3.local": {"host": "server3.local"},
"eu-mysql-1.local": {"host": "eu-mysql-1.local"},
"eu-mysql-2": {"host": "eu-mysql-2"},
"eu-mysql-2.local": {"host": "eu-mysql-2.local"},
}
@pytest.fixture
def GLOB_DICT():
yield {
"server1": {"host": "server1"},
"server2": {"host": "server2"},
"server3.local": {"host": "server3.local"},
}
@pytest.fixture
def PCRE_DICT():
yield {"eu-mysql-2": {"host": "eu-mysql-2"}}
@pytest.fixture
def known_hosts():
yield """
server1 ecdsa-sha2-nistp256 AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAyNTYAAABBBD0vTE0R76xiKEXAdebZW0a3xGLeP2Fet/5YHQgprry3wuXzjBJwGcm8PVFNfbK/C7oAgFUg8NVX7xqQnScekJg=
server2 ssh-rsa AAAAB3NzaC1yc2EAAAABIwAAAQEAui+dKujjKF92dDdM9hZzCd+BdTDRnvsWqMf88NjushOmFCt/8zXbB1TvYQmdCcy1qXqhmkbgUdtVuLHnhncf/niCtyih3K3ZR7NpecBydcC+0xv0UeXk/xCGcwM2V0BuukrV/5qRqhyG0rK1hd+Iv9fkB0/s8D/HLcEB1/V4g77XxPGnI7lNANFbZpWs1LrnAec7JIkHO9MHEfuhQWZR6+/iIXIwQoc1RCToQbWQFCYFwrnDrAUHC2+izJiP2VDNW6xboVcf6DpwydfYvFdM8Mo97DEcchlwIWhmGl//LpnwafujFZCE5vDveA8X4uKZEXxoCmUPIGfkx6xIzzTkqQ==
server3.local ecdsa-sha2-nistp256 AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAyNTYAAABBBAgKWpCT7JIeK/qzwE5lUQkLRfkRa5WnyyeF+aYCKDUHB4b4Pn+acm8FOca+riulPDY/gJhb0MX3Rf/t6MrEHQA=
eu-mysql-1.local ssh-rsa AAAAB3NzaC1yc2EAAAABIwAAAQEAsuToIp6iqJ3lHPQzCTiNf5F8uf/CjAljuxRjURYCbQydts2lnnqTpjamL1b8/FpvB1dDlA71G79yTftVZ8EqL2VaN0tL242MXaqy2nmeVjy89dtOyk35IHwQe8Bi6mu3vLYCFnysiAvrtLQMFe8jNjndsvf27LNKox8pIAyOyN3hONL+bXEcPB2RjIUL8wS8uTeOueuPbVwc1cHkUuMjlNzsH3l6KMVjJZ8keFdRj8iogV8oZGR3KGoPfX4aZDt9S+L/k97fWkOhSKLWkKbplEcmIjuF5pgZLO3Wf35eLZN12PcHuX7WFWZi+UxjJDW2VLaP867La4YXDEU3LNdPEQ==
eu-mysql-2,eu-mysql-2.local ecdsa-sha2-nistp256 AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAyNTYAAABBBHFnjWT+gnUGRA2zW+LGZdebSkUVKBb6F/XCcDrtBZmaxCNaS/+F6SYzXP4MghCQhXFEPd7MpFnwPV8giU1NUag="""
@pytest.fixture
def configure_loader_modules(salt_master_factory, tmp_path):
opts = salt_master_factory.config.copy()
utils = salt.loader.utils(opts)
runner = salt.loader.runner(opts, utils=utils)
return {
sshknownhosts: {
"__opts__": {},
"__runner__": runner,
"__utils__": utils,
}
}
def test_all(known_hosts, tmp_path, ALL_DICT):
with pytest.helpers.temp_file(
"known_hosts", known_hosts, directory=tmp_path
) as known_hosts_file:
opts = {"ssh_known_hosts_file": str(known_hosts_file)}
with patch.dict(sshknownhosts.__opts__, opts):
targets = sshknownhosts.targets(tgt="*")
assert targets == ALL_DICT
def test_glob(known_hosts, tmp_path, GLOB_DICT):
with pytest.helpers.temp_file(
"known_hosts", known_hosts, directory=tmp_path
) as known_hosts_file:
opts = {"ssh_known_hosts_file": str(known_hosts_file)}
with patch.dict(sshknownhosts.__opts__, opts):
targets = sshknownhosts.targets(tgt="server*")
assert targets == GLOB_DICT
def test_pcre(known_hosts, tmp_path, PCRE_DICT):
with pytest.helpers.temp_file(
"known_hosts", known_hosts, directory=tmp_path
) as known_hosts_file:
opts = {"ssh_known_hosts_file": str(known_hosts_file)}
with patch.dict(sshknownhosts.__opts__, opts):
targets = sshknownhosts.targets(tgt="eu-mysql-2$", tgt_type="pcre")
assert targets == PCRE_DICT

69
tests/unit/roster/test_sshknownhosts.py
View file

@ -1,69 +0,0 @@
import os
import salt.config
import salt.loader
import salt.roster.sshknownhosts as sshknownhosts
from tests.support import mixins
from tests.support.mock import patch
from tests.support.runtests import RUNTIME_VARS
from tests.support.unit import TestCase
_ALL = {
"server1": {"host": "server1"},
"server2": {"host": "server2"},
"server3.local": {"host": "server3.local"},
"eu-mysql-1.local": {"host": "eu-mysql-1.local"},
"eu-mysql-2": {"host": "eu-mysql-2"},
"eu-mysql-2.local": {"host": "eu-mysql-2.local"},
}
_TEST_GLOB = {
"server1": {"host": "server1"},
"server2": {"host": "server2"},
"server3.local": {"host": "server3.local"},
}
_TEST_PCRE = {"eu-mysql-2": {"host": "eu-mysql-2"}}
class SSHKnownHostsRosterTestCase(TestCase, mixins.LoaderModuleMockMixin):
@classmethod
def setUpClass(cls):
cls.tests_dir = os.path.join(
RUNTIME_VARS.TESTS_DIR, "unit/files/rosters/sshknownhosts/"
)
cls.opts = {"ssh_known_hosts_file": "known_hosts"}
@classmethod
def tearDownClass(cls):
delattr(cls, "tests_dir")
delattr(cls, "opts")
def setup_loader_modules(self):
opts = salt.config.master_config(
os.path.join(RUNTIME_VARS.TMP_CONF_DIR, "master")
)
utils = salt.loader.utils(opts)
runner = salt.loader.runner(opts, utils=utils)
return {
sshknownhosts: {"__utils__": utils, "__opts__": {}, "__runner__": runner}
}
def test_all(self):
self.opts["ssh_known_hosts_file"] = os.path.join(self.tests_dir, "known_hosts")
with patch.dict(sshknownhosts.__opts__, self.opts):
targets = sshknownhosts.targets(tgt="*")
self.assertDictEqual(targets, _ALL)
def test_glob(self):
self.opts["ssh_known_hosts_file"] = os.path.join(self.tests_dir, "known_hosts")
with patch.dict(sshknownhosts.__opts__, self.opts):
targets = sshknownhosts.targets(tgt="server*")
self.assertDictEqual(targets, _TEST_GLOB)
def test_pcre(self):
self.opts["ssh_known_hosts_file"] = os.path.join(self.tests_dir, "known_hosts")
with patch.dict(sshknownhosts.__opts__, self.opts):
targets = sshknownhosts.targets(tgt="eu-mysql-2$", tgt_type="pcre")
self.assertDictEqual(targets, _TEST_PCRE)