saltstack/salt
1
0
Fork 0
mirror of https://github.com/saltstack/salt.git synced 2025-04-17 10:10:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

migrate test_rdp to pytest

Browse source
This commit is contained in:
Frode Gundersen 2022-12-08 16:59:37 +00:00
parent 2bb2604f74
commit 01e96bc293
No known key found for this signature in database
GPG key ID: DAB4C1C375D2EF45
2 changed files with 56 additions and 52 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

56
tests/pytests/unit/modules/test_rdp.py Normal file
View file

@ -0,0 +1,56 @@
"""
:codeauthor: Jayesh Kariya <jayeshk@saltstack.com>
Test cases for salt.modules.rdp
"""
import pytest
import salt.modules.rdp as rdp
from tests.support.mock import MagicMock, patch
@pytest.fixture
def configure_loader_modules():
return {rdp: {}}
# 'enable' function tests: 1
def test_enable():
"""
Test if it enables RDP the service on the server
"""
mock = MagicMock(return_value=True)
with patch.dict(rdp.__salt__, {"cmd.run": mock}), patch(
"salt.modules.rdp._parse_return_code_powershell", MagicMock(return_value=0)
):
assert rdp.enable()
# 'disable' function tests: 1
def test_disable():
"""
Test if it disables RDP the service on the server
"""
mock = MagicMock(return_value=True)
with patch.dict(rdp.__salt__, {"cmd.run": mock}), patch(
"salt.modules.rdp._parse_return_code_powershell", MagicMock(return_value=0)
):
assert rdp.disable()
# 'status' function tests: 1
def test_status():
"""
Test if it shows rdp is enabled on the server
"""
mock = MagicMock(return_value="1")
with patch.dict(rdp.__salt__, {"cmd.run": mock}):
assert rdp.status()

52
tests/unit/modules/test_rdp.py
View file

@ -1,52 +0,0 @@
"""
:codeauthor: Jayesh Kariya <jayeshk@saltstack.com>
"""
import salt.modules.rdp as rdp
from tests.support.mixins import LoaderModuleMockMixin
from tests.support.mock import MagicMock, patch
from tests.support.unit import TestCase
class RdpTestCase(TestCase, LoaderModuleMockMixin):
"""
Test cases for salt.modules.rdp
"""
def setup_loader_modules(self):
return {rdp: {}}
# 'enable' function tests: 1
def test_enable(self):
"""
Test if it enables RDP the service on the server
"""
mock = MagicMock(return_value=True)
with patch.dict(rdp.__salt__, {"cmd.run": mock}), patch(
"salt.modules.rdp._parse_return_code_powershell", MagicMock(return_value=0)
):
self.assertTrue(rdp.enable())
# 'disable' function tests: 1
def test_disable(self):
"""
Test if it disables RDP the service on the server
"""
mock = MagicMock(return_value=True)
with patch.dict(rdp.__salt__, {"cmd.run": mock}), patch(
"salt.modules.rdp._parse_return_code_powershell", MagicMock(return_value=0)
):
self.assertTrue(rdp.disable())
# 'status' function tests: 1
def test_status(self):
"""
Test if it shows rdp is enabled on the server
"""
mock = MagicMock(return_value="1")
with patch.dict(rdp.__salt__, {"cmd.run": mock}):
self.assertTrue(rdp.status())