import os import unittest from ansible.inventory import Inventory 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') 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 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', 'cerberus001','cerberus002','cerberus003', 'cottus99', 'cottus100', 'poseidon', 'thor', 'odin', 'loki', 'thrudgelmir0', 'thrudgelmir1', 'thrudgelmir2', 'thrudgelmir3', 'thrudgelmir4', 'thrudgelmir5'] 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', 'cerberus001','cerberus002','cerberus003', 'cottus99', 'cottus100', 'poseidon', 'thor', 'odin', 'loki', 'thrudgelmir0', 'thrudgelmir1', 'thrudgelmir2', 'thrudgelmir3', 'thrudgelmir4', 'thrudgelmir5'] 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', 'thrudgelmir0', 'thrudgelmir1', 'thrudgelmir2', 'thrudgelmir3', 'thrudgelmir4', 'thrudgelmir5'] 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', 'cerberus001','cerberus002','cerberus003', 'cottus99','cottus100', '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', 'cerberus001','cerberus002','cerberus003', 'cottus99', 'cottus100', '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', 'thrudgelmir0', 'thrudgelmir1', 'thrudgelmir2', 'thrudgelmir3', 'thrudgelmir4', 'thrudgelmir5'] assert sorted(hosts) == sorted(expected_hosts) hosts = inventory.list_hosts("all:!norse:!greek") expected_hosts=['jupiter', 'saturn', 'thrudgelmir0', 'thrudgelmir1', 'thrudgelmir2', 'thrudgelmir3', 'thrudgelmir4', 'thrudgelmir5'] 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', 'inventory_hostname_short': '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', 'inventory_hostname_short': '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', inventory_hostname_short='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', 'inventory_hostname_short': 'thor'}