salt/tests/unit/states/test_postgres_initdb.py
rallytime 3273bbdab7
Merge branch '2017.7' into '2018.3'
Conflicts:
  - doc/ref/configuration/master.rst
  - doc/ref/modules/all/index.rst
  - doc/topics/grains/index.rst
  - doc/topics/releases/2016.3.4.rst
  - doc/topics/spm/spm_formula.rst
  - doc/topics/tutorials/cron.rst
  - doc/topics/tutorials/index.rst
  - doc/topics/tutorials/stormpath.rst
  - salt/engines/slack.py
  - salt/log/handlers/fluent_mod.py
  - salt/modules/cyg.py
  - salt/modules/junos.py
  - salt/modules/namecheap_dns.py
  - salt/modules/namecheap_domains.py
  - salt/modules/namecheap_ns.py
  - salt/modules/namecheap_ssl.py
  - salt/modules/namecheap_users.py
  - salt/modules/reg.py
  - salt/modules/tomcat.py
  - salt/modules/vault.py
  - salt/modules/win_file.py
  - salt/modules/zpool.py
  - salt/output/highstate.py
  - salt/renderers/pass.py
  - salt/runners/cache.py
  - salt/states/boto_apigateway.py
  - salt/states/boto_iam.py
  - salt/states/boto_route53.py
  - salt/states/msteams.py
  - salt/states/reg.py
  - salt/states/win_iis.py
  - tests/integration/modules/test_cmdmod.py
  - tests/integration/states/test_user.py
  - tests/support/helpers.py
  - tests/unit/cloud/clouds/test_openstack.py
  - tests/unit/fileserver/test_gitfs.py
  - tests/unit/modules/test_junos.py
  - tests/unit/pillar/test_git.py
  - tests/unit/states/test_win_path.py
  - tests/unit/test_pillar.py
  - tests/unit/utils/test_format_call.py
  - tests/unit/utils/test_utils.py
  - tests/unit/utils/test_warnings.py
2018-06-01 14:54:12 -04:00

98 lines
3.4 KiB
Python

# -*- coding: utf-8 -*-
'''
:codeauthor: Andrew Colin Kissa <andrew@topdog.za.net>
'''
from __future__ import absolute_import, unicode_literals, print_function
from tests.support.mixins import LoaderModuleMockMixin
from tests.support.unit import skipIf, TestCase
from tests.support.mock import (
NO_MOCK,
NO_MOCK_REASON,
MagicMock,
patch
)
import salt.states.postgres_initdb as postgres_initdb
@skipIf(NO_MOCK, NO_MOCK_REASON)
class PostgresInitdbTestCase(TestCase, LoaderModuleMockMixin):
'''
Test cases for salt.states.postgres_initdb
'''
def setup_loader_modules(self):
return {postgres_initdb: {}}
def setUp(self):
'''
Setup data for the tests
'''
self.name = '/var/lib/psql/data'
self.ret = {
'name': self.name,
'changes': {},
'result': False,
'comment': ''}
self.mock_true = MagicMock(return_value=True)
self.mock_false = MagicMock(return_value=False)
def tearDown(self):
del self.ret
del self.mock_true
del self.mock_false
def test_present_existing(self):
'''
Test existing data directory handled correctly
'''
with patch.dict(postgres_initdb.__salt__,
{'postgres.datadir_exists': self.mock_true}):
_comt = 'Postgres data directory {0} is already present'\
.format(self.name)
self.ret.update({'comment': _comt, 'result': True})
self.assertDictEqual(postgres_initdb.present(self.name), self.ret)
def test_present_non_existing_pass(self):
'''
Test non existing data directory ok
'''
with patch.dict(postgres_initdb.__salt__,
{'postgres.datadir_exists': self.mock_false,
'postgres.datadir_init': self.mock_true}):
with patch.dict(postgres_initdb.__opts__, {'test': True}):
_comt = 'Postgres data directory {0} is set to be initialized'\
.format(self.name)
self.ret.update({'comment': _comt, 'result': None})
self.assertDictEqual(
postgres_initdb.present(self.name), self.ret)
with patch.dict(postgres_initdb.__opts__, {'test': False}):
_comt = 'Postgres data directory {0} has been initialized'\
.format(self.name)
_changes = {self.name: 'Present'}
self.ret.update({
'comment': _comt,
'result': True,
'changes': _changes})
self.assertDictEqual(
postgres_initdb.present(self.name), self.ret)
def test_present_non_existing_fail(self):
'''
Test non existing data directory fail
'''
with patch.dict(postgres_initdb.__salt__,
{'postgres.datadir_exists': self.mock_false,
'postgres.datadir_init': self.mock_false}):
with patch.dict(postgres_initdb.__opts__, {'test': False}):
_comt = 'Postgres data directory {0} initialization failed'\
.format(self.name)
self.ret.update({
'comment': _comt,
'result': False
})
self.assertDictEqual(
postgres_initdb.present(self.name), self.ret)