mirror of
https://github.com/saltstack/salt.git
synced 2025-04-17 10:10:20 +00:00
migrate test_ret to pytest
This commit is contained in:
parent
3fff9eae63
commit
f37a1e60ca
2 changed files with 81 additions and 75 deletions
81
tests/pytests/unit/modules/test_ret.py
Normal file
81
tests/pytests/unit/modules/test_ret.py
Normal 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"
|
|
@ -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")
|
Loading…
Add table
Reference in a new issue