Fix inventory.get_hosts when hosts is a list.

This commit is contained in:
John Kleint
2012-09-11 13:00:40 -04:00
parent 565f336182
commit 1f8696f5c1
2 changed files with 11 additions and 1 deletions

View File

@@ -238,3 +238,11 @@ class TestInventory(unittest.TestCase):
'group_names': ['norse'],
'inventory_hostname': 'thor',
'inventory_hostname_short': 'thor'}
def test_hosts_list(self):
"""Test the case when playbook 'hosts' var is a list."""
inventory = self.script_inventory()
host_names = sorted(['thor', 'loki', 'odin']) # Not sure if sorting is in the contract or not
actual_hosts = inventory.get_hosts(host_names)
actual_host_names = [host.name for host in actual_hosts]
assert host_names == actual_host_names