mirror of
https://github.com/saltstack/salt.git
synced 2025-04-17 10:10:20 +00:00
migrate test_riak to pytest
This commit is contained in:
Frode Gundersen
parent
3fff9eae63
commit
57b1dfbde2
2 changed files with 129 additions and 130 deletions
129
tests/pytests/unit/modules/test_riak.py
Normal file
129
tests/pytests/unit/modules/test_riak.py
Normal file
|
|
@ -0,0 +1,129 @@
|
|||
"""
|
||||
:codeauthor: Jayesh Kariya <jayeshk@saltstack.com>
|
||||
|
||||
Test cases for salt.modules.riak
|
||||
"""
|
||||
|
||||
import pytest
|
||||
|
||||
import salt.modules.riak as riak
|
||||
from tests.support.mock import patch
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def configure_loader_modules():
|
||||
return {riak: {}}
|
||||
|
||||
|
||||
def test_start():
|
||||
"""
|
||||
Test for start Riak
|
||||
"""
|
||||
with patch.object(
|
||||
riak, "__execute_cmd", return_value={"retcode": 0, "stdout": "success"}
|
||||
):
|
||||
assert riak.start() == {"success": True, "comment": "success"}
|
||||
|
||||
|
||||
def test_stop():
|
||||
"""
|
||||
Test for stop Riak
|
||||
"""
|
||||
with patch.object(
|
||||
riak, "__execute_cmd", return_value={"retcode": 0, "stdout": "success"}
|
||||
):
|
||||
assert riak.stop() == {"success": True, "comment": "success"}
|
||||
|
||||
|
||||
def test_cluster_join():
|
||||
"""
|
||||
Test for Join a Riak cluster
|
||||
"""
|
||||
with patch.object(
|
||||
riak, "__execute_cmd", return_value={"retcode": 0, "stdout": "success"}
|
||||
):
|
||||
assert riak.cluster_join("A", "B") == {"success": True, "comment": "success"}
|
||||
|
||||
|
||||
def test_cluster_leave():
|
||||
"""
|
||||
Test for leaving a Riak cluster
|
||||
"""
|
||||
with patch.object(
|
||||
riak, "__execute_cmd", return_value={"retcode": 0, "stdout": "success"}
|
||||
):
|
||||
assert riak.cluster_leave("A", "B") == {"success": True, "comment": "success"}
|
||||
|
||||
|
||||
def test_cluster_plan():
|
||||
"""
|
||||
Test for Review Cluster Plan
|
||||
"""
|
||||
with patch.object(
|
||||
riak, "__execute_cmd", return_value={"retcode": 0, "stdout": "success"}
|
||||
):
|
||||
assert riak.cluster_plan()
|
||||
|
||||
|
||||
def test_cluster_commit():
|
||||
"""
|
||||
Test for Commit Cluster Changes
|
||||
"""
|
||||
with patch.object(
|
||||
riak, "__execute_cmd", return_value={"retcode": 0, "stdout": "success"}
|
||||
):
|
||||
assert riak.cluster_commit() == {"success": True, "comment": "success"}
|
||||
|
||||
|
||||
def test_member_status():
|
||||
"""
|
||||
Test for Get cluster member status
|
||||
"""
|
||||
with patch.object(
|
||||
riak, "__execute_cmd", return_value={"stdout": "A:a/B:b\nC:c/D:d"}
|
||||
):
|
||||
assert riak.member_status() == {
|
||||
"membership": {},
|
||||
"summary": {
|
||||
"A": "a",
|
||||
"C": "c",
|
||||
"B": "b",
|
||||
"D": "d",
|
||||
"Exiting": 0,
|
||||
"Down": 0,
|
||||
"Valid": 0,
|
||||
"Leaving": 0,
|
||||
"Joining": 0,
|
||||
},
|
||||
}
|
||||
|
||||
|
||||
def test_status():
|
||||
"""
|
||||
Test status information
|
||||
"""
|
||||
ret = {"stdout": "vnode_map_update_time_95 : 0\nvnode_map_update_time_99 : 0"}
|
||||
|
||||
with patch.object(riak, "__execute_cmd", return_value=ret):
|
||||
assert riak.status() == {
|
||||
"vnode_map_update_time_95": "0",
|
||||
"vnode_map_update_time_99": "0",
|
||||
}
|
||||
|
||||
|
||||
def test_test():
|
||||
"""
|
||||
Test the Riak test
|
||||
"""
|
||||
with patch.object(
|
||||
riak, "__execute_cmd", return_value={"retcode": 0, "stdout": "success"}
|
||||
):
|
||||
assert riak.test() == {"success": True, "comment": "success"}
|
||||
|
||||
|
||||
def test_services():
|
||||
"""
|
||||
Test Riak Service List
|
||||
"""
|
||||
with patch.object(riak, "__execute_cmd", return_value={"stdout": "[a,b,c]"}):
|
||||
assert riak.services() == ["a", "b", "c"]
|
||||
|
|
@ -1,130 +0,0 @@
|
|||
"""
|
||||
:codeauthor: Jayesh Kariya <jayeshk@saltstack.com>
|
||||
"""
|
||||
|
||||
import salt.modules.riak as riak
|
||||
from tests.support.mixins import LoaderModuleMockMixin
|
||||
from tests.support.mock import patch
|
||||
from tests.support.unit import TestCase
|
||||
|
||||
|
||||
class RiakTestCase(TestCase, LoaderModuleMockMixin):
|
||||
"""
|
||||
Test cases for salt.modules.riak
|
||||
"""
|
||||
|
||||
def setup_loader_modules(self):
|
||||
return {riak: {}}
|
||||
|
||||
def test_start(self):
|
||||
"""
|
||||
Test for start Riak
|
||||
"""
|
||||
with patch.object(
|
||||
riak, "__execute_cmd", return_value={"retcode": 0, "stdout": "success"}
|
||||
):
|
||||
self.assertEqual(riak.start(), {"success": True, "comment": "success"})
|
||||
|
||||
def test_stop(self):
|
||||
"""
|
||||
Test for stop Riak
|
||||
"""
|
||||
with patch.object(
|
||||
riak, "__execute_cmd", return_value={"retcode": 0, "stdout": "success"}
|
||||
):
|
||||
self.assertEqual(riak.stop(), {"success": True, "comment": "success"})
|
||||
|
||||
def test_cluster_join(self):
|
||||
"""
|
||||
Test for Join a Riak cluster
|
||||
"""
|
||||
with patch.object(
|
||||
riak, "__execute_cmd", return_value={"retcode": 0, "stdout": "success"}
|
||||
):
|
||||
self.assertEqual(
|
||||
riak.cluster_join("A", "B"), {"success": True, "comment": "success"}
|
||||
)
|
||||
|
||||
def test_cluster_leave(self):
|
||||
"""
|
||||
Test for leaving a Riak cluster
|
||||
"""
|
||||
with patch.object(
|
||||
riak, "__execute_cmd", return_value={"retcode": 0, "stdout": "success"}
|
||||
):
|
||||
self.assertEqual(
|
||||
riak.cluster_leave("A", "B"), {"success": True, "comment": "success"}
|
||||
)
|
||||
|
||||
def test_cluster_plan(self):
|
||||
"""
|
||||
Test for Review Cluster Plan
|
||||
"""
|
||||
with patch.object(
|
||||
riak, "__execute_cmd", return_value={"retcode": 0, "stdout": "success"}
|
||||
):
|
||||
self.assertTrue(riak.cluster_plan())
|
||||
|
||||
def test_cluster_commit(self):
|
||||
"""
|
||||
Test for Commit Cluster Changes
|
||||
"""
|
||||
with patch.object(
|
||||
riak, "__execute_cmd", return_value={"retcode": 0, "stdout": "success"}
|
||||
):
|
||||
self.assertEqual(
|
||||
riak.cluster_commit(), {"success": True, "comment": "success"}
|
||||
)
|
||||
|
||||
def test_member_status(self):
|
||||
"""
|
||||
Test for Get cluster member status
|
||||
"""
|
||||
with patch.object(
|
||||
riak, "__execute_cmd", return_value={"stdout": "A:a/B:b\nC:c/D:d"}
|
||||
):
|
||||
self.assertDictEqual(
|
||||
riak.member_status(),
|
||||
{
|
||||
"membership": {},
|
||||
"summary": {
|
||||
"A": "a",
|
||||
"C": "c",
|
||||
"B": "b",
|
||||
"D": "d",
|
||||
"Exiting": 0,
|
||||
"Down": 0,
|
||||
"Valid": 0,
|
||||
"Leaving": 0,
|
||||
"Joining": 0,
|
||||
},
|
||||
},
|
||||
)
|
||||
|
||||
def test_status(self):
|
||||
"""
|
||||
Test status information
|
||||
"""
|
||||
ret = {"stdout": "vnode_map_update_time_95 : 0\nvnode_map_update_time_99 : 0"}
|
||||
|
||||
with patch.object(riak, "__execute_cmd", return_value=ret):
|
||||
self.assertEqual(
|
||||
riak.status(),
|
||||
{"vnode_map_update_time_95": "0", "vnode_map_update_time_99": "0"},
|
||||
)
|
||||
|
||||
def test_test(self):
|
||||
"""
|
||||
Test the Riak test
|
||||
"""
|
||||
with patch.object(
|
||||
riak, "__execute_cmd", return_value={"retcode": 0, "stdout": "success"}
|
||||
):
|
||||
self.assertEqual(riak.test(), {"success": True, "comment": "success"})
|
||||
|
||||
def test_services(self):
|
||||
"""
|
||||
Test Riak Service List
|
||||
"""
|
||||
with patch.object(riak, "__execute_cmd", return_value={"stdout": "[a,b,c]"}):
|
||||
self.assertEqual(riak.services(), ["a", "b", "c"])
|
||||
Loading…
Add table
Reference in a new issue