mirror of
https://github.com/saltstack/salt.git
synced 2025-04-17 10:10:20 +00:00
migrate test_genesis to pytest
This commit is contained in:
Frode Gundersen
Megan Wilhite
parent
72bc1c8664
commit
3367c18bf8
2 changed files with 190 additions and 196 deletions
190
tests/pytests/unit/modules/test_genesis.py
Normal file
190
tests/pytests/unit/modules/test_genesis.py
Normal file
|
|
@ -0,0 +1,190 @@
|
|||
"""
|
||||
:codeauthor: Rupesh Tare <rupesht@saltstack.com>
|
||||
"""
|
||||
|
||||
|
||||
import sys
|
||||
|
||||
import pytest
|
||||
|
||||
import salt.modules.genesis as genesis
|
||||
from tests.support.mock import MagicMock, patch
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def configure_loader_modules():
|
||||
return {genesis: {}}
|
||||
|
||||
|
||||
def test_bootstrap():
|
||||
"""
|
||||
Test for Create an image for a specific platform.
|
||||
"""
|
||||
# Changed in 3.7.0 pformat no longer includes the comma
|
||||
if sys.version_info >= (3, 7):
|
||||
exception_string = "Exception({})".format(repr("foo"))
|
||||
else:
|
||||
exception_string = "Exception({},)".format(repr("foo"))
|
||||
mock = MagicMock(return_value=False)
|
||||
with patch.dict(genesis.__salt__, {"file.directory_exists": mock}):
|
||||
mock = MagicMock(side_effect=Exception("foo"))
|
||||
with patch.dict(genesis.__salt__, {"file.mkdir": mock}):
|
||||
assert genesis.bootstrap("platform", "root") == {"Error": exception_string}
|
||||
|
||||
with patch.object(genesis, "_bootstrap_yum", return_value="A"):
|
||||
with patch.dict(
|
||||
genesis.__salt__,
|
||||
{
|
||||
"mount.umount": MagicMock(),
|
||||
"file.rmdir": MagicMock(),
|
||||
"file.directory_exists": MagicMock(),
|
||||
},
|
||||
):
|
||||
with patch.dict(
|
||||
genesis.__salt__, {"disk.blkid": MagicMock(return_value={})}
|
||||
):
|
||||
assert genesis.bootstrap("rpm", "root", "dir") is None
|
||||
|
||||
common_parms = {
|
||||
"platform": "deb",
|
||||
"root": "root",
|
||||
"img_format": "dir",
|
||||
"arch": "amd64",
|
||||
"flavor": "stable",
|
||||
"static_qemu": "qemu",
|
||||
}
|
||||
|
||||
param_sets = [
|
||||
{
|
||||
"params": {},
|
||||
"cmd": [
|
||||
"debootstrap",
|
||||
"--foreign",
|
||||
"--arch",
|
||||
"amd64",
|
||||
"stable",
|
||||
"root",
|
||||
"http://ftp.debian.org/debian/",
|
||||
],
|
||||
},
|
||||
{
|
||||
"params": {"pkgs": "vim"},
|
||||
"cmd": [
|
||||
"debootstrap",
|
||||
"--foreign",
|
||||
"--arch",
|
||||
"amd64",
|
||||
"--include",
|
||||
"vim",
|
||||
"stable",
|
||||
"root",
|
||||
"http://ftp.debian.org/debian/",
|
||||
],
|
||||
},
|
||||
{
|
||||
"params": {"pkgs": "vim,emacs"},
|
||||
"cmd": [
|
||||
"debootstrap",
|
||||
"--foreign",
|
||||
"--arch",
|
||||
"amd64",
|
||||
"--include",
|
||||
"vim,emacs",
|
||||
"stable",
|
||||
"root",
|
||||
"http://ftp.debian.org/debian/",
|
||||
],
|
||||
},
|
||||
{
|
||||
"params": {"pkgs": ["vim", "emacs"]},
|
||||
"cmd": [
|
||||
"debootstrap",
|
||||
"--foreign",
|
||||
"--arch",
|
||||
"amd64",
|
||||
"--include",
|
||||
"vim,emacs",
|
||||
"stable",
|
||||
"root",
|
||||
"http://ftp.debian.org/debian/",
|
||||
],
|
||||
},
|
||||
{
|
||||
"params": {"pkgs": ["vim", "emacs"], "exclude_pkgs": ["vim", "foo"]},
|
||||
"cmd": [
|
||||
"debootstrap",
|
||||
"--foreign",
|
||||
"--arch",
|
||||
"amd64",
|
||||
"--include",
|
||||
"vim,emacs",
|
||||
"--exclude",
|
||||
"vim,foo",
|
||||
"stable",
|
||||
"root",
|
||||
"http://ftp.debian.org/debian/",
|
||||
],
|
||||
},
|
||||
]
|
||||
|
||||
for param_set in param_sets:
|
||||
|
||||
with patch.dict(
|
||||
genesis.__salt__,
|
||||
{
|
||||
"mount.umount": MagicMock(),
|
||||
"file.rmdir": MagicMock(),
|
||||
"file.directory_exists": MagicMock(),
|
||||
"cmd.run": MagicMock(),
|
||||
"disk.blkid": MagicMock(return_value={}),
|
||||
},
|
||||
):
|
||||
with patch("salt.modules.genesis.salt.utils.path.which", return_value=True):
|
||||
with patch(
|
||||
"salt.modules.genesis.salt.utils.validate.path.is_executable",
|
||||
return_value=True,
|
||||
):
|
||||
param_set["params"].update(common_parms)
|
||||
assert genesis.bootstrap(**param_set["params"]) is None
|
||||
genesis.__salt__["cmd.run"].assert_any_call(
|
||||
param_set["cmd"], python_shell=False
|
||||
)
|
||||
|
||||
with patch.object(genesis, "_bootstrap_pacman", return_value="A") as pacman_patch:
|
||||
with patch.dict(
|
||||
genesis.__salt__,
|
||||
{
|
||||
"mount.umount": MagicMock(),
|
||||
"file.rmdir": MagicMock(),
|
||||
"file.directory_exists": MagicMock(),
|
||||
"disk.blkid": MagicMock(return_value={}),
|
||||
},
|
||||
):
|
||||
genesis.bootstrap("pacman", "root", "dir")
|
||||
pacman_patch.assert_called_with(
|
||||
"root", img_format="dir", exclude_pkgs=[], pkgs=[]
|
||||
)
|
||||
|
||||
|
||||
def test_avail_platforms():
|
||||
"""
|
||||
Test for Return which platforms are available
|
||||
"""
|
||||
with patch("salt.utils.path.which", MagicMock(return_value=False)):
|
||||
assert not genesis.avail_platforms()["deb"]
|
||||
|
||||
|
||||
def test_pack():
|
||||
"""
|
||||
Test for Pack up a directory structure, into a specific format
|
||||
"""
|
||||
with patch.object(genesis, "_tar", return_value="tar"):
|
||||
assert genesis.pack("name", "root") is None
|
||||
|
||||
|
||||
def test_unpack():
|
||||
"""
|
||||
Test for Unpack an image into a directory structure
|
||||
"""
|
||||
with patch.object(genesis, "_untar", return_value="untar"):
|
||||
assert genesis.unpack("name", "root") is None
|
||||
|
|
@ -1,196 +0,0 @@
|
|||
"""
|
||||
:codeauthor: Rupesh Tare <rupesht@saltstack.com>
|
||||
"""
|
||||
|
||||
|
||||
import sys
|
||||
|
||||
import salt.modules.genesis as genesis
|
||||
from tests.support.mixins import LoaderModuleMockMixin
|
||||
from tests.support.mock import MagicMock, patch
|
||||
from tests.support.unit import TestCase
|
||||
|
||||
|
||||
class GenesisTestCase(TestCase, LoaderModuleMockMixin):
|
||||
"""
|
||||
Test cases for salt.modules.genesis
|
||||
"""
|
||||
|
||||
def setup_loader_modules(self):
|
||||
return {genesis: {}}
|
||||
|
||||
def test_bootstrap(self):
|
||||
"""
|
||||
Test for Create an image for a specific platform.
|
||||
"""
|
||||
# Changed in 3.7.0 pformat no longer includes the comma
|
||||
if sys.version_info >= (3, 7):
|
||||
exception_string = "Exception({})".format(repr("foo"))
|
||||
else:
|
||||
exception_string = "Exception({},)".format(repr("foo"))
|
||||
mock = MagicMock(return_value=False)
|
||||
with patch.dict(genesis.__salt__, {"file.directory_exists": mock}):
|
||||
mock = MagicMock(side_effect=Exception("foo"))
|
||||
with patch.dict(genesis.__salt__, {"file.mkdir": mock}):
|
||||
self.assertEqual(
|
||||
genesis.bootstrap("platform", "root"), {"Error": exception_string}
|
||||
)
|
||||
|
||||
with patch.object(genesis, "_bootstrap_yum", return_value="A"):
|
||||
with patch.dict(
|
||||
genesis.__salt__,
|
||||
{
|
||||
"mount.umount": MagicMock(),
|
||||
"file.rmdir": MagicMock(),
|
||||
"file.directory_exists": MagicMock(),
|
||||
},
|
||||
):
|
||||
with patch.dict(
|
||||
genesis.__salt__, {"disk.blkid": MagicMock(return_value={})}
|
||||
):
|
||||
self.assertEqual(genesis.bootstrap("rpm", "root", "dir"), None)
|
||||
|
||||
common_parms = {
|
||||
"platform": "deb",
|
||||
"root": "root",
|
||||
"img_format": "dir",
|
||||
"arch": "amd64",
|
||||
"flavor": "stable",
|
||||
"static_qemu": "qemu",
|
||||
}
|
||||
|
||||
param_sets = [
|
||||
{
|
||||
"params": {},
|
||||
"cmd": [
|
||||
"debootstrap",
|
||||
"--foreign",
|
||||
"--arch",
|
||||
"amd64",
|
||||
"stable",
|
||||
"root",
|
||||
"http://ftp.debian.org/debian/",
|
||||
],
|
||||
},
|
||||
{
|
||||
"params": {"pkgs": "vim"},
|
||||
"cmd": [
|
||||
"debootstrap",
|
||||
"--foreign",
|
||||
"--arch",
|
||||
"amd64",
|
||||
"--include",
|
||||
"vim",
|
||||
"stable",
|
||||
"root",
|
||||
"http://ftp.debian.org/debian/",
|
||||
],
|
||||
},
|
||||
{
|
||||
"params": {"pkgs": "vim,emacs"},
|
||||
"cmd": [
|
||||
"debootstrap",
|
||||
"--foreign",
|
||||
"--arch",
|
||||
"amd64",
|
||||
"--include",
|
||||
"vim,emacs",
|
||||
"stable",
|
||||
"root",
|
||||
"http://ftp.debian.org/debian/",
|
||||
],
|
||||
},
|
||||
{
|
||||
"params": {"pkgs": ["vim", "emacs"]},
|
||||
"cmd": [
|
||||
"debootstrap",
|
||||
"--foreign",
|
||||
"--arch",
|
||||
"amd64",
|
||||
"--include",
|
||||
"vim,emacs",
|
||||
"stable",
|
||||
"root",
|
||||
"http://ftp.debian.org/debian/",
|
||||
],
|
||||
},
|
||||
{
|
||||
"params": {"pkgs": ["vim", "emacs"], "exclude_pkgs": ["vim", "foo"]},
|
||||
"cmd": [
|
||||
"debootstrap",
|
||||
"--foreign",
|
||||
"--arch",
|
||||
"amd64",
|
||||
"--include",
|
||||
"vim,emacs",
|
||||
"--exclude",
|
||||
"vim,foo",
|
||||
"stable",
|
||||
"root",
|
||||
"http://ftp.debian.org/debian/",
|
||||
],
|
||||
},
|
||||
]
|
||||
|
||||
for param_set in param_sets:
|
||||
|
||||
with patch.dict(
|
||||
genesis.__salt__,
|
||||
{
|
||||
"mount.umount": MagicMock(),
|
||||
"file.rmdir": MagicMock(),
|
||||
"file.directory_exists": MagicMock(),
|
||||
"cmd.run": MagicMock(),
|
||||
"disk.blkid": MagicMock(return_value={}),
|
||||
},
|
||||
):
|
||||
with patch(
|
||||
"salt.modules.genesis.salt.utils.path.which", return_value=True
|
||||
):
|
||||
with patch(
|
||||
"salt.modules.genesis.salt.utils.validate.path.is_executable",
|
||||
return_value=True,
|
||||
):
|
||||
param_set["params"].update(common_parms)
|
||||
self.assertEqual(genesis.bootstrap(**param_set["params"]), None)
|
||||
genesis.__salt__["cmd.run"].assert_any_call(
|
||||
param_set["cmd"], python_shell=False
|
||||
)
|
||||
|
||||
with patch.object(
|
||||
genesis, "_bootstrap_pacman", return_value="A"
|
||||
) as pacman_patch:
|
||||
with patch.dict(
|
||||
genesis.__salt__,
|
||||
{
|
||||
"mount.umount": MagicMock(),
|
||||
"file.rmdir": MagicMock(),
|
||||
"file.directory_exists": MagicMock(),
|
||||
"disk.blkid": MagicMock(return_value={}),
|
||||
},
|
||||
):
|
||||
genesis.bootstrap("pacman", "root", "dir")
|
||||
pacman_patch.assert_called_with(
|
||||
"root", img_format="dir", exclude_pkgs=[], pkgs=[]
|
||||
)
|
||||
|
||||
def test_avail_platforms(self):
|
||||
"""
|
||||
Test for Return which platforms are available
|
||||
"""
|
||||
with patch("salt.utils.path.which", MagicMock(return_value=False)):
|
||||
self.assertFalse(genesis.avail_platforms()["deb"])
|
||||
|
||||
def test_pack(self):
|
||||
"""
|
||||
Test for Pack up a directory structure, into a specific format
|
||||
"""
|
||||
with patch.object(genesis, "_tar", return_value="tar"):
|
||||
self.assertEqual(genesis.pack("name", "root"), None)
|
||||
|
||||
def test_unpack(self):
|
||||
"""
|
||||
Test for Unpack an image into a directory structure
|
||||
"""
|
||||
with patch.object(genesis, "_untar", return_value="untar"):
|
||||
self.assertEqual(genesis.unpack("name", "root"), None)
|
||||
Loading…
Add table
Reference in a new issue