salt/tests/unit/states/test_eselect.py
2017-03-28 19:04:01 +01:00

49 lines
1.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 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 Libs
import salt.states.eselect as eselect
@skipIf(NO_MOCK, NO_MOCK_REASON)
class EselectTestCase(TestCase, LoaderModuleMockMixin):
'''
Test cases for salt.states.eselect
'''
def setup_loader_modules(self):
return {eselect: {}}
# 'set_' function tests: 1
def test_set_(self):
'''
Test to verify that the given module is set to the given target
'''
name = 'myeselect'
target = 'hardened/linux/amd64'
ret = {'name': name,
'result': True,
'comment': '',
'changes': {}}
mock = MagicMock(return_value=target)
with patch.dict(eselect.__salt__, {'eselect.get_current_target': mock}):
comt = ('Target \'{0}\' is already set on \'{1}\' module.'
.format(target, name))
ret.update({'comment': comt})
self.assertDictEqual(eselect.set_(name, target), ret)