mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 08:58:59 +00:00
adding states/alternatives unit test case
This commit is contained in:
Jayesh Kariya
parent
d14710acce
commit
283899e939
205
tests/unit/states/alternatives_test.py
Normal file
205
tests/unit/states/alternatives_test.py
Normal file
@ -0,0 +1,205 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
'''
|
||||
:codeauthor: :email:`Jayesh Kariya <jayeshk@saltstack.com>`
|
||||
'''
|
||||
# Import Python libs
|
||||
from __future__ import absolute_import
|
||||
|
||||
# Import Salt Testing Libs
|
||||
from salttesting import skipIf, TestCase
|
||||
from salttesting.mock import (
|
||||
NO_MOCK,
|
||||
NO_MOCK_REASON,
|
||||
MagicMock,
|
||||
patch
|
||||
)
|
||||
|
||||
from salttesting.helpers import ensure_in_syspath
|
||||
|
||||
ensure_in_syspath('../../')
|
||||
|
||||
# Import Salt Libs
|
||||
from salt.states import alternatives
|
||||
|
||||
alternatives.__opts__ = {}
|
||||
alternatives.__salt__ = {}
|
||||
|
||||
|
||||
@skipIf(NO_MOCK, NO_MOCK_REASON)
|
||||
class AlternativesTestCase(TestCase):
|
||||
'''
|
||||
Test cases for salt.states.alternatives
|
||||
'''
|
||||
# 'install' function tests: 1
|
||||
|
||||
def test_install(self):
|
||||
'''
|
||||
Test to install new alternative for defined <name>
|
||||
'''
|
||||
name = 'pager'
|
||||
link = '/usr/bin/pager'
|
||||
path = '/usr/bin/less'
|
||||
priority = 5
|
||||
|
||||
ret = {'name': name,
|
||||
'link': link,
|
||||
'path': path,
|
||||
'priority': priority,
|
||||
'result': None,
|
||||
'changes': {},
|
||||
'comment': ''}
|
||||
|
||||
mock = MagicMock(side_effect=[True, False, False])
|
||||
mock_bool = MagicMock(return_value=True)
|
||||
with patch.dict(alternatives.__salt__,
|
||||
{'alternatives.check_installed': mock,
|
||||
'alternatives.install': mock_bool}):
|
||||
comt = ('Alternatives for {0} is already set to {1}'
|
||||
).format(name, path)
|
||||
ret.update({'comment': comt, 'result': True})
|
||||
self.assertDictEqual(alternatives.install(name, link, path,
|
||||
priority), ret)
|
||||
|
||||
comt = (('Alternative will be set for {0} to {1} with priority {2}'
|
||||
).format(name, path, priority))
|
||||
ret.update({'comment': comt, 'result': None})
|
||||
with patch.dict(alternatives.__opts__, {'test': True}):
|
||||
self.assertDictEqual(alternatives.install(name, link, path,
|
||||
priority), ret)
|
||||
|
||||
comt = ('Setting alternative for {0} to {1} with priority {2}'
|
||||
).format(name, path, priority)
|
||||
ret.update({'comment': comt, 'result': True,
|
||||
'changes': {'name': name, 'link': link, 'path': path,
|
||||
'priority': priority}})
|
||||
with patch.dict(alternatives.__opts__, {'test': False}):
|
||||
self.assertDictEqual(alternatives.install(name, link, path,
|
||||
priority), ret)
|
||||
|
||||
# 'remove' function tests: 1
|
||||
|
||||
def test_remove(self):
|
||||
'''
|
||||
Test to removes installed alternative for defined <name> and <path>
|
||||
or fallback to default alternative, if some defined before.
|
||||
'''
|
||||
name = 'pager'
|
||||
path = '/usr/bin/less'
|
||||
|
||||
ret = {'name': name,
|
||||
'path': path,
|
||||
'result': None,
|
||||
'changes': {},
|
||||
'comment': ''}
|
||||
|
||||
mock = MagicMock(side_effect=[True, True, True, False, False])
|
||||
mock_bool = MagicMock(return_value=True)
|
||||
mock_bol = MagicMock(side_effect=[False, True, True, False])
|
||||
with patch.dict(alternatives.__salt__,
|
||||
{'alternatives.check_installed': mock,
|
||||
'alternatives.show_current': mock_bol,
|
||||
'alternatives.remove': mock_bool}):
|
||||
comt = ('Alternative for {0} will be removed'.format(name))
|
||||
ret.update({'comment': comt})
|
||||
with patch.dict(alternatives.__opts__, {'test': True}):
|
||||
self.assertDictEqual(alternatives.remove(name, path), ret)
|
||||
|
||||
comt = ('Alternative for {0} removed'.format(name))
|
||||
ret.update({'comment': comt, 'result': True})
|
||||
with patch.dict(alternatives.__opts__, {'test': False}):
|
||||
self.assertDictEqual(alternatives.remove(name, path), ret)
|
||||
|
||||
comt = ('Alternative for pager removed. Falling back to path True')
|
||||
ret.update({'comment': comt, 'result': True,
|
||||
'changes': {'path': True}})
|
||||
with patch.dict(alternatives.__opts__, {'test': False}):
|
||||
self.assertDictEqual(alternatives.remove(name, path), ret)
|
||||
|
||||
comt = ('Alternative for {0} is set to it\'s default path True'
|
||||
).format(name)
|
||||
ret.update({'comment': comt, 'result': True, 'changes': {}})
|
||||
self.assertDictEqual(alternatives.remove(name, path), ret)
|
||||
|
||||
comt = ('Alternative for {0} doesn\'t exist').format(name)
|
||||
ret.update({'comment': comt, 'result': False})
|
||||
self.assertDictEqual(alternatives.remove(name, path), ret)
|
||||
|
||||
# 'auto' function tests: 1
|
||||
|
||||
def test_auto(self):
|
||||
'''
|
||||
Test to instruct alternatives to use the highest priority
|
||||
path for <name>
|
||||
'''
|
||||
name = 'pager'
|
||||
|
||||
ret = {'name': name,
|
||||
'result': True,
|
||||
'changes': {},
|
||||
'comment': ''}
|
||||
|
||||
mock = MagicMock(side_effect=[' auto mode', ' ', ' '])
|
||||
mock_auto = MagicMock(return_value=True)
|
||||
with patch.dict(alternatives.__salt__,
|
||||
{'alternatives.display': mock,
|
||||
'alternatives.auto': mock_auto}):
|
||||
comt = ('{0} already in auto mode'.format(name))
|
||||
ret.update({'comment': comt})
|
||||
self.assertDictEqual(alternatives.auto(name), ret)
|
||||
|
||||
comt = ('{0} will be put in auto mode'.format(name))
|
||||
ret.update({'comment': comt, 'result': None})
|
||||
with patch.dict(alternatives.__opts__, {'test': True}):
|
||||
self.assertDictEqual(alternatives.auto(name), ret)
|
||||
|
||||
ret.update({'comment': '', 'result': True,
|
||||
'changes': {'result': True}})
|
||||
with patch.dict(alternatives.__opts__, {'test': False}):
|
||||
self.assertDictEqual(alternatives.auto(name), ret)
|
||||
|
||||
# 'set_' function tests: 1
|
||||
|
||||
def test_set(self):
|
||||
'''
|
||||
Test to sets alternative for <name> to <path>, if <path> is defined
|
||||
as an alternative for <name>.
|
||||
'''
|
||||
name = 'pager'
|
||||
path = '/usr/bin/less'
|
||||
|
||||
ret = {'name': name,
|
||||
'path': path,
|
||||
'result': True,
|
||||
'changes': {},
|
||||
'comment': ''}
|
||||
|
||||
mock = MagicMock(side_effect=[path, path, ''])
|
||||
mock_bool = MagicMock(return_value=True)
|
||||
mock_bol = MagicMock(side_effect=[path, False, False, False, False])
|
||||
with patch.dict(alternatives.__salt__,
|
||||
{'alternatives.display': mock,
|
||||
'alternatives.show_current': mock_bol,
|
||||
'alternatives.set': mock_bool}):
|
||||
comt = ('Alternative for {0} already set to {1}'.format(name, path))
|
||||
ret.update({'comment': comt})
|
||||
self.assertDictEqual(alternatives.set_(name, path), ret)
|
||||
|
||||
comt = ('Alternative for {0} will be set to path False'
|
||||
).format(name)
|
||||
ret.update({'comment': comt, 'result': None})
|
||||
with patch.dict(alternatives.__opts__, {'test': True}):
|
||||
self.assertDictEqual(alternatives.set_(name, path), ret)
|
||||
|
||||
comt = 'Alternative for {0} not updated'.format(name)
|
||||
ret.update({'comment': comt, 'result': True})
|
||||
with patch.dict(alternatives.__opts__, {'test': False}):
|
||||
self.assertDictEqual(alternatives.set_(name, path), ret)
|
||||
|
||||
comt = ('Alternative {0} for {1} doesn\'t exist').format(path, name)
|
||||
ret.update({'comment': comt, 'result': False})
|
||||
self.assertDictEqual(alternatives.set_(name, path), ret)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
from integration import run_tests
|
||||
run_tests(AlternativesTestCase, needs_daemon=False)
|
||||
Loading…
Reference in New Issue
Block a user