migrate test_ret to pytest

This commit is contained in:
Frode Gundersen 2022-12-08 21:08:17 +00:00
parent 3fff9eae63
commit f37a1e60ca
No known key found for this signature in database
GPG key ID: DAB4C1C375D2EF45
2 changed files with 81 additions and 75 deletions

View file

@ -0,0 +1,81 @@
"""
:codeauthor: Jayesh Kariya <jayeshk@saltstack.com>
Test cases for salt.modules.ret
"""
import pytest
import salt.loader
import salt.modules.ret as ret
from tests.support.mock import MagicMock, patch
@pytest.fixture
def configure_loader_modules():
return {ret: {}}
# 'get_jid' function tests: 1
def test_get_jid():
"""
Test if it return the information for a specified job id
"""
mock_ret = MagicMock(return_value="DB")
with patch.object(
salt.loader,
"returners",
MagicMock(return_value={"redis.get_jid": mock_ret}),
):
assert ret.get_jid("redis", "net") == "DB"
# 'get_fun' function tests: 1
def test_get_fun():
"""
Test if it return info about last time fun was called on each minion
"""
mock_ret = MagicMock(return_value="DB")
with patch.object(
salt.loader,
"returners",
MagicMock(return_value={"mysql.get_fun": mock_ret}),
):
assert ret.get_fun("mysql", "net") == "DB"
# 'get_jids' function tests: 1
def test_get_jids():
"""
Test if it return a list of all job ids
"""
mock_ret = MagicMock(return_value="DB")
with patch.object(
salt.loader,
"returners",
MagicMock(return_value={"mysql.get_jids": mock_ret}),
):
assert ret.get_jids("mysql") == "DB"
# 'get_minions' function tests: 1
def test_get_minions():
"""
Test if it return a list of all minions
"""
mock_ret = MagicMock(return_value="DB")
with patch.object(
salt.loader,
"returners",
MagicMock(return_value={"mysql.get_minions": mock_ret}),
):
assert ret.get_minions("mysql") == "DB"

View file

@ -1,75 +0,0 @@
"""
:codeauthor: Jayesh Kariya <jayeshk@saltstack.com>
"""
import salt.loader
import salt.modules.ret as ret
from tests.support.mixins import LoaderModuleMockMixin
from tests.support.mock import MagicMock, patch
from tests.support.unit import TestCase
class RetTestCase(TestCase, LoaderModuleMockMixin):
"""
Test cases for salt.modules.ret
"""
def setup_loader_modules(self):
return {ret: {}}
# 'get_jid' function tests: 1
def test_get_jid(self):
"""
Test if it return the information for a specified job id
"""
mock_ret = MagicMock(return_value="DB")
with patch.object(
salt.loader,
"returners",
MagicMock(return_value={"redis.get_jid": mock_ret}),
):
self.assertEqual(ret.get_jid("redis", "net"), "DB")
# 'get_fun' function tests: 1
def test_get_fun(self):
"""
Test if it return info about last time fun was called on each minion
"""
mock_ret = MagicMock(return_value="DB")
with patch.object(
salt.loader,
"returners",
MagicMock(return_value={"mysql.get_fun": mock_ret}),
):
self.assertEqual(ret.get_fun("mysql", "net"), "DB")
# 'get_jids' function tests: 1
def test_get_jids(self):
"""
Test if it return a list of all job ids
"""
mock_ret = MagicMock(return_value="DB")
with patch.object(
salt.loader,
"returners",
MagicMock(return_value={"mysql.get_jids": mock_ret}),
):
self.assertEqual(ret.get_jids("mysql"), "DB")
# 'get_minions' function tests: 1
def test_get_minions(self):
"""
Test if it return a list of all minions
"""
mock_ret = MagicMock(return_value="DB")
with patch.object(
salt.loader,
"returners",
MagicMock(return_value={"mysql.get_minions": mock_ret}),
):
self.assertEqual(ret.get_minions("mysql"), "DB")