System configuration for the OpenDev Collaboratory
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

98 lines
3.7 KiB

# Copyright (C) 2018 Red Hat, Inc.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# Make coding more python3-ish
from __future__ import (absolute_import, division, print_function)
__metaclass__ = type
import os
import testtools
import mock
import yaml
from import Host
from .yamlgroup import InventoryModule
FIXTURE_DIR = os.path.join(os.path.dirname(__file__),
class TestInventory(testtools.TestCase):
def test_yaml_groups(self):
inventory = mock.MagicMock()
results_yaml = os.path.join(FIXTURE_DIR, 'results.yaml')
with open(results_yaml) as f:
results = yaml.load(f, Loader=yaml.FullLoader)
results = results['results']
# Build the inventory list. This is a list of Host objects
# which are the keys in our results.yaml file, keyed by the
# hostname (... I dunno, we're just tricking the inventory and
# making something it's happy with)
inventory.hosts = {}
for host in results.keys():
inventory.hosts[host] = Host(name=host)
# Fake out add_group() and add_child() for the inventory
# object to store our groups.
inventory.groups = {}
def add_group(group):
inventory.groups[group] = []
inventory.add_group = add_group
def add_child(group, host):
inventory.add_child = add_child
# Not really needed for unit test
loader = mock.MagicMock()
# This is all setup by ansible magic plugin/inventory stuff in
# real-life, which gets the groups into the config object
path = os.path.join(FIXTURE_DIR, 'groups.yaml')
with open(path) as f:
config_groups = yaml.load(f, Loader=yaml.FullLoader)
config_groups = config_groups['groups']
im = InventoryModule()
im._read_config_data = mock.MagicMock()
im._load_name = 'yamlgroup'
im.get_option = mock.MagicMock(side_effect=lambda x: config_groups)
im.parse(inventory, loader, path)
# Now, for every host we have in our results, we should be
# able to see it listed as a child of the groups it wants to
# be in
for host, groups in results.items():
for group in groups:
"The inventory does not have a group <%s>;"
"host <%s> should be in this group" % (group, host))
self.assertEquals(group in inventory.groups, True, message)
"The group <%s> does not contain host <%s>"
% (group, host))
self.assertIn(host, inventory.groups[group], message)
# Additionally, check this host hasn't managed to get into
# any groups it is *not* supposed to be in
for inventory_group, inventory_hosts in inventory.groups.items():
if host in inventory_hosts:
message = ("The host <%s> should not be in group <%s>"
% (host, inventory_group))
self.assertTrue(inventory_group in groups, message)