migrate test_grub_legacy to pytest

This commit is contained in:
Frode Gundersen 2022-12-05 16:22:35 +00:00
parent 36a4c99a24
commit 51b58567df
No known key found for this signature in database
GPG key ID: DAB4C1C375D2EF45
2 changed files with 47 additions and 47 deletions

View file

@ -0,0 +1,47 @@
"""
:codeauthor: Rupesh Tare <rupesht@saltstack.com>
Test cases for salt.modules.grub_legacy
"""
import errno
import pytest
import salt.modules.grub_legacy as grub_legacy
import salt.utils.stringutils
from salt.exceptions import CommandExecutionError
from tests.support.mock import MagicMock, mock_open, patch
@pytest.fixture
def configure_loader_modules():
return {grub_legacy: {}}
def test_version():
"""
Test for Return server version from grub --version
"""
mock = MagicMock(return_value="out")
with patch.dict(grub_legacy.__salt__, {"cmd.run": mock}):
assert grub_legacy.version() == "out"
def test_conf():
"""
Test for Parse GRUB conf file
"""
file_data = IOError(errno.EACCES, "Permission denied")
with patch("salt.utils.files.fopen", mock_open(read_data=file_data)), patch.object(
grub_legacy, "_detect_conf", return_value="A"
):
pytest.raises(CommandExecutionError, grub_legacy.conf)
file_data = salt.utils.stringutils.to_str("\n".join(["#", "A B C D,E,F G H"]))
with patch("salt.utils.files.fopen", mock_open(read_data=file_data)), patch.object(
grub_legacy, "_detect_conf", return_value="A"
):
conf = grub_legacy.conf()
assert conf == {"A": "B C D,E,F G H", "stanzas": []}, conf

View file

@ -1,47 +0,0 @@
"""
:codeauthor: Rupesh Tare <rupesht@saltstack.com>
"""
import errno
import salt.modules.grub_legacy as grub_legacy
import salt.utils.stringutils
from salt.exceptions import CommandExecutionError
from tests.support.mixins import LoaderModuleMockMixin
from tests.support.mock import MagicMock, mock_open, patch
from tests.support.unit import TestCase
class GrublegacyTestCase(TestCase, LoaderModuleMockMixin):
"""
Test cases for salt.modules.grub_legacy
"""
def setup_loader_modules(self):
return {grub_legacy: {}}
def test_version(self):
"""
Test for Return server version from grub --version
"""
mock = MagicMock(return_value="out")
with patch.dict(grub_legacy.__salt__, {"cmd.run": mock}):
self.assertEqual(grub_legacy.version(), "out")
def test_conf(self):
"""
Test for Parse GRUB conf file
"""
file_data = IOError(errno.EACCES, "Permission denied")
with patch(
"salt.utils.files.fopen", mock_open(read_data=file_data)
), patch.object(grub_legacy, "_detect_conf", return_value="A"):
self.assertRaises(CommandExecutionError, grub_legacy.conf)
file_data = salt.utils.stringutils.to_str("\n".join(["#", "A B C D,E,F G H"]))
with patch(
"salt.utils.files.fopen", mock_open(read_data=file_data)
), patch.object(grub_legacy, "_detect_conf", return_value="A"):
conf = grub_legacy.conf()
assert conf == {"A": "B C D,E,F G H", "stanzas": []}, conf