salt/tests/unit/states/layman_test.py
2015-05-12 18:05:35 +05:30

84 lines
2.3 KiB
Python

# -*- 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 layman
layman.__salt__ = {}
layman.__opts__ = {}
@skipIf(NO_MOCK, NO_MOCK_REASON)
class LaymanTestCase(TestCase):
'''
Test cases for salt.states.layman
'''
# 'present' function tests: 1
def test_present(self):
'''
Test to verify that the overlay is present.
'''
name = 'sunrise'
ret = {'name': name,
'result': True,
'comment': '',
'changes': {}}
mock = MagicMock(side_effect=[[name], []])
with patch.dict(layman.__salt__, {'layman.list_local': mock}):
comt = ('Overlay {0} already present'.format(name))
ret.update({'comment': comt})
self.assertDictEqual(layman.present(name), ret)
with patch.dict(layman.__opts__, {'test': True}):
comt = ('Overlay {0} is set to be added'.format(name))
ret.update({'comment': comt, 'result': None})
self.assertDictEqual(layman.present(name), ret)
# 'absent' function tests: 1
def test_absent(self):
'''
Test to verify that the overlay is absent.
'''
name = 'sunrise'
ret = {'name': name,
'result': True,
'comment': '',
'changes': {}}
mock = MagicMock(side_effect=[[], [name]])
with patch.dict(layman.__salt__, {'layman.list_local': mock}):
comt = ('Overlay {0} already absent'.format(name))
ret.update({'comment': comt})
self.assertDictEqual(layman.absent(name), ret)
with patch.dict(layman.__opts__, {'test': True}):
comt = ('Overlay {0} is set to be deleted'.format(name))
ret.update({'comment': comt, 'result': None})
self.assertDictEqual(layman.absent(name), ret)
if __name__ == '__main__':
from integration import run_tests
run_tests(LaymanTestCase, needs_daemon=False)