salt/tests/unit/states/test_openstack_config.py
Erik Johnson 93ee5ee2b0
Fix all Sphinx warnings
Well, all but one, which we expect to see
2018-05-31 15:28:25 -05:00

99 lines
3.5 KiB
Python

# -*- coding: utf-8 -*-
'''
:codeauthor: 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)
from salt.exceptions import CommandExecutionError
# Import Salt Libs
import salt.states.openstack_config as openstack_config
@skipIf(NO_MOCK, NO_MOCK_REASON)
class OpenstackConfigTestCase(TestCase, LoaderModuleMockMixin):
'''
Test cases for salt.states.openstack_config
'''
def setup_loader_modules(self):
return {openstack_config: {'__opts__': {'test': False}}}
# 'present' function tests: 1
def test_present(self):
'''
Test to ensure a value is set in an OpenStack configuration file.
'''
name = 'salt'
filename = '/tmp/salt'
section = 'A'
value = 'SALT'
ret = {'name': name,
'result': False,
'comment': '',
'changes': {}}
mock_lst = MagicMock(side_effect=[value, CommandExecutionError, 'A'])
mock_t = MagicMock(return_value=True)
with patch.dict(openstack_config.__salt__,
{'openstack_config.get': mock_lst,
'openstack_config.set': mock_t}):
comt = ('The value is already set to the correct value')
ret.update({'comment': comt, 'result': True})
self.assertDictEqual(openstack_config.present(name, filename,
section, value), ret)
self.assertRaises(CommandExecutionError, openstack_config.present,
name, filename, section, value)
comt = ('The value has been updated')
ret.update({'comment': comt, 'changes': {'Value': 'Updated'}})
self.assertDictEqual(openstack_config.present(name, filename,
section, value), ret)
# 'absent' function tests: 1
def test_absent(self):
'''
Test to ensure a value is not set in an OpenStack configuration file.
'''
name = 'salt'
filename = '/tmp/salt'
section = 'A'
ret = {'name': name,
'result': False,
'comment': '',
'changes': {}}
mock_lst = MagicMock(side_effect=[CommandExecutionError
('parameter not found:'),
CommandExecutionError, 'A'])
mock_t = MagicMock(return_value=True)
with patch.dict(openstack_config.__salt__,
{'openstack_config.get': mock_lst,
'openstack_config.delete': mock_t}):
comt = ('The value is already absent')
ret.update({'comment': comt, 'result': True})
self.assertDictEqual(openstack_config.absent(name, filename,
section), ret)
self.assertRaises(CommandExecutionError, openstack_config.absent,
name, filename, section)
comt = ('The value has been deleted')
ret.update({'comment': comt, 'changes': {'Value': 'Deleted'}})
self.assertDictEqual(openstack_config.absent(name, filename,
section), ret)