def test_add_host_vars_host_not_exist_raises_exception(self):
        from inventory import HostDoesNotExist

        inv = AwxInventory()

        test_host_vars = {
            'var1': 'val1',
            'var2': 'val2',
        }

        with self.assertRaises(HostDoesNotExist):
            inv.add_host_vars('host1', test_host_vars)
    def test_add_host_vars(self):

        inv = AwxInventory()
        inv.add_host('host1')

        test_host_vars = {
            'var1': 'val1',
            'var2': 'val2',
        }

        inv.add_host_vars('host1', test_host_vars)

        test_dict = {'host1': test_host_vars}

        self.assertDictEqual(inv.hosts, test_dict)