diff --git a/lib/ansible/inventory.py b/lib/ansible/inventory.py index 7cb8c7253eb..ef790b6d313 100644 --- a/lib/ansible/inventory.py +++ b/lib/ansible/inventory.py @@ -81,13 +81,19 @@ class Inventory(object): def get_variables(self, host): """ Return the variables associated with this host. """ + variables = {} if host in self._variables: - return self._variables[host].copy() + variables.update(self._variables[host].copy()) - if not self._is_script: - return {} + if self._is_script: + variables.update(self._get_variables_from_script(host)) - return self._get_variables_from_script(host) + variables['group_names'] = [] + for name,hosts in self.groups.iteritems(): + if host in hosts: + variables['group_names'].append(name) + + return variables # ***************************************************** diff --git a/test/TestInventory.py b/test/TestInventory.py index 96991d77d00..c136430d18d 100644 --- a/test/TestInventory.py +++ b/test/TestInventory.py @@ -85,13 +85,13 @@ class TestInventory(unittest.TestCase): inventory = self.simple_inventory() vars = inventory.get_variables('thor') - assert vars == {} + assert vars == {'group_names': ['norse']} def test_simple_port(self): inventory = self.simple_inventory() vars = inventory.get_variables('hera') - assert vars == {'ansible_ssh_port': 3000} + assert vars == {'ansible_ssh_port': 3000, 'group_names': ['greek']} ### Inventory API tests @@ -146,7 +146,7 @@ class TestInventory(unittest.TestCase): inventory = self.script_inventory() vars = inventory.get_variables('thor') - assert vars == {"hammer":True} + assert vars == {"hammer":True, 'group_names': ['norse']} ### Tests for yaml inventory file @@ -205,7 +205,7 @@ class TestInventory(unittest.TestCase): inventory = self.yaml_inventory() vars = inventory.get_variables('thor') - assert vars == {"hammer":True} + assert vars == {"hammer":True, 'group_names': ['norse']} def test_yaml_change_vars(self): inventory = self.yaml_inventory() @@ -214,19 +214,30 @@ class TestInventory(unittest.TestCase): vars["hammer"] = False vars = inventory.get_variables('thor') - assert vars == {"hammer":True} + assert vars == {"hammer":True, 'group_names': ['norse']} def test_yaml_host_vars(self): inventory = self.yaml_inventory() vars = inventory.get_variables('saturn') - assert vars == {"moon":"titan", "moon2":"enceladus"} + assert vars == {"moon":"titan", + "moon2":"enceladus", + 'group_names': ['multiple']} def test_yaml_port(self): inventory = self.yaml_inventory() vars = inventory.get_variables('hera') - assert vars == {'ansible_ssh_port': 3000, 'ntp_server': 'olympus.example.com'} + assert vars == {'ansible_ssh_port': 3000, + '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' ] ### Test Runner class method diff --git a/test/yaml_hosts b/test/yaml_hosts index 628c2843506..8d98d5133e4 100644 --- a/test/yaml_hosts +++ b/test/yaml_hosts @@ -25,6 +25,11 @@ - odin - loki +- group: ruler + hosts: + - zeus + - odin + - group: multiple hosts: - saturn