mirror of
https://github.com/ansible-collections/community.general.git
synced 2024-09-14 20:13:21 +02:00
60f295f7a0
commit 4430ce3eefcdff0b0ceffea0ef66ea8e876a807d
Merge: 631783b 649963c
Author: Michael DeHaan <michael.dehaan@gmail.com>
Date: Thu Jul 12 01:28:43 2012 -0400
Merge branch 'host-groups' of https://github.com/dagwieers/ansible into daggroups
commit 649963ca2c2610b97a90d2449132a57ae6b39ec9
Author: Dag Wieërs <dag@wieers.com>
Date: Thu Jul 12 23:01:00 2012 +0200
Added comments in the example yaml file as requested
commit 7f9718f185ec991bc165c4a52b2468cf41f4c349
Author: Dag Wieërs <dag@wieers.com>
Date: Thu Jul 12 22:49:38 2012 +0200
Add the default nose color too, to test specific overrides
commit eb63b9e899318ce0c26902ca73af50135a6224e4
Author: Dag Wieërs <dag@wieers.com>
Date: Thu Jul 12 22:44:35 2012 +0200
Introduce comics and cartoons to test yaml groups defined on a per-node basis
commit aa13d233078b825a8057bebf35ed478342cf4e43
Author: Dag Wieërs <dag@wieers.com>
Date: Thu Jul 12 19:33:15 2012 +0200
A small fix to revert to old state
commit 264ebaa77c4243f2e9117e8d1168dc2f2eed7ee2
Author: Dag Wieërs <dag@wieers.com>
Date: Thu Jul 12 19:31:51 2012 +0200
Combine both yaml unit tests into one example file
commit 7db49a8048e78402c4c9a0a6cb2604689280fbbb
Author: Dag Wieërs <dag@wieers.com>
Date: Thu Jul 12 16:46:53 2012 +0200
Might as well fix this too
commit f36c6c8c5b419865939c7e2b0b26f6c97255fdc8
Author: Dag Wieërs <dag@wieers.com>
Date: Thu Jul 12 16:42:00 2012 +0200
Added unit tests for host-groups patch
For the unit test I chose to keep the original yaml file in place as a reference.
This patch also includes a fix.
commit a96f6813522c5ae8b2be4514a2de56a775c6b7b0
Author: Dag Wieërs <dag@wieers.com>
Date: Thu Jul 12 12:30:43 2012 +0200
Allow groups to be defined on a per-host basis
This makes it possible to define on a per-host basis what groups a host is in.
When managing a large set of systems it makes it easier to ensure each of the
systems is defined in a set of groups (e.g. production/qa/development,
linux/solaris/aix) rather than having to add systems to multiple disconnected
groups.
----
- host: system01
- host: system02
- host: system03
- group: linux
hosts:
- system01
- system02
- group: solaris
hosts:
- system03
- group: production
hosts:
- system01
- system03
- group: qa
- system02
- group: dbserver
hosts:
- system01
- group: ntpserver
hosts:
- system02
- group: webserver
- system03
----
Can be redefined as:
----
- host: system01
groups: [ linux, production, dbserver ]
- host: system02
groups: [ linux, qa, ntpserver ]
- host: system03
groups: [ solaris, production, webserver ]
----
350 lines
12 KiB
Python
350 lines
12 KiB
Python
import os
|
|
import unittest
|
|
|
|
from ansible.inventory import Inventory
|
|
from ansible.runner import Runner
|
|
# from nose.plugins.skip import SkipTest
|
|
|
|
class TestInventory(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.cwd = os.getcwd()
|
|
self.test_dir = os.path.join(self.cwd, 'test')
|
|
|
|
self.inventory_file = os.path.join(self.test_dir, 'simple_hosts')
|
|
self.complex_inventory_file = os.path.join(self.test_dir, 'complex_hosts')
|
|
self.inventory_script = os.path.join(self.test_dir, 'inventory_api.py')
|
|
self.inventory_yaml = os.path.join(self.test_dir, 'yaml_hosts')
|
|
|
|
os.chmod(self.inventory_script, 0755)
|
|
|
|
def tearDown(self):
|
|
os.chmod(self.inventory_script, 0644)
|
|
|
|
def compare(self, left, right):
|
|
left = sorted(left)
|
|
right = sorted(right)
|
|
print left
|
|
print right
|
|
assert left == right
|
|
|
|
|
|
def simple_inventory(self):
|
|
return Inventory(self.inventory_file)
|
|
|
|
def script_inventory(self):
|
|
return Inventory(self.inventory_script)
|
|
|
|
def yaml_inventory(self):
|
|
return Inventory(self.inventory_yaml)
|
|
|
|
def complex_inventory(self):
|
|
return Inventory(self.complex_inventory_file)
|
|
|
|
#####################################
|
|
### Simple inventory format tests
|
|
|
|
def test_simple(self):
|
|
inventory = self.simple_inventory()
|
|
hosts = inventory.list_hosts()
|
|
|
|
expected_hosts=['jupiter', 'saturn', 'zeus', 'hera', 'poseidon', 'thor', 'odin', 'loki']
|
|
assert sorted(hosts) == sorted(expected_hosts)
|
|
|
|
def test_simple_all(self):
|
|
inventory = self.simple_inventory()
|
|
hosts = inventory.list_hosts('all')
|
|
|
|
expected_hosts=['jupiter', 'saturn', 'zeus', 'hera', 'poseidon', 'thor', 'odin', 'loki']
|
|
assert sorted(hosts) == sorted(expected_hosts)
|
|
|
|
def test_simple_norse(self):
|
|
inventory = self.simple_inventory()
|
|
hosts = inventory.list_hosts("norse")
|
|
|
|
expected_hosts=['thor', 'odin', 'loki']
|
|
assert sorted(hosts) == sorted(expected_hosts)
|
|
|
|
def test_simple_ungrouped(self):
|
|
inventory = self.simple_inventory()
|
|
hosts = inventory.list_hosts("ungrouped")
|
|
|
|
expected_hosts=['jupiter', 'saturn']
|
|
assert sorted(hosts) == sorted(expected_hosts)
|
|
|
|
def test_simple_combined(self):
|
|
inventory = self.simple_inventory()
|
|
hosts = inventory.list_hosts("norse:greek")
|
|
|
|
expected_hosts=['zeus', 'hera', 'poseidon', 'thor', 'odin', 'loki']
|
|
assert sorted(hosts) == sorted(expected_hosts)
|
|
|
|
def test_simple_restrict(self):
|
|
inventory = self.simple_inventory()
|
|
|
|
restricted_hosts = ['hera', 'poseidon', 'thor']
|
|
expected_hosts=['zeus', 'hera', 'poseidon', 'thor', 'odin', 'loki']
|
|
|
|
inventory.restrict_to(restricted_hosts)
|
|
hosts = inventory.list_hosts("norse:greek")
|
|
|
|
print "Hosts=%s" % hosts
|
|
print "Restricted=%s" % restricted_hosts
|
|
assert sorted(hosts) == sorted(restricted_hosts)
|
|
|
|
inventory.lift_restriction()
|
|
hosts = inventory.list_hosts("norse:greek")
|
|
|
|
print hosts
|
|
print expected_hosts
|
|
assert sorted(hosts) == sorted(expected_hosts)
|
|
|
|
def test_simple_exclude(self):
|
|
inventory = self.simple_inventory()
|
|
|
|
hosts = inventory.list_hosts("all:!greek")
|
|
expected_hosts=['jupiter', 'saturn', 'thor', 'odin', 'loki']
|
|
assert sorted(hosts) == sorted(expected_hosts)
|
|
|
|
hosts = inventory.list_hosts("all:!norse:!greek")
|
|
expected_hosts=['jupiter', 'saturn']
|
|
assert sorted(hosts) == sorted(expected_hosts)
|
|
|
|
def test_simple_vars(self):
|
|
inventory = self.simple_inventory()
|
|
vars = inventory.get_variables('thor')
|
|
|
|
print vars
|
|
assert vars == {'group_names': ['norse'],
|
|
'inventory_hostname': 'thor'}
|
|
|
|
def test_simple_port(self):
|
|
inventory = self.simple_inventory()
|
|
vars = inventory.get_variables('hera')
|
|
|
|
print vars
|
|
expected = {'ansible_ssh_port': 3000,
|
|
'group_names': ['greek'],
|
|
'inventory_hostname': 'hera'}
|
|
print expected
|
|
assert vars == expected
|
|
|
|
###################################################
|
|
### INI file advanced tests
|
|
|
|
def test_complex_vars(self):
|
|
inventory = self.complex_inventory()
|
|
|
|
vars = inventory.get_variables('rtp_a')
|
|
print vars
|
|
|
|
expected = dict(
|
|
a='1', b='2', c='3', d='100002', rga='1', rgb='2', rgc='3',
|
|
inventory_hostname='rtp_a',
|
|
group_names=[ 'eastcoast', 'nc', 'redundantgroup', 'redundantgroup2', 'redundantgroup3', 'rtp', 'us' ]
|
|
)
|
|
print vars
|
|
print expected
|
|
assert vars == expected
|
|
|
|
def test_complex_exclude(self):
|
|
inventory = self.complex_inventory()
|
|
|
|
hosts = inventory.list_hosts("nc:!triangle:florida:!orlando")
|
|
expected_hosts=['rtp_a', 'rtp_b', 'rtb_c', 'miami']
|
|
assert sorted(hosts) == sorted(expected_hosts)
|
|
|
|
|
|
|
|
###################################################
|
|
### Inventory API tests
|
|
|
|
def test_script(self):
|
|
inventory = self.script_inventory()
|
|
hosts = inventory.list_hosts()
|
|
|
|
expected_hosts=['jupiter', 'saturn', 'zeus', 'hera', 'poseidon', 'thor', 'odin', 'loki']
|
|
|
|
print "Expected: %s"%(expected_hosts)
|
|
print "Got : %s"%(hosts)
|
|
assert sorted(hosts) == sorted(expected_hosts)
|
|
|
|
def test_script_all(self):
|
|
inventory = self.script_inventory()
|
|
hosts = inventory.list_hosts('all')
|
|
|
|
expected_hosts=['jupiter', 'saturn', 'zeus', 'hera', 'poseidon', 'thor', 'odin', 'loki']
|
|
assert sorted(hosts) == sorted(expected_hosts)
|
|
|
|
def test_script_norse(self):
|
|
inventory = self.script_inventory()
|
|
hosts = inventory.list_hosts("norse")
|
|
|
|
expected_hosts=['thor', 'odin', 'loki']
|
|
assert sorted(hosts) == sorted(expected_hosts)
|
|
|
|
def test_script_combined(self):
|
|
inventory = self.script_inventory()
|
|
hosts = inventory.list_hosts("norse:greek")
|
|
|
|
expected_hosts=['zeus', 'hera', 'poseidon', 'thor', 'odin', 'loki']
|
|
assert sorted(hosts) == sorted(expected_hosts)
|
|
|
|
def test_script_restrict(self):
|
|
inventory = self.script_inventory()
|
|
|
|
restricted_hosts = ['hera', 'poseidon', 'thor']
|
|
expected_hosts=['zeus', 'hera', 'poseidon', 'thor', 'odin', 'loki']
|
|
|
|
inventory.restrict_to(restricted_hosts)
|
|
hosts = inventory.list_hosts("norse:greek")
|
|
|
|
assert sorted(hosts) == sorted(restricted_hosts)
|
|
|
|
inventory.lift_restriction()
|
|
hosts = inventory.list_hosts("norse:greek")
|
|
|
|
assert sorted(hosts) == sorted(expected_hosts)
|
|
|
|
def test_script_vars(self):
|
|
inventory = self.script_inventory()
|
|
vars = inventory.get_variables('thor')
|
|
|
|
print "VARS=%s" % vars
|
|
|
|
assert vars == {'hammer':True,
|
|
'group_names': ['norse'],
|
|
'inventory_hostname': 'thor'}
|
|
|
|
### Tests for yaml inventory file
|
|
|
|
def test_yaml(self):
|
|
inventory = self.yaml_inventory()
|
|
hosts = inventory.list_hosts()
|
|
expected_hosts=['garfield', 'goofy', 'hera', 'jerry', 'jupiter', 'loki', 'mars', 'mickey', 'odie', 'odin', 'poseidon', 'saturn', 'thor', 'tom', 'zeus']
|
|
self.compare(hosts, expected_hosts)
|
|
|
|
def test_yaml_all(self):
|
|
inventory = self.yaml_inventory()
|
|
hosts = inventory.list_hosts('all')
|
|
|
|
expected_hosts=['garfield', 'goofy', 'hera', 'jerry', 'jupiter', 'loki', 'mars', 'mickey', 'odie', 'odin', 'poseidon', 'saturn', 'thor', 'tom', 'zeus']
|
|
self.compare(hosts, expected_hosts)
|
|
|
|
def test_yaml_norse(self):
|
|
inventory = self.yaml_inventory()
|
|
hosts = inventory.list_hosts("norse")
|
|
|
|
expected_hosts=['thor', 'odin', 'loki']
|
|
self.compare(hosts, expected_hosts)
|
|
|
|
def test_yaml_ungrouped(self):
|
|
inventory = self.yaml_inventory()
|
|
hosts = inventory.list_hosts("ungrouped")
|
|
|
|
expected_hosts=['jupiter', 'mars']
|
|
self.compare(hosts, expected_hosts)
|
|
|
|
def test_yaml_combined(self):
|
|
inventory = self.yaml_inventory()
|
|
hosts = inventory.list_hosts("norse:greek")
|
|
|
|
expected_hosts=['zeus', 'hera', 'poseidon', 'thor', 'odin', 'loki']
|
|
self.compare(hosts, expected_hosts)
|
|
|
|
def test_yaml_restrict(self):
|
|
inventory = self.yaml_inventory()
|
|
|
|
restricted_hosts = ['hera', 'poseidon', 'thor']
|
|
expected_hosts=['zeus', 'hera', 'poseidon', 'thor', 'odin', 'loki']
|
|
|
|
inventory.restrict_to(restricted_hosts)
|
|
hosts = inventory.list_hosts("norse:greek")
|
|
|
|
self.compare(hosts, restricted_hosts)
|
|
|
|
inventory.lift_restriction()
|
|
hosts = inventory.list_hosts("norse:greek")
|
|
|
|
self.compare(hosts, expected_hosts)
|
|
|
|
def test_yaml_vars(self):
|
|
inventory = self.yaml_inventory()
|
|
vars = inventory.get_variables('thor')
|
|
assert vars == {'group_names': ['norse'],
|
|
'hammer':True,
|
|
'inventory_hostname': 'thor'}
|
|
|
|
def test_yaml_list_vars(self):
|
|
inventory = self.yaml_inventory()
|
|
vars = inventory.get_variables('zeus')
|
|
assert vars == {'ansible_ssh_port': 3001,
|
|
'group_names': ['greek', 'ruler'],
|
|
'inventory_hostname': 'zeus',
|
|
'ntp_server': 'olympus.example.com'}
|
|
|
|
def test_yaml_change_vars(self):
|
|
inventory = self.yaml_inventory()
|
|
vars = inventory.get_variables('thor')
|
|
|
|
vars["hammer"] = False
|
|
|
|
vars = inventory.get_variables('thor')
|
|
print vars
|
|
assert vars == {'hammer':True,
|
|
'inventory_hostname': 'thor',
|
|
'group_names': ['norse']}
|
|
|
|
def test_yaml_host_vars(self):
|
|
inventory = self.yaml_inventory()
|
|
vars = inventory.get_variables('saturn')
|
|
|
|
print vars
|
|
assert vars == {'inventory_hostname': 'saturn',
|
|
'moon': 'titan',
|
|
'moon2': 'enceladus',
|
|
'group_names': ['multiple']}
|
|
|
|
def test_yaml_port(self):
|
|
inventory = self.yaml_inventory()
|
|
vars = inventory.get_variables('hera')
|
|
|
|
print vars
|
|
assert vars == {'ansible_ssh_port': 3000,
|
|
'inventory_hostname': 'hera',
|
|
'ntp_server': 'olympus.example.com',
|
|
'group_names': ['greek']}
|
|
|
|
def test_yaml_multiple_groups(self):
|
|
inventory = self.yaml_inventory()
|
|
vars = inventory.get_variables('odin')
|
|
|
|
assert 'group_names' in vars
|
|
assert sorted(vars['group_names']) == [ 'norse', 'ruler' ]
|
|
|
|
def test_yaml_some_animals(self):
|
|
inventory = self.yaml_inventory()
|
|
hosts = inventory.list_hosts("cat:mouse")
|
|
expected_hosts=['garfield', 'jerry', 'mickey', 'tom']
|
|
self.compare(hosts, expected_hosts)
|
|
|
|
def test_yaml_comic(self):
|
|
inventory = self.yaml_inventory()
|
|
hosts = inventory.list_hosts("comic")
|
|
expected_hosts=['garfield', 'odie']
|
|
self.compare(hosts, expected_hosts)
|
|
|
|
def test_yaml_orange(self):
|
|
inventory = self.yaml_inventory()
|
|
hosts = inventory.list_hosts("orange")
|
|
expected_hosts=['garfield', 'goofy']
|
|
self.compare(hosts, expected_hosts)
|
|
|
|
def test_yaml_garfield_vars(self):
|
|
inventory = self.yaml_inventory()
|
|
vars = inventory.get_variables('garfield')
|
|
assert vars == {'ears': 'pointy',
|
|
'inventory_hostname': 'garfield',
|
|
'group_names': ['cat', 'comic', 'orange'],
|
|
'nose': 'pink'}
|
|
|