mirror of
https://github.com/saltstack/salt.git
synced 2025-04-17 10:10:20 +00:00
migrate test_drbd to pytest
This commit is contained in:
Frode Gundersen
Pedro Algarvio
parent
9864347cac
commit
5aa89c01e9
2 changed files with 186 additions and 191 deletions
186
tests/pytests/unit/modules/test_drbd.py
Normal file
186
tests/pytests/unit/modules/test_drbd.py
Normal file
|
|
@ -0,0 +1,186 @@
|
|||
"""
|
||||
:codeauthor: Jayesh Kariya <jayeshk@saltstack.com>
|
||||
|
||||
Test cases for salt.modules.drbd
|
||||
"""
|
||||
|
||||
|
||||
import pytest
|
||||
|
||||
import salt.modules.drbd as drbd
|
||||
from tests.support.mock import MagicMock, patch
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def configure_loader_modules():
|
||||
return {drbd: {}}
|
||||
|
||||
|
||||
# 'overview' function tests: 1
|
||||
|
||||
|
||||
def test_overview():
|
||||
"""
|
||||
Test if it shows status of the DRBD devices
|
||||
"""
|
||||
ret = {
|
||||
"connection state": "True",
|
||||
"device": "Stack",
|
||||
"fs": "None",
|
||||
"local disk state": "UpToDate",
|
||||
"local role": "master",
|
||||
"minor number": "Salt",
|
||||
"mountpoint": "True",
|
||||
"partner disk state": "UpToDate",
|
||||
"partner role": "minion",
|
||||
"percent": "888",
|
||||
"remains": "666",
|
||||
"total size": "50",
|
||||
"used": "50",
|
||||
}
|
||||
mock = MagicMock(
|
||||
return_value=(
|
||||
"Salt:Stack True master/minion UpToDate/UpToDate True None 50 50 666 888"
|
||||
)
|
||||
)
|
||||
with patch.dict(drbd.__salt__, {"cmd.run": mock}):
|
||||
assert drbd.overview() == ret
|
||||
|
||||
ret = {
|
||||
"connection state": "True",
|
||||
"device": "Stack",
|
||||
"local disk state": "UpToDate",
|
||||
"local role": "master",
|
||||
"minor number": "Salt",
|
||||
"partner disk state": "partner",
|
||||
"partner role": "minion",
|
||||
"synched": "5050",
|
||||
"synchronisation: ": "syncbar",
|
||||
}
|
||||
mock = MagicMock(
|
||||
return_value=(
|
||||
"Salt:Stack True master/minion UpToDate/partner syncbar None 50 50"
|
||||
)
|
||||
)
|
||||
with patch.dict(drbd.__salt__, {"cmd.run": mock}):
|
||||
assert drbd.overview() == ret
|
||||
|
||||
|
||||
def test_status():
|
||||
"""
|
||||
Test if it shows status of the DRBD resources via drbdadm
|
||||
"""
|
||||
ret = [
|
||||
{
|
||||
"local role": "Primary",
|
||||
"local volumes": [{"disk": "UpToDate"}],
|
||||
"peer nodes": [
|
||||
{
|
||||
"peer volumes": [
|
||||
{
|
||||
"done": "96.47",
|
||||
"peer-disk": "Inconsistent",
|
||||
"replication": "SyncSource",
|
||||
}
|
||||
],
|
||||
"peernode name": "opensuse-node2",
|
||||
"role": "Secondary",
|
||||
}
|
||||
],
|
||||
"resource name": "single",
|
||||
}
|
||||
]
|
||||
|
||||
mock = MagicMock(
|
||||
return_value="""
|
||||
single role:Primary
|
||||
disk:UpToDate
|
||||
opensuse-node2 role:Secondary
|
||||
replication:SyncSource peer-disk:Inconsistent done:96.47
|
||||
"""
|
||||
)
|
||||
|
||||
with patch.dict(drbd.__salt__, {"cmd.run": mock}):
|
||||
assert drbd.status() == ret
|
||||
|
||||
ret = [
|
||||
{
|
||||
"local role": "Primary",
|
||||
"local volumes": [
|
||||
{"disk": "UpToDate", "volume": "0"},
|
||||
{"disk": "UpToDate", "volume": "1"},
|
||||
],
|
||||
"peer nodes": [
|
||||
{
|
||||
"peer volumes": [
|
||||
{"peer-disk": "UpToDate", "volume": "0"},
|
||||
{"peer-disk": "UpToDate", "volume": "1"},
|
||||
],
|
||||
"peernode name": "node2",
|
||||
"role": "Secondary",
|
||||
},
|
||||
{
|
||||
"peer volumes": [
|
||||
{"peer-disk": "UpToDate", "volume": "0"},
|
||||
{"peer-disk": "UpToDate", "volume": "1"},
|
||||
],
|
||||
"peernode name": "node3",
|
||||
"role": "Secondary",
|
||||
},
|
||||
],
|
||||
"resource name": "res",
|
||||
},
|
||||
{
|
||||
"local role": "Primary",
|
||||
"local volumes": [
|
||||
{"disk": "UpToDate", "volume": "0"},
|
||||
{"disk": "UpToDate", "volume": "1"},
|
||||
],
|
||||
"peer nodes": [
|
||||
{
|
||||
"peer volumes": [
|
||||
{"peer-disk": "UpToDate", "volume": "0"},
|
||||
{"peer-disk": "UpToDate", "volume": "1"},
|
||||
],
|
||||
"peernode name": "node2",
|
||||
"role": "Secondary",
|
||||
},
|
||||
{
|
||||
"peer volumes": [
|
||||
{"peer-disk": "UpToDate", "volume": "0"},
|
||||
{"peer-disk": "UpToDate", "volume": "1"},
|
||||
],
|
||||
"peernode name": "node3",
|
||||
"role": "Secondary",
|
||||
},
|
||||
],
|
||||
"resource name": "test",
|
||||
},
|
||||
]
|
||||
|
||||
mock = MagicMock(
|
||||
return_value="""
|
||||
res role:Primary
|
||||
volume:0 disk:UpToDate
|
||||
volume:1 disk:UpToDate
|
||||
node2 role:Secondary
|
||||
volume:0 peer-disk:UpToDate
|
||||
volume:1 peer-disk:UpToDate
|
||||
node3 role:Secondary
|
||||
volume:0 peer-disk:UpToDate
|
||||
volume:1 peer-disk:UpToDate
|
||||
|
||||
test role:Primary
|
||||
volume:0 disk:UpToDate
|
||||
volume:1 disk:UpToDate
|
||||
node2 role:Secondary
|
||||
volume:0 peer-disk:UpToDate
|
||||
volume:1 peer-disk:UpToDate
|
||||
node3 role:Secondary
|
||||
volume:0 peer-disk:UpToDate
|
||||
volume:1 peer-disk:UpToDate
|
||||
|
||||
"""
|
||||
)
|
||||
with patch.dict(drbd.__salt__, {"cmd.run": mock}):
|
||||
assert drbd.status() == ret
|
||||
|
|
@ -1,191 +0,0 @@
|
|||
"""
|
||||
:codeauthor: Jayesh Kariya <jayeshk@saltstack.com>
|
||||
"""
|
||||
|
||||
|
||||
import salt.modules.drbd as drbd
|
||||
from tests.support.mixins import LoaderModuleMockMixin
|
||||
from tests.support.mock import MagicMock, patch
|
||||
from tests.support.unit import TestCase
|
||||
|
||||
|
||||
class DrbdTestCase(TestCase, LoaderModuleMockMixin):
|
||||
"""
|
||||
Test cases for salt.modules.drbd
|
||||
"""
|
||||
|
||||
def setup_loader_modules(self):
|
||||
return {drbd: {}}
|
||||
|
||||
# 'overview' function tests: 1
|
||||
|
||||
def test_overview(self):
|
||||
"""
|
||||
Test if it shows status of the DRBD devices
|
||||
"""
|
||||
ret = {
|
||||
"connection state": "True",
|
||||
"device": "Stack",
|
||||
"fs": "None",
|
||||
"local disk state": "UpToDate",
|
||||
"local role": "master",
|
||||
"minor number": "Salt",
|
||||
"mountpoint": "True",
|
||||
"partner disk state": "UpToDate",
|
||||
"partner role": "minion",
|
||||
"percent": "888",
|
||||
"remains": "666",
|
||||
"total size": "50",
|
||||
"used": "50",
|
||||
}
|
||||
mock = MagicMock(
|
||||
return_value=(
|
||||
"Salt:Stack True master/minion UpToDate/UpToDate True None 50 50 666 888"
|
||||
)
|
||||
)
|
||||
with patch.dict(drbd.__salt__, {"cmd.run": mock}):
|
||||
self.assertDictEqual(drbd.overview(), ret)
|
||||
|
||||
ret = {
|
||||
"connection state": "True",
|
||||
"device": "Stack",
|
||||
"local disk state": "UpToDate",
|
||||
"local role": "master",
|
||||
"minor number": "Salt",
|
||||
"partner disk state": "partner",
|
||||
"partner role": "minion",
|
||||
"synched": "5050",
|
||||
"synchronisation: ": "syncbar",
|
||||
}
|
||||
mock = MagicMock(
|
||||
return_value=(
|
||||
"Salt:Stack True master/minion UpToDate/partner syncbar None 50 50"
|
||||
)
|
||||
)
|
||||
with patch.dict(drbd.__salt__, {"cmd.run": mock}):
|
||||
self.assertDictEqual(drbd.overview(), ret)
|
||||
|
||||
def test_status(self):
|
||||
"""
|
||||
Test if it shows status of the DRBD resources via drbdadm
|
||||
"""
|
||||
ret = [
|
||||
{
|
||||
"local role": "Primary",
|
||||
"local volumes": [{"disk": "UpToDate"}],
|
||||
"peer nodes": [
|
||||
{
|
||||
"peer volumes": [
|
||||
{
|
||||
"done": "96.47",
|
||||
"peer-disk": "Inconsistent",
|
||||
"replication": "SyncSource",
|
||||
}
|
||||
],
|
||||
"peernode name": "opensuse-node2",
|
||||
"role": "Secondary",
|
||||
}
|
||||
],
|
||||
"resource name": "single",
|
||||
}
|
||||
]
|
||||
|
||||
mock = MagicMock(
|
||||
return_value="""
|
||||
single role:Primary
|
||||
disk:UpToDate
|
||||
opensuse-node2 role:Secondary
|
||||
replication:SyncSource peer-disk:Inconsistent done:96.47
|
||||
"""
|
||||
)
|
||||
|
||||
with patch.dict(drbd.__salt__, {"cmd.run": mock}):
|
||||
try: # python2
|
||||
self.assertItemsEqual(drbd.status(), ret)
|
||||
except AttributeError: # python3
|
||||
self.assertCountEqual(drbd.status(), ret)
|
||||
|
||||
ret = [
|
||||
{
|
||||
"local role": "Primary",
|
||||
"local volumes": [
|
||||
{"disk": "UpToDate", "volume": "0"},
|
||||
{"disk": "UpToDate", "volume": "1"},
|
||||
],
|
||||
"peer nodes": [
|
||||
{
|
||||
"peer volumes": [
|
||||
{"peer-disk": "UpToDate", "volume": "0"},
|
||||
{"peer-disk": "UpToDate", "volume": "1"},
|
||||
],
|
||||
"peernode name": "node2",
|
||||
"role": "Secondary",
|
||||
},
|
||||
{
|
||||
"peer volumes": [
|
||||
{"peer-disk": "UpToDate", "volume": "0"},
|
||||
{"peer-disk": "UpToDate", "volume": "1"},
|
||||
],
|
||||
"peernode name": "node3",
|
||||
"role": "Secondary",
|
||||
},
|
||||
],
|
||||
"resource name": "test",
|
||||
},
|
||||
{
|
||||
"local role": "Primary",
|
||||
"local volumes": [
|
||||
{"disk": "UpToDate", "volume": "0"},
|
||||
{"disk": "UpToDate", "volume": "1"},
|
||||
],
|
||||
"peer nodes": [
|
||||
{
|
||||
"peer volumes": [
|
||||
{"peer-disk": "UpToDate", "volume": "0"},
|
||||
{"peer-disk": "UpToDate", "volume": "1"},
|
||||
],
|
||||
"peernode name": "node2",
|
||||
"role": "Secondary",
|
||||
},
|
||||
{
|
||||
"peer volumes": [
|
||||
{"peer-disk": "UpToDate", "volume": "0"},
|
||||
{"peer-disk": "UpToDate", "volume": "1"},
|
||||
],
|
||||
"peernode name": "node3",
|
||||
"role": "Secondary",
|
||||
},
|
||||
],
|
||||
"resource name": "res",
|
||||
},
|
||||
]
|
||||
|
||||
mock = MagicMock(
|
||||
return_value="""
|
||||
res role:Primary
|
||||
volume:0 disk:UpToDate
|
||||
volume:1 disk:UpToDate
|
||||
node2 role:Secondary
|
||||
volume:0 peer-disk:UpToDate
|
||||
volume:1 peer-disk:UpToDate
|
||||
node3 role:Secondary
|
||||
volume:0 peer-disk:UpToDate
|
||||
volume:1 peer-disk:UpToDate
|
||||
|
||||
test role:Primary
|
||||
volume:0 disk:UpToDate
|
||||
volume:1 disk:UpToDate
|
||||
node2 role:Secondary
|
||||
volume:0 peer-disk:UpToDate
|
||||
volume:1 peer-disk:UpToDate
|
||||
node3 role:Secondary
|
||||
volume:0 peer-disk:UpToDate
|
||||
volume:1 peer-disk:UpToDate
|
||||
|
||||
"""
|
||||
)
|
||||
with patch.dict(drbd.__salt__, {"cmd.run": mock}):
|
||||
try: # python2
|
||||
self.assertItemsEqual(drbd.status(), ret)
|
||||
except AttributeError: # python3
|
||||
self.assertCountEqual(drbd.status(), ret)
|
||||
Loading…
Add table
Reference in a new issue