126 lines
3.5 KiB
Python
126 lines
3.5 KiB
Python
# Copyright 2020 Red Hat, Inc.
|
|
# All Rights Reserved.
|
|
#
|
|
# 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
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import os
|
|
|
|
from tripleo_ansible.ansible_plugins.filter import tripleo_inventory
|
|
from tripleo_ansible.tests import base as tests_base
|
|
|
|
static_inventory = """
|
|
overcloud:
|
|
children:
|
|
allovercloud: {}
|
|
allovercloud:
|
|
children:
|
|
Compute: {}
|
|
Controller: {}
|
|
vars: {}
|
|
Controller:
|
|
children:
|
|
overcloud_Controller: {}
|
|
overcloud_Controller:
|
|
hosts:
|
|
overcloud-controller-0: {foo_bar: baz}
|
|
vars:
|
|
tripleo_role_name: Controller
|
|
Compute:
|
|
children:
|
|
overcloud_Compute: {}
|
|
overcloud_Compute:
|
|
hosts:
|
|
overcloud-novacompute-0: {foo_bar: baz}
|
|
vars:
|
|
tripleo_role_name: Compute
|
|
"""
|
|
|
|
dynamic_inventory = """
|
|
{
|
|
"overcloud": {
|
|
"children": [ "allovercloud" ]
|
|
},
|
|
"allovercloud": {
|
|
"children": [ "Compute", "Controller" ],
|
|
"vars" : {}
|
|
},
|
|
"Controller" : {
|
|
"children": [ "overcloud_Controller" ]
|
|
},
|
|
"overcloud_Controller" : {
|
|
"hosts": [ "overcloud-controller-0" ],
|
|
"vars": { "tripleo_role_name": "Controller" }
|
|
},
|
|
"Compute" : {
|
|
"children": [ "overcloud_Compute" ]
|
|
},
|
|
"overcloud_Compute" : {
|
|
"hosts": ["overcloud-novacompute-0" ],
|
|
"vars": { "tripleo_role_name": "Compute" }
|
|
}
|
|
}
|
|
"""
|
|
|
|
|
|
class TestInventoryFilters(tests_base.TestCase):
|
|
|
|
def setUp(self):
|
|
super(TestInventoryFilters, self).setUp()
|
|
|
|
def _test_hostmap(self, inventory):
|
|
expected = {
|
|
"Compute": ["overcloud-novacompute-0"],
|
|
"overcloud_Compute": ["overcloud-novacompute-0"],
|
|
"Controller": ["overcloud-controller-0"],
|
|
"overcloud_Controller": ["overcloud-controller-0"],
|
|
"overcloud": ["overcloud-controller-0", "overcloud-novacompute-0"],
|
|
"allovercloud": ["overcloud-controller-0", "overcloud-novacompute-0"],
|
|
}
|
|
result = tripleo_inventory.to_inventory_hostmap(inventory)
|
|
self.assertEqual(expected, result)
|
|
|
|
def test_hostmap_static(self):
|
|
self._test_hostmap(static_inventory)
|
|
|
|
def test_hostmap_dynamic(self):
|
|
self._test_hostmap(dynamic_inventory)
|
|
|
|
def _test_rolemap(self, inventory):
|
|
expected = {
|
|
"overcloud": ["Compute", "Controller"],
|
|
"allovercloud": ["Compute", "Controller"],
|
|
"Controller": ["Controller"],
|
|
"overcloud_Controller": ["Controller"],
|
|
"Compute": ["Compute"],
|
|
"overcloud_Compute": ["Compute"]
|
|
}
|
|
result = tripleo_inventory.to_inventory_rolemap(inventory)
|
|
self.assertEqual(expected, result)
|
|
|
|
def test_rolemap_static(self):
|
|
self._test_rolemap(static_inventory)
|
|
|
|
def test_rolemap_dynamic(self):
|
|
self._test_rolemap(dynamic_inventory)
|
|
|
|
def _test_roles(self, inventory):
|
|
expected = ["Compute", "Controller"]
|
|
result = tripleo_inventory.to_inventory_roles(inventory)
|
|
self.assertEqual(expected, result)
|
|
|
|
def test_roles_static(self):
|
|
self._test_roles(static_inventory)
|
|
|
|
def test_roles_dynamic(self):
|
|
self._test_roles(dynamic_inventory)
|