mirror of
https://github.com/valitydev/salt.git
synced 2024-11-07 17:09:03 +00:00
68 lines
2.3 KiB
Python
68 lines
2.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
'''
|
|
:codeauthor: Nicole Thomas <nicole@saltstack.com>
|
|
'''
|
|
# Import Python libs
|
|
from __future__ import absolute_import
|
|
import salt.utils
|
|
|
|
# Import Salt Testing Libs
|
|
from tests.support.case import ShellCase
|
|
|
|
|
|
class BatchTest(ShellCase):
|
|
'''
|
|
Integration tests for the salt.cli.batch module
|
|
'''
|
|
|
|
def test_batch_run(self):
|
|
'''
|
|
Tests executing a simple batch command to help catch regressions
|
|
'''
|
|
ret = 'Executing run on [\'sub_minion\']'
|
|
if salt.utils.is_windows():
|
|
cmd = self.run_salt('"*minion" test.echo "batch testing" -b 50%')
|
|
else:
|
|
cmd = self.run_salt('\'*minion\' test.echo \'batch testing\' -b 50%')
|
|
self.assertIn(ret, cmd)
|
|
|
|
def test_batch_run_number(self):
|
|
'''
|
|
Tests executing a simple batch command using a number division instead of
|
|
a percentage with full batch CLI call.
|
|
'''
|
|
ret = "Executing run on ['minion', 'sub_minion']"
|
|
if salt.utils.is_windows():
|
|
cmd = self.run_salt('"*minion" test.ping --batch-size 2')
|
|
else:
|
|
cmd = self.run_salt('\'*minion\' test.ping --batch-size 2')
|
|
self.assertIn(ret, cmd)
|
|
|
|
def test_batch_run_grains_targeting(self):
|
|
'''
|
|
Tests executing a batch command using a percentage divisor as well as grains
|
|
targeting.
|
|
'''
|
|
os_grain = ''
|
|
sub_min_ret = "Executing run on ['sub_minion']"
|
|
min_ret = "Executing run on ['minion']"
|
|
|
|
for item in self.run_salt('minion grains.get os'):
|
|
if item != 'minion':
|
|
os_grain = item
|
|
|
|
os_grain = os_grain.strip()
|
|
if salt.utils.is_windows():
|
|
cmd = self.run_salt('-C "G@os:{0} and not localhost" -b 25% test.ping'.format(os_grain))
|
|
else:
|
|
cmd = self.run_salt('-C \'G@os:{0} and not localhost\' -b 25% test.ping'.format(os_grain))
|
|
self.assertIn(sub_min_ret, cmd)
|
|
self.assertIn(min_ret, cmd)
|
|
|
|
def test_batch_exit_code(self):
|
|
'''
|
|
Test that a failed state returns a non-zero exit code in batch mode
|
|
'''
|
|
cmd = self.run_salt(' "*" state.single test.fail_without_changes name=test_me -b 25%', with_retcode=True)
|
|
self.assertEqual(cmd[-1], 2)
|