valitydev/salt
14
0
Fork 0
mirror of https://github.com/valitydev/salt.git synced 2024-11-07 08:58:59 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add postgres_privileges state module tests

Browse Source
This commit is contained in:
Andrew Colin Kissa 2015-12-29 12:57:21 +02:00
parent 0a0b3985c1
commit 7be5eebfc7
1 changed files with 132 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

135
tests/unit/states/postgres_privileges_test.py
View File

@ -31,11 +31,140 @@ class PostgresPrivilegesTestCase(TestCase):
'''
postgres_privileges.__opts__ = {}
postgres_privileges.__salt__ = {}
self.table_name = 'awl'
self.group_name = 'admins'
self.name = 'baruwa'
self.ret = {'name': self.name,
'changes': {},
'result': False,
'comment': ''}
self.mock_true = MagicMock(return_value=True)
self.mock_false = MagicMock(return_value=False)
def test_present_existing(self):
def test_present_table(self):
'''
Test present, privilege(s) are already granted
Test present
'''
pass
with patch.dict(postgres_privileges.__salt__,
{'postgres.has_privileges': self.mock_true}):
comt = 'The requested privilege(s) are already set'
self.ret.update({'comment': comt, 'result': True})
self.assertDictEqual(
postgres_privileges.present(
self.name,
self.table_name,
'table'),
self.ret)
with patch.dict(postgres_privileges.__salt__,
{'postgres.has_privileges': self.mock_false,
'postgres.privileges_grant': self.mock_true}):
with patch.dict(postgres_privileges.__opts__, {'test': True}):
comt = ('The privilege(s): {0} are'
' set to be granted to {1}').format('ALL', self.name)
self.ret.update({'comment': comt, 'result': None})
self.assertDictEqual(
postgres_privileges.present(self.name,
self.table_name, 'table', privileges=['ALL']), self.ret)
with patch.dict(postgres_privileges.__opts__, {'test': False}):
comt = ('The privilege(s): {0} have '
'been granted to {1}').format('ALL', self.name)
self.ret.update({'comment': comt,
'result': True,
'changes': {'baruwa': 'Present'}})
self.assertDictEqual(
postgres_privileges.present(self.name,
self.table_name, 'table', privileges=['ALL']), self.ret)
def test_present_group(self):
'''
Test present group
'''
with patch.dict(postgres_privileges.__salt__,
{'postgres.has_privileges': self.mock_false,
'postgres.privileges_grant': self.mock_true}):
with patch.dict(postgres_privileges.__opts__, {'test': True}):
comt = ('The privilege(s): {0} are'
' set to be granted to {1}').format(self.group_name,
self.name)
self.ret.update({'comment': comt, 'result': None})
self.assertDictEqual(
postgres_privileges.present(self.name,
self.group_name, 'group'), self.ret)
with patch.dict(postgres_privileges.__opts__, {'test': False}):
comt = ('The privilege(s): {0} have '
'been granted to {1}').format(self.group_name,
self.name)
self.ret.update({'comment': comt,
'result': True,
'changes': {'baruwa': 'Present'}})
self.assertDictEqual(
postgres_privileges.present(self.name,
self.group_name, 'group'), self.ret)
def test_absent_table(self):
'''
Test absent
'''
with patch.dict(postgres_privileges.__salt__,
{'postgres.has_privileges': self.mock_false}):
with patch.dict(postgres_privileges.__opts__, {'test': True}):
comt = ('The requested privilege(s)'
' are not set so cannot be revoked')
self.ret.update({'comment': comt, 'result': True})
self.assertDictEqual(
postgres_privileges.absent(
self.name,
self.table_name,
'table'),
self.ret)
with patch.dict(postgres_privileges.__salt__,
{'postgres.has_privileges': self.mock_true,
'postgres.privileges_revoke': self.mock_true}):
with patch.dict(postgres_privileges.__opts__, {'test': True}):
comt = ('The privilege(s): {0} are'
' set to be revoked from {1}').format('ALL', self.name)
self.ret.update({'comment': comt, 'result': None})
self.assertDictEqual(
postgres_privileges.absent(self.name,
self.table_name, 'table', privileges=['ALL']), self.ret)
with patch.dict(postgres_privileges.__opts__, {'test': False}):
comt = ('The privilege(s): {0} have '
'been revoked from {1}').format('ALL', self.name)
self.ret.update({'comment': comt,
'result': True,
'changes': {'baruwa': 'Absent'}})
self.assertDictEqual(
postgres_privileges.absent(self.name,
self.table_name, 'table', privileges=['ALL']), self.ret)
def test_absent_group(self):
'''
Test absent group
'''
with patch.dict(postgres_privileges.__salt__,
{'postgres.has_privileges': self.mock_true,
'postgres.privileges_revoke': self.mock_true}):
with patch.dict(postgres_privileges.__opts__, {'test': True}):
comt = ('The privilege(s): {0} are'
' set to be revoked from {1}').format(self.group_name,
self.name)
self.ret.update({'comment': comt, 'result': None})
self.assertDictEqual(
postgres_privileges.absent(self.name,
self.group_name, 'group'), self.ret)
with patch.dict(postgres_privileges.__opts__, {'test': False}):
comt = ('The privilege(s): {0} have '
'been revoked from {1}').format(self.group_name,
self.name)
self.ret.update({'comment': comt,
'result': True,
'changes': {'baruwa': 'Absent'}})
self.assertDictEqual(
postgres_privileges.absent(self.name,
self.group_name, 'group'), self.ret)