mirror of
https://github.com/saltstack/salt.git
synced 2025-04-17 10:10:20 +00:00
Add new option minions_state_test to enable test on master side
This commit is contained in:
Ch3LL
parent
1463e036d7
commit
499541a8e8
4 changed files with 187 additions and 2 deletions
|
|
@ -4763,7 +4763,7 @@ def check_file_meta(
|
|||
if mode is not None and mode != smode:
|
||||
changes['mode'] = mode
|
||||
|
||||
if lsattr_cmd:
|
||||
if lsattr_cmd and attrs:
|
||||
diff_attrs = _cmp_attrs(name, attrs)
|
||||
if (
|
||||
attrs is not None and
|
||||
|
|
|
|||
|
|
@ -429,6 +429,8 @@ def _get_test_value(test=None, **kwargs):
|
|||
if test is None:
|
||||
if salt.utils.args.test_mode(test=test, **kwargs):
|
||||
ret = True
|
||||
elif __salt__['config.get']('minions_state_test') is True:
|
||||
ret = True
|
||||
else:
|
||||
ret = __opts__.get('test', None)
|
||||
else:
|
||||
|
|
|
|||
|
|
@ -11,7 +11,7 @@ import time
|
|||
# Import Salt Testing libs
|
||||
from tests.support.case import ModuleCase
|
||||
from tests.support.unit import skipIf
|
||||
from tests.support.paths import TMP
|
||||
from tests.support.paths import TMP, TMP_PILLAR_TREE
|
||||
from tests.support.mixins import SaltReturnAssertsMixin
|
||||
|
||||
# Import Salt libs
|
||||
|
|
@ -1634,7 +1634,104 @@ class StateModuleTest(ModuleCase, SaltReturnAssertsMixin):
|
|||
self.assertTrue(state_run['file_|-test_file_|-/tmp/nonbase_env_|-managed']['result'])
|
||||
self.assertTrue(os.path.isfile('/tmp/nonbase_env'))
|
||||
|
||||
def _add_runtime_pillar(self, pillar):
|
||||
'''
|
||||
helper class to add pillar data at runtime
|
||||
'''
|
||||
import yaml
|
||||
with salt.utils.files.fopen(os.path.join(TMP_PILLAR_TREE,
|
||||
'pillar.sls'), 'w') as fp:
|
||||
fp.write(yaml.dump(pillar))
|
||||
|
||||
with salt.utils.files.fopen(os.path.join(TMP_PILLAR_TREE, 'top.sls'), 'w') as fp:
|
||||
fp.write(textwrap.dedent('''\
|
||||
base:
|
||||
'*':
|
||||
- pillar
|
||||
'''))
|
||||
|
||||
self.run_function('saltutil.refresh_pillar')
|
||||
self.run_function('test.sleep', [5])
|
||||
|
||||
def test_state_sls_id_minions_state_test(self):
|
||||
'''
|
||||
test state.sls_id when minions_state_test is set
|
||||
to true in pillar data
|
||||
'''
|
||||
self._add_runtime_pillar(pillar={'minions_state_test': True})
|
||||
ret = self.run_function('state.sls', ['core'])
|
||||
|
||||
for key, val in ret.items():
|
||||
self.assertEqual(val['comment'], 'The file /tmp/salt-tests-tmpdir/testfile is set to be changed')
|
||||
self.assertEqual(val['changes'], {})
|
||||
|
||||
def test_state_sls_id_minions_state_test_post_run(self):
|
||||
'''
|
||||
test state.sls_id when minions_state_test is set to
|
||||
true post the state already being run previously
|
||||
'''
|
||||
ret = self.run_function('state.sls', ['core'])
|
||||
for key, val in ret.items():
|
||||
self.assertEqual(val['comment'], 'File /tmp/salt-tests-tmpdir/testfile updated')
|
||||
self.assertEqual(val['changes']['diff'], 'New file')
|
||||
|
||||
self._add_runtime_pillar(pillar={'minion_state_test': True})
|
||||
ret = self.run_function('state.sls', ['core'])
|
||||
|
||||
for key, val in ret.items():
|
||||
self.assertEqual(val['comment'], 'File /tmp/salt-tests-tmpdir/testfile is in the correct state')
|
||||
self.assertEqual(val['changes'], {})
|
||||
|
||||
def test_state_sls_id_test_true(self):
|
||||
'''
|
||||
test state.sls_id when test=True is passed as arg
|
||||
'''
|
||||
ret = self.run_function('state.sls', ['core'], test=True)
|
||||
for key, val in ret.items():
|
||||
self.assertEqual(val['comment'], 'The file /tmp/salt-tests-tmpdir/testfile is set to be changed')
|
||||
self.assertEqual(val['changes'], {})
|
||||
|
||||
def test_state_sls_id_test_true_post_run(self):
|
||||
'''
|
||||
test state.sls_id when test is set to true as an
|
||||
arg post the state already being run previously
|
||||
'''
|
||||
ret = self.run_function('state.sls', ['core'])
|
||||
for key, val in ret.items():
|
||||
self.assertEqual(val['comment'], 'File /tmp/salt-tests-tmpdir/testfile updated')
|
||||
self.assertEqual(val['changes']['diff'], 'New file')
|
||||
|
||||
ret = self.run_function('state.sls', ['core'], test=True)
|
||||
|
||||
for key, val in ret.items():
|
||||
self.assertEqual(val['comment'], 'The file /tmp/salt-tests-tmpdir/testfile is in the correct state')
|
||||
self.assertEqual(val['changes'], {})
|
||||
|
||||
def test_state_sls_id_test_false_pillar_true(self):
|
||||
'''
|
||||
test state.sls_id when test is set to false as an
|
||||
arg and minion_state_test is set to True. Should
|
||||
return test=False.
|
||||
'''
|
||||
self._add_runtime_pillar(pillar={'minion_state_test': True})
|
||||
ret = self.run_function('state.sls', ['core'], test=False)
|
||||
|
||||
for key, val in ret.items():
|
||||
self.assertEqual(val['comment'], 'File /tmp/salt-tests-tmpdir/testfile updated')
|
||||
self.assertEqual(val['changes']['diff'], 'New file')
|
||||
|
||||
def tearDown(self):
|
||||
nonbase_file = '/tmp/nonbase_env'
|
||||
if os.path.isfile(nonbase_file):
|
||||
os.remove(nonbase_file)
|
||||
|
||||
# remove old pillar data
|
||||
for filename in os.listdir(TMP_PILLAR_TREE):
|
||||
os.remove(os.path.join(TMP_PILLAR_TREE, filename))
|
||||
self.run_function('saltutil.refresh_pillar')
|
||||
self.run_function('test.sleep', [5])
|
||||
|
||||
# remove testfile added in core.sls state file
|
||||
state_file = os.path.join(TMP, 'testfile')
|
||||
if os.path.isfile(state_file):
|
||||
os.remove(state_file)
|
||||
|
|
|
|||
|
|
@ -26,6 +26,7 @@ import salt.utils.odict
|
|||
import salt.utils.platform
|
||||
import salt.modules.state as state
|
||||
from salt.exceptions import CommandExecutionError, SaltInvocationError
|
||||
import salt.modules.config as config
|
||||
from salt.ext import six
|
||||
|
||||
|
||||
|
|
@ -366,7 +367,15 @@ class StateTestCase(TestCase, LoaderModuleMockMixin):
|
|||
'__cli': 'salt',
|
||||
},
|
||||
'__utils__': utils,
|
||||
'__salt__': {
|
||||
'config.get': config.get,
|
||||
}
|
||||
},
|
||||
config: {
|
||||
'__opts__': {},
|
||||
'__pillar__': {},
|
||||
},
|
||||
|
||||
}
|
||||
|
||||
def test_running(self):
|
||||
|
|
@ -939,6 +948,83 @@ class StateTestCase(TestCase, LoaderModuleMockMixin):
|
|||
mock):
|
||||
self.sub_test_sls()
|
||||
|
||||
def test_get_test_value(self):
|
||||
'''
|
||||
Test _get_test_value when opts contains different values
|
||||
'''
|
||||
# test when minion_state_test is True in __opts__
|
||||
test_arg = 'minions_state_test'
|
||||
with patch.dict(config.__opts__, {test_arg: True}):
|
||||
with patch.dict(state.__salt__, {'config.option':
|
||||
config.get(test_arg)}):
|
||||
self.assertTrue(state._get_test_value(test=None),
|
||||
msg='Failure when {0} is True in __opts__'.format(test_arg))
|
||||
|
||||
# test when minions_state_test is blah in __opts__
|
||||
with patch.dict(config.__opts__, {test_arg: 'blah'}):
|
||||
with patch.dict(state.__salt__, {'config.option':
|
||||
config.get(test_arg)}):
|
||||
self.assertFalse(state._get_test_value(test=None),
|
||||
msg='Failure when {0} is blah in __opts__'.format(test_arg))
|
||||
|
||||
# test when minions_state_test is False in __opts__
|
||||
with patch.dict(config.__opts__, {test_arg: False}):
|
||||
with patch.dict(state.__salt__, {'config.option':
|
||||
config.get(test_arg)}):
|
||||
self.assertFalse(state._get_test_value(test=None),
|
||||
msg='Failure when {0} is False in __opts__'.format(test_arg))
|
||||
|
||||
# test when minions_state_test does not exist in __opts__
|
||||
with patch.dict(config.__opts__, {}):
|
||||
with patch.dict(state.__salt__, {'config.option':
|
||||
config.get(test_arg)}):
|
||||
self.assertFalse(state._get_test_value(test=None),
|
||||
msg='Failure when {0} does not exist in __opts__'.format(test_arg))
|
||||
|
||||
# test when minions_state_test is None in __opts__
|
||||
with patch.dict(config.__opts__, {test_arg: None}):
|
||||
with patch.dict(state.__salt__, {'config.option':
|
||||
config.get(test_arg)}):
|
||||
self.assertEqual(state._get_test_value(test=None), None,
|
||||
msg='Failure when {0} is None in __opts__'.format(test_arg))
|
||||
|
||||
# test when minions_state_test is True in __pillar__
|
||||
with patch.dict(config.__pillar__, {test_arg: True}):
|
||||
with patch.dict(state.__salt__, {'config.option':
|
||||
config.get(test_arg)}):
|
||||
self.assertTrue(state._get_test_value(test=None),
|
||||
msg='Failure when {0} is True in __pillar__'.format(test_arg))
|
||||
|
||||
# test when minions_state_test is True in master __pillar__
|
||||
with patch.dict(config.__pillar__, {'master': {test_arg: True}}):
|
||||
with patch.dict(state.__salt__, {'config.option':
|
||||
config.get(test_arg)}):
|
||||
self.assertTrue(state._get_test_value(test=None),
|
||||
msg='Failure when {0} is True in master __pillar__'.format(test_arg))
|
||||
|
||||
# test when minions_state_test is True in __opts__
|
||||
# and test is False in opts
|
||||
with patch.dict(config.__opts__, {test_arg: True}):
|
||||
with patch.dict(state.__opts__, {'test': False}):
|
||||
with patch.dict(state.__salt__, {'config.option':
|
||||
config.get(test_arg)}):
|
||||
self.assertTrue(state._get_test_value(test=None),
|
||||
msg='Failure when {0} is True in __opts__ and test is False in __opts__'.format(test_arg))
|
||||
|
||||
# test when test is True in __opts__
|
||||
with patch.dict(state.__opts__, {'test': True}):
|
||||
with patch.dict(state.__salt__, {'config.option':
|
||||
config.get(test_arg)}):
|
||||
self.assertTrue(state._get_test_value(test=None),
|
||||
msg='Failure when test is True in __opts__')
|
||||
|
||||
# test when __opts__ is empty and test=True is passed to method
|
||||
with patch.dict(state.__opts__, {}):
|
||||
with patch.dict(state.__salt__, {'config.option':
|
||||
config.get(test_arg)}):
|
||||
self.assertTrue(state._get_test_value(test=True),
|
||||
msg='Failure when test is True as arg')
|
||||
|
||||
def sub_test_sls(self):
|
||||
'''
|
||||
Sub function of test_sls
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue