7e92ff823e
The goal of breaking apart the base_parser() function is to get rid of a bunch of conditionals and parameters in the code and, instead, make code look like simple composition. When splitting, a choice had to be made as to whether this would operate by side effect (modifying a passed in parser) or side effect-free (returning a new parser everytime). Making a version that's side-effect-free appears to be fighting with the optparse API (it wants to work by creating a parser object, configuring the object, and then parsing the arguments with it) so instead, make it clear that our helper functions are modifying the passed in parser by (1) not returning the parser and (2) changing the function names to be more clear that it is operating by side-effect. Also move all of the generic optparse code, along with the argument context classes, into a new subdirectory.
148 lines
6.3 KiB
Python
148 lines
6.3 KiB
Python
# (c) 2012-2014, Michael DeHaan <michael.dehaan@gmail.com>
|
|
#
|
|
# 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/>.
|
|
|
|
# Make coding more python3-ish
|
|
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
from units.compat import unittest
|
|
from units.compat.mock import MagicMock
|
|
|
|
from ansible.arguments import context_objects as co
|
|
from ansible.executor.playbook_executor import PlaybookExecutor
|
|
from ansible.playbook import Playbook
|
|
from ansible.template import Templar
|
|
|
|
from units.mock.loader import DictDataLoader
|
|
|
|
|
|
class TestPlaybookExecutor(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
# Reset command line args for every test
|
|
co.GlobalCLIArgs._Singleton__instance = None
|
|
|
|
def tearDown(self):
|
|
# And cleanup after ourselves too
|
|
co.GlobalCLIArgs._Singleton__instance = None
|
|
|
|
def test_get_serialized_batches(self):
|
|
fake_loader = DictDataLoader({
|
|
'no_serial.yml': '''
|
|
- hosts: all
|
|
gather_facts: no
|
|
tasks:
|
|
- debug: var=inventory_hostname
|
|
''',
|
|
'serial_int.yml': '''
|
|
- hosts: all
|
|
gather_facts: no
|
|
serial: 2
|
|
tasks:
|
|
- debug: var=inventory_hostname
|
|
''',
|
|
'serial_pct.yml': '''
|
|
- hosts: all
|
|
gather_facts: no
|
|
serial: 20%
|
|
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()
|
|
mock_var_manager = MagicMock()
|
|
|
|
templar = Templar(loader=fake_loader)
|
|
|
|
pbe = PlaybookExecutor(
|
|
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,
|
|
passwords=[],
|
|
)
|
|
|
|
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']]
|
|
)
|