def test_add_host_to_group_raises_exception_no_group(self):
        from inventory import GroupDoesNotExist

        inv = AwxInventory()

        inv.add_host('host1')

        with self.assertRaises(GroupDoesNotExist):
            inv.add_host_to_group('host1', 'group1')
    def test_add_host_to_group(self):

        inv = AwxInventory()

        inv.add_host('host1')
        inv.add_group('group1')

        inv.add_host_to_group('host1', 'group1')

        test_dict = {'group1': {'hosts': ['host1']}}

        self.assertDictEqual(inv.groups, test_dict)
    def test_remove_group_with_host_check_hosts_host_removed(self):
        inv = AwxInventory()

        inv.add_host('host1')
        inv.add_host('host2')
        inv.add_host('host3')

        inv.add_group('group1')
        inv.add_group('group2')

        inv.add_host_to_group('host1', 'group1')
        inv.add_host_to_group('host2', 'group2')
        inv.add_host_to_group('host3', 'group1')

        inv.remove_group('group1', delete_host=True)

        test_dict = {'host2': {}}

        self.assertDictEqual(inv.hosts, test_dict)