mirror of
https://github.com/ansible-collections/community.general.git
synced 2025-10-23 04:24:00 -07:00
FEATURE: adding variable serial batches
This feature changes the scalar value of `serial:` to a list, which allows users to specify a list of values, so batches can be ramped up (commonly called "canary" setups): - hosts: all serial: [1, 5, 10, "100%"] tasks: ...
This commit is contained in:
parent
3a83333ef2
commit
159aa26b36
8 changed files with 176 additions and 25 deletions
|
@ -25,10 +25,11 @@ from ansible.compat.tests.mock import patch, MagicMock
|
|||
from ansible.errors import AnsibleError, AnsibleParserError
|
||||
from ansible.executor.playbook_executor import PlaybookExecutor
|
||||
from ansible.playbook import Playbook
|
||||
from ansible.template import Templar
|
||||
|
||||
from units.mock.loader import DictDataLoader
|
||||
|
||||
class TestPlayIterator(unittest.TestCase):
|
||||
class TestPlaybookExecutor(unittest.TestCase):
|
||||
|
||||
def setUp(self):
|
||||
pass
|
||||
|
@ -58,6 +59,20 @@ class TestPlayIterator(unittest.TestCase):
|
|||
tasks:
|
||||
- debug: var=inventory_hostname
|
||||
''',
|
||||
'serial_list.yml': '''
|
||||
- hosts: all
|
||||
gather_facts: no
|
||||
serial: [1, 2, 3]
|
||||
tasks:
|
||||
- debug: var=inventory_hostname
|
||||
''',
|
||||
'serial_list_mixed.yml': '''
|
||||
- hosts: all
|
||||
gather_facts: no
|
||||
serial: [1, "20%", -1]
|
||||
tasks:
|
||||
- debug: var=inventory_hostname
|
||||
''',
|
||||
})
|
||||
|
||||
mock_inventory = MagicMock()
|
||||
|
@ -68,8 +83,10 @@ class TestPlayIterator(unittest.TestCase):
|
|||
mock_options = MagicMock()
|
||||
mock_options.syntax.value = True
|
||||
|
||||
templar = Templar(loader=fake_loader)
|
||||
|
||||
pbe = PlaybookExecutor(
|
||||
playbooks=['no_serial.yml', 'serial_int.yml', 'serial_pct.yml'],
|
||||
playbooks=['no_serial.yml', 'serial_int.yml', 'serial_pct.yml', 'serial_list.yml', 'serial_list_mixed.yml'],
|
||||
inventory=mock_inventory,
|
||||
variable_manager=mock_var_manager,
|
||||
loader=fake_loader,
|
||||
|
@ -79,27 +96,44 @@ class TestPlayIterator(unittest.TestCase):
|
|||
|
||||
playbook = Playbook.load(pbe._playbooks[0], variable_manager=mock_var_manager, loader=fake_loader)
|
||||
play = playbook.get_plays()[0]
|
||||
play.post_validate(templar)
|
||||
mock_inventory.get_hosts.return_value = ['host0','host1','host2','host3','host4','host5','host6','host7','host8','host9']
|
||||
self.assertEqual(pbe._get_serialized_batches(play), [['host0','host1','host2','host3','host4','host5','host6','host7','host8','host9']])
|
||||
|
||||
playbook = Playbook.load(pbe._playbooks[1], variable_manager=mock_var_manager, loader=fake_loader)
|
||||
play = playbook.get_plays()[0]
|
||||
play.post_validate(templar)
|
||||
mock_inventory.get_hosts.return_value = ['host0','host1','host2','host3','host4','host5','host6','host7','host8','host9']
|
||||
self.assertEqual(pbe._get_serialized_batches(play), [['host0','host1'],['host2','host3'],['host4','host5'],['host6','host7'],['host8','host9']])
|
||||
|
||||
playbook = Playbook.load(pbe._playbooks[2], variable_manager=mock_var_manager, loader=fake_loader)
|
||||
play = playbook.get_plays()[0]
|
||||
play.post_validate(templar)
|
||||
mock_inventory.get_hosts.return_value = ['host0','host1','host2','host3','host4','host5','host6','host7','host8','host9']
|
||||
self.assertEqual(pbe._get_serialized_batches(play), [['host0','host1'],['host2','host3'],['host4','host5'],['host6','host7'],['host8','host9']])
|
||||
|
||||
playbook = Playbook.load(pbe._playbooks[3], variable_manager=mock_var_manager, loader=fake_loader)
|
||||
play = playbook.get_plays()[0]
|
||||
play.post_validate(templar)
|
||||
mock_inventory.get_hosts.return_value = ['host0','host1','host2','host3','host4','host5','host6','host7','host8','host9']
|
||||
self.assertEqual(pbe._get_serialized_batches(play), [['host0'],['host1','host2'],['host3','host4','host5'],['host6','host7','host8'],['host9']])
|
||||
|
||||
playbook = Playbook.load(pbe._playbooks[4], variable_manager=mock_var_manager, loader=fake_loader)
|
||||
play = playbook.get_plays()[0]
|
||||
play.post_validate(templar)
|
||||
mock_inventory.get_hosts.return_value = ['host0','host1','host2','host3','host4','host5','host6','host7','host8','host9']
|
||||
self.assertEqual(pbe._get_serialized_batches(play), [['host0'],['host1','host2'],['host3','host4','host5','host6','host7','host8','host9']])
|
||||
|
||||
# Test when serial percent is under 1.0
|
||||
playbook = Playbook.load(pbe._playbooks[2], variable_manager=mock_var_manager, loader=fake_loader)
|
||||
play = playbook.get_plays()[0]
|
||||
play.post_validate(templar)
|
||||
mock_inventory.get_hosts.return_value = ['host0','host1','host2']
|
||||
self.assertEqual(pbe._get_serialized_batches(play), [['host0'],['host1'],['host2']])
|
||||
|
||||
# Test when there is a remainder for serial as a percent
|
||||
playbook = Playbook.load(pbe._playbooks[2], variable_manager=mock_var_manager, loader=fake_loader)
|
||||
play = playbook.get_plays()[0]
|
||||
play.post_validate(templar)
|
||||
mock_inventory.get_hosts.return_value = ['host0','host1','host2','host3','host4','host5','host6','host7','host8','host9','host10']
|
||||
self.assertEqual(pbe._get_serialized_batches(play), [['host0','host1'],['host2','host3'],['host4','host5'],['host6','host7'],['host8','host9'],['host10']])
|
||||
|
|
31
test/units/utils/test_helpers.py
Normal file
31
test/units/utils/test_helpers.py
Normal file
|
@ -0,0 +1,31 @@
|
|||
# (c) 2015, Marius Gedminas <marius@gedmin.as>
|
||||
#
|
||||
# This file is part of Ansible
|
||||
#
|
||||
# Ansible is free software: you can redistribute it and/or modify
|
||||
# it under the terms of the GNU General Public License as published by
|
||||
# the Free Software Foundation, either version 3 of the License, or
|
||||
# (at your option) any later version.
|
||||
#
|
||||
# Ansible is distributed in the hope that it will be useful,
|
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
# GNU General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU General Public License
|
||||
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
import unittest
|
||||
|
||||
from ansible.utils.helpers import pct_to_int
|
||||
|
||||
class TestHelpers(unittest.TestCase):
|
||||
|
||||
def test_pct_to_int(self):
|
||||
self.assertEqual(pct_to_int(1, 100), 1)
|
||||
self.assertEqual(pct_to_int(-1, 100), -1)
|
||||
self.assertEqual(pct_to_int("1%", 10), 1)
|
||||
self.assertEqual(pct_to_int("1%", 10, 0), 0)
|
||||
self.assertEqual(pct_to_int("1", 100), 1)
|
||||
self.assertEqual(pct_to_int("10%", 100), 10)
|
||||
|
Loading…
Add table
Add a link
Reference in a new issue