mirror of
https://github.com/saltstack/salt.git
synced 2025-04-17 10:10:20 +00:00
Migrated test_deb_postgre.py to pytests
This commit is contained in:
David Murphy
Pedro Algarvio
parent
4dbd8caec2
commit
0963209cc3
2 changed files with 159 additions and 184 deletions
159
tests/pytests/unit/modules/test_deb_postgres.py
Normal file
159
tests/pytests/unit/modules/test_deb_postgres.py
Normal file
|
|
@ -0,0 +1,159 @@
|
|||
import logging
|
||||
|
||||
import pytest
|
||||
|
||||
import salt.modules.deb_postgres as deb_postgres
|
||||
from tests.support.mock import Mock, patch
|
||||
|
||||
log = logging.getLogger(__name__)
|
||||
|
||||
|
||||
LSCLUSTER = """\
|
||||
8.4 main 5432 online postgres /srv/8.4/main \
|
||||
/var/log/postgresql/postgresql-8.4-main.log
|
||||
9.1 main 5433 online postgres /srv/9.1/main \
|
||||
/var/log/postgresql/postgresql-9.1-main.log
|
||||
"""
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def configure_loader_modules():
|
||||
return {
|
||||
deb_postgres: {
|
||||
"__salt__": {
|
||||
"config.option": Mock(),
|
||||
"cmd.run_all": Mock(return_value={"stdout": LSCLUSTER}),
|
||||
"file.chown": Mock(),
|
||||
"file.remove": Mock(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
def test_cluster_create():
|
||||
with patch("salt.utils.path.which", Mock(return_value="/usr/bin/pg_createcluster")):
|
||||
expected_cmdstr = (
|
||||
"/usr/bin/pg_createcluster "
|
||||
"--port 5432 --locale fr_FR --encoding UTF-8 "
|
||||
"--datadir /opt/postgresql "
|
||||
"9.3 main"
|
||||
)
|
||||
|
||||
deb_postgres.cluster_create(
|
||||
"9.3",
|
||||
"main",
|
||||
port="5432",
|
||||
locale="fr_FR",
|
||||
encoding="UTF-8",
|
||||
datadir="/opt/postgresql",
|
||||
)
|
||||
assert deb_postgres.__salt__["cmd.run_all"].call_args[0][0] == expected_cmdstr
|
||||
|
||||
|
||||
def test_cluster_create_with_initdb_options():
|
||||
with patch("salt.utils.path.which", Mock(return_value="/usr/bin/pg_createcluster")):
|
||||
expected_cmdstr = (
|
||||
"/usr/bin/pg_createcluster "
|
||||
"--port 5432 --locale fr_FR --encoding UTF-8 "
|
||||
"--datadir /opt/postgresql "
|
||||
"11 main "
|
||||
"-- "
|
||||
"--allow-group-access "
|
||||
"--data-checksums "
|
||||
"--wal-segsize 32"
|
||||
)
|
||||
|
||||
deb_postgres.cluster_create(
|
||||
"11",
|
||||
"main",
|
||||
port="5432",
|
||||
locale="fr_FR",
|
||||
encoding="UTF-8",
|
||||
datadir="/opt/postgresql",
|
||||
allow_group_access=True,
|
||||
data_checksums=True,
|
||||
wal_segsize="32",
|
||||
)
|
||||
assert deb_postgres.__salt__["cmd.run_all"].call_args[0][0] == expected_cmdstr
|
||||
|
||||
|
||||
def test_cluster_create_with_float():
|
||||
with patch("salt.utils.path.which", Mock(return_value="/usr/bin/pg_createcluster")):
|
||||
expected_cmdstr = (
|
||||
"/usr/bin/pg_createcluster "
|
||||
"--port 5432 --locale fr_FR --encoding UTF-8 "
|
||||
"--datadir /opt/postgresql "
|
||||
"9.3 main"
|
||||
)
|
||||
|
||||
deb_postgres.cluster_create(
|
||||
9.3,
|
||||
"main",
|
||||
port="5432",
|
||||
locale="fr_FR",
|
||||
encoding="UTF-8",
|
||||
datadir="/opt/postgresql",
|
||||
)
|
||||
assert deb_postgres.__salt__["cmd.run_all"].call_args[0][0] == expected_cmdstr
|
||||
|
||||
|
||||
def test_parse_pg_lsclusters():
|
||||
with patch("salt.utils.path.which", Mock(return_value="/usr/bin/pg_lsclusters")):
|
||||
stdout = LSCLUSTER
|
||||
maxDiff = None
|
||||
expected = {
|
||||
"8.4/main": {
|
||||
"port": 5432,
|
||||
"status": "online",
|
||||
"user": "postgres",
|
||||
"datadir": "/srv/8.4/main",
|
||||
"log": "/var/log/postgresql/postgresql-8.4-main.log",
|
||||
},
|
||||
"9.1/main": {
|
||||
"port": 5433,
|
||||
"status": "online",
|
||||
"user": "postgres",
|
||||
"datadir": "/srv/9.1/main",
|
||||
"log": "/var/log/postgresql/postgresql-9.1-main.log",
|
||||
},
|
||||
}
|
||||
assert deb_postgres._parse_pg_lscluster(stdout) == expected
|
||||
|
||||
|
||||
def test_cluster_list():
|
||||
with patch("salt.utils.path.which", Mock(return_value="/usr/bin/pg_lsclusters")):
|
||||
return_list = deb_postgres.cluster_list()
|
||||
assert (
|
||||
deb_postgres.__salt__["cmd.run_all"].call_args[0][0]
|
||||
== "/usr/bin/pg_lsclusters --no-header"
|
||||
)
|
||||
|
||||
return_dict = deb_postgres.cluster_list(verbose=True)
|
||||
assert isinstance(return_dict, dict)
|
||||
|
||||
|
||||
def test_cluster_exists():
|
||||
assert deb_postgres.cluster_exists("8.4")
|
||||
assert deb_postgres.cluster_exists("8.4", "main")
|
||||
assert not deb_postgres.cluster_exists("3.4", "main")
|
||||
|
||||
|
||||
def test_cluster_delete():
|
||||
with patch("salt.utils.path.which", Mock(return_value="/usr/bin/pg_dropcluster")):
|
||||
deb_postgres.cluster_remove("9.3", "main")
|
||||
assert (
|
||||
deb_postgres.__salt__["cmd.run_all"].call_args[0][0]
|
||||
== "/usr/bin/pg_dropcluster 9.3 main"
|
||||
)
|
||||
|
||||
deb_postgres.cluster_remove("9.3", "main", stop=True)
|
||||
assert (
|
||||
deb_postgres.__salt__["cmd.run_all"].call_args[0][0]
|
||||
== "/usr/bin/pg_dropcluster --stop 9.3 main"
|
||||
)
|
||||
|
||||
deb_postgres.cluster_remove(9.3, "main", stop=True)
|
||||
assert (
|
||||
deb_postgres.__salt__["cmd.run_all"].call_args[0][0]
|
||||
== "/usr/bin/pg_dropcluster --stop 9.3 main"
|
||||
)
|
||||
|
|
@ -1,184 +0,0 @@
|
|||
import salt.modules.deb_postgres as deb_postgres
|
||||
from tests.support.mixins import LoaderModuleMockMixin
|
||||
from tests.support.mock import Mock, patch
|
||||
from tests.support.unit import TestCase
|
||||
|
||||
LSCLUSTER = """\
|
||||
8.4 main 5432 online postgres /srv/8.4/main \
|
||||
/var/log/postgresql/postgresql-8.4-main.log
|
||||
9.1 main 5433 online postgres /srv/9.1/main \
|
||||
/var/log/postgresql/postgresql-9.1-main.log
|
||||
"""
|
||||
|
||||
|
||||
class PostgresClusterTestCase(TestCase, LoaderModuleMockMixin):
|
||||
def setup_loader_modules(self):
|
||||
self.cmd_run_all_mock = Mock(return_value={"stdout": LSCLUSTER})
|
||||
self.addCleanup(delattr, self, "cmd_run_all_mock")
|
||||
patcher = patch(
|
||||
"salt.utils.path.which", Mock(return_value="/usr/bin/pg_createcluster")
|
||||
)
|
||||
patcher.start()
|
||||
self.addCleanup(patcher.stop)
|
||||
return {
|
||||
deb_postgres: {
|
||||
"__salt__": {
|
||||
"config.option": Mock(),
|
||||
"cmd.run_all": self.cmd_run_all_mock,
|
||||
"file.chown": Mock(),
|
||||
"file.remove": Mock(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
def test_cluster_create(self):
|
||||
deb_postgres.cluster_create(
|
||||
"9.3",
|
||||
"main",
|
||||
port="5432",
|
||||
locale="fr_FR",
|
||||
encoding="UTF-8",
|
||||
datadir="/opt/postgresql",
|
||||
)
|
||||
cmdstr = (
|
||||
"/usr/bin/pg_createcluster "
|
||||
"--port 5432 --locale fr_FR --encoding UTF-8 "
|
||||
"--datadir /opt/postgresql "
|
||||
"9.3 main"
|
||||
)
|
||||
self.assertEqual(cmdstr, self.cmd_run_all_mock.call_args[0][0])
|
||||
|
||||
def test_cluster_create_with_initdb_options(self):
|
||||
deb_postgres.cluster_create(
|
||||
"11",
|
||||
"main",
|
||||
port="5432",
|
||||
locale="fr_FR",
|
||||
encoding="UTF-8",
|
||||
datadir="/opt/postgresql",
|
||||
allow_group_access=True,
|
||||
data_checksums=True,
|
||||
wal_segsize="32",
|
||||
)
|
||||
cmdstr = (
|
||||
"/usr/bin/pg_createcluster "
|
||||
"--port 5432 --locale fr_FR --encoding UTF-8 "
|
||||
"--datadir /opt/postgresql "
|
||||
"11 main "
|
||||
"-- "
|
||||
"--allow-group-access "
|
||||
"--data-checksums "
|
||||
"--wal-segsize 32"
|
||||
)
|
||||
self.assertEqual(cmdstr, self.cmd_run_all_mock.call_args[0][0])
|
||||
|
||||
def test_cluster_create_with_float(self):
|
||||
deb_postgres.cluster_create(
|
||||
9.3,
|
||||
"main",
|
||||
port="5432",
|
||||
locale="fr_FR",
|
||||
encoding="UTF-8",
|
||||
datadir="/opt/postgresql",
|
||||
)
|
||||
cmdstr = (
|
||||
"/usr/bin/pg_createcluster "
|
||||
"--port 5432 --locale fr_FR --encoding UTF-8 "
|
||||
"--datadir /opt/postgresql "
|
||||
"9.3 main"
|
||||
)
|
||||
self.assertEqual(cmdstr, self.cmd_run_all_mock.call_args[0][0])
|
||||
|
||||
|
||||
class PostgresLsClusterTestCase(TestCase, LoaderModuleMockMixin):
|
||||
def setup_loader_modules(self):
|
||||
self.cmd_run_all_mock = Mock(return_value={"stdout": LSCLUSTER})
|
||||
self.addCleanup(delattr, self, "cmd_run_all_mock")
|
||||
patcher = patch(
|
||||
"salt.utils.path.which", Mock(return_value="/usr/bin/pg_lsclusters")
|
||||
)
|
||||
patcher.start()
|
||||
self.addCleanup(patcher.stop)
|
||||
return {
|
||||
deb_postgres: {
|
||||
"__salt__": {
|
||||
"config.option": Mock(),
|
||||
"cmd.run_all": self.cmd_run_all_mock,
|
||||
"file.chown": Mock(),
|
||||
"file.remove": Mock(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
def test_parse_pg_lsclusters(self):
|
||||
stdout = LSCLUSTER
|
||||
self.maxDiff = None
|
||||
self.assertDictEqual(
|
||||
{
|
||||
"8.4/main": {
|
||||
"port": 5432,
|
||||
"status": "online",
|
||||
"user": "postgres",
|
||||
"datadir": "/srv/8.4/main",
|
||||
"log": "/var/log/postgresql/postgresql-8.4-main.log",
|
||||
},
|
||||
"9.1/main": {
|
||||
"port": 5433,
|
||||
"status": "online",
|
||||
"user": "postgres",
|
||||
"datadir": "/srv/9.1/main",
|
||||
"log": "/var/log/postgresql/postgresql-9.1-main.log",
|
||||
},
|
||||
},
|
||||
deb_postgres._parse_pg_lscluster(stdout),
|
||||
)
|
||||
|
||||
def test_cluster_list(self):
|
||||
return_list = deb_postgres.cluster_list()
|
||||
self.assertEqual(
|
||||
"/usr/bin/pg_lsclusters --no-header", self.cmd_run_all_mock.call_args[0][0]
|
||||
)
|
||||
return_dict = deb_postgres.cluster_list(verbose=True)
|
||||
self.assertIsInstance(return_dict, dict)
|
||||
|
||||
def test_cluster_exists(self):
|
||||
self.assertTrue(deb_postgres.cluster_exists("8.4") is True)
|
||||
self.assertTrue(deb_postgres.cluster_exists("8.4", "main") is True)
|
||||
self.assertFalse(deb_postgres.cluster_exists("3.4", "main"))
|
||||
|
||||
|
||||
class PostgresDeleteClusterTestCase(TestCase, LoaderModuleMockMixin):
|
||||
def setup_loader_modules(self):
|
||||
self.cmd_run_all_mock = Mock(return_value={"stdout": LSCLUSTER})
|
||||
self.addCleanup(delattr, self, "cmd_run_all_mock")
|
||||
patcher = patch(
|
||||
"salt.utils.path.which", Mock(return_value="/usr/bin/pg_dropcluster")
|
||||
)
|
||||
patcher.start()
|
||||
self.addCleanup(patcher.stop)
|
||||
return {
|
||||
deb_postgres: {
|
||||
"__salt__": {
|
||||
"config.option": Mock(),
|
||||
"cmd.run_all": self.cmd_run_all_mock,
|
||||
"file.chown": Mock(),
|
||||
"file.remove": Mock(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
def test_cluster_delete(self):
|
||||
deb_postgres.cluster_remove("9.3", "main")
|
||||
self.assertEqual(
|
||||
"/usr/bin/pg_dropcluster 9.3 main", self.cmd_run_all_mock.call_args[0][0]
|
||||
)
|
||||
deb_postgres.cluster_remove("9.3", "main", stop=True)
|
||||
self.assertEqual(
|
||||
"/usr/bin/pg_dropcluster --stop 9.3 main",
|
||||
self.cmd_run_all_mock.call_args[0][0],
|
||||
)
|
||||
deb_postgres.cluster_remove(9.3, "main", stop=True)
|
||||
self.assertEqual(
|
||||
"/usr/bin/pg_dropcluster --stop 9.3 main",
|
||||
self.cmd_run_all_mock.call_args[0][0],
|
||||
)
|
||||
Loading…
Add table
Reference in a new issue