Merge "Expand roles honstname to role mapping"

This commit is contained in:
Zuul 2021-01-12 23:37:37 +00:00 committed by Gerrit Code Review
commit 9363129f99
3 changed files with 80 additions and 22 deletions

View File

@ -171,6 +171,7 @@ def expand(roles, stack_name, expand_provisioned=True, default_image=None,
hostname_map = {}
parameter_defaults = {'HostnameMap': hostname_map}
role_net_map = {}
hostname_role_map = {}
for role in roles:
name = role['name']
role_net_map.setdefault(name, set())
@ -264,6 +265,7 @@ def expand(roles, stack_name, expand_provisioned=True, default_image=None,
role_net_map[name].update(
[x['network'] for x in inst.get('networks', [])])
hostname_role_map[inst['hostname']] = name
parameter_defaults['%sCount' % name] = (
provisioned_count)
@ -273,7 +275,7 @@ def expand(roles, stack_name, expand_provisioned=True, default_image=None,
env = {'parameter_defaults': parameter_defaults}
else:
env = {}
return instances, env, role_net_map
return instances, env, role_net_map, hostname_role_map
def merge_networks_defaults(defaults, instance):

View File

@ -216,7 +216,7 @@ def main():
try:
if state in ('present', 'all'):
present, env, role_net_map = bd.expand(
present, env, role_net_map, hostname_role_map = bd.expand(
roles=module.params['baremetal_deployment'],
stack_name=module.params['stack_name'],
expand_provisioned=True,
@ -226,7 +226,7 @@ def main():
ssh_public_keys=module.params['ssh_public_keys'],
)
if state in ('absent', 'all'):
absent, _, _ = bd.expand(
absent, _, _, _ = bd.expand(
roles=module.params['baremetal_deployment'],
stack_name=module.params['stack_name'],
expand_provisioned=False,
@ -234,6 +234,7 @@ def main():
)
env = {}
role_net_map = {}
hostname_role_map = {}
if state == 'present':
instances = present
elif state == 'absent':
@ -246,7 +247,8 @@ def main():
msg='Expanded to %d instances' % len(instances),
instances=instances,
environment=env,
role_net_map=role_net_map
role_net_map=role_net_map,
hostname_role_map=hostname_role_map,
)
except Exception as e:
module.fail_json(msg=str(e))

View File

@ -102,7 +102,7 @@ class TestExpandRoles(base.TestCase):
{'name': 'Compute'},
{'name': 'Controller'},
]
instances, environment, role_net_map = bd.expand(
instances, environment, role_net_map, hostname_role_map = bd.expand(
roles, 'overcloud', True, self.default_image
)
@ -128,13 +128,16 @@ class TestExpandRoles(base.TestCase):
}
},
environment['parameter_defaults'])
self.assertEqual({'overcloud-controller-0': 'Controller',
'overcloud-novacompute-0': 'Compute'},
hostname_role_map)
def test_default_network(self):
roles = [
{'name': 'Compute'},
{'name': 'Controller'},
]
instances, environment, role_net_map = bd.expand(
instances, environment, role_net_map, hostname_role_map = bd.expand(
roles, 'overcloud', True, self.default_image, self.default_network
)
self.assertEqual(
@ -149,6 +152,9 @@ class TestExpandRoles(base.TestCase):
'nics': [{'network': 'ctlplane'}]},
],
instances)
self.assertEqual({'overcloud-controller-0': 'Controller',
'overcloud-novacompute-0': 'Compute'},
hostname_role_map)
def test_networks_set_no_default_network(self):
roles = [
@ -165,7 +171,7 @@ class TestExpandRoles(base.TestCase):
]}
},
]
instances, environment, role_net_map = bd.expand(
instances, environment, role_net_map, hostname_role_map = bd.expand(
roles, 'overcloud', True, self.default_image, None
)
self.assertEqual(
@ -180,6 +186,9 @@ class TestExpandRoles(base.TestCase):
'nics': [{'network': 'some_net'}]},
],
instances)
self.assertEqual({'overcloud-controller-0': 'Controller',
'overcloud-novacompute-0': 'Compute'},
hostname_role_map)
def test_networks_set_default_appended(self):
roles = [
@ -196,7 +205,7 @@ class TestExpandRoles(base.TestCase):
]}
},
]
instances, environment, role_net_map = bd.expand(
instances, environment, role_net_map, hostname_role_map = bd.expand(
roles, 'overcloud', True, self.default_image, self.default_network
)
self.assertEqual(
@ -213,6 +222,9 @@ class TestExpandRoles(base.TestCase):
'nics': [{'network': 'ctlplane'}]},
],
instances)
self.assertEqual({'overcloud-controller-0': 'Controller',
'overcloud-novacompute-0': 'Compute'},
hostname_role_map)
def test_networks_vif_set_default_appended(self):
roles = [
@ -229,7 +241,7 @@ class TestExpandRoles(base.TestCase):
]}
},
]
instances, environment, role_net_map = bd.expand(
instances, environment, role_net_map, hostname_role_map = bd.expand(
roles, 'overcloud', True, self.default_image, self.default_network
)
self.assertEqual(
@ -255,6 +267,9 @@ class TestExpandRoles(base.TestCase):
]},
],
instances)
self.assertEqual({'overcloud-controller-0': 'Controller',
'overcloud-novacompute-0': 'Compute'},
hostname_role_map)
def test_networks_nics_are_mutually_exclusive(self):
# Neither 'nics' nor 'networks' - OK
@ -289,7 +304,7 @@ class TestExpandRoles(base.TestCase):
'hostname': 'overcloud-controller-1',
}]
}]
instances, environment, role_net_map = bd.expand(
instances, environment, role_net_map, hostname_role_map = bd.expand(
roles, 'overcloud', True, self.default_image
)
self.assertEqual(
@ -321,7 +336,7 @@ class TestExpandRoles(base.TestCase):
},
'hostname_format': 'controller-%index%.example.com'
}]
instances, environment, role_net_map = bd.expand(
instances, environment, role_net_map, hostname_role_map = bd.expand(
roles, 'overcloud', True, self.default_image,
user_name='heat-admin', ssh_public_keys='aaaa'
)
@ -366,6 +381,12 @@ class TestExpandRoles(base.TestCase):
}
},
environment['parameter_defaults'])
self.assertEqual({'compute-0.example.com': 'Compute',
'compute-1.example.com': 'Compute',
'controller-0.example.com': 'Controller',
'controller-1.example.com': 'Controller',
'controller-2.example.com': 'Controller'},
hostname_role_map)
def test_explicit_instances(self):
roles = [{
@ -397,7 +418,7 @@ class TestExpandRoles(base.TestCase):
'vif': True}]},
]},
]
instances, environment, role_net_map = bd.expand(
instances, environment, role_net_map, hostname_role_map = bd.expand(
roles, 'overcloud', True, self.default_image
)
self.assertEqual(
@ -440,6 +461,11 @@ class TestExpandRoles(base.TestCase):
}
},
environment['parameter_defaults'])
self.assertEqual({'compute-0.example.com': 'Compute',
'compute-1.example.com': 'Compute',
'controller-X.example.com': 'Controller',
'node-0': 'Controller'},
hostname_role_map)
def test_count_with_instances(self):
roles = [{
@ -464,7 +490,7 @@ class TestExpandRoles(base.TestCase):
'nics': [{'subnet': 'leaf-2'}]},
]},
]
instances, environment, role_net_map = bd.expand(
instances, environment, role_net_map, hostname_role_map = bd.expand(
roles, 'overcloud', True, self.default_image
)
self.assertEqual([
@ -508,6 +534,12 @@ class TestExpandRoles(base.TestCase):
'overcloud-controller-2': 'overcloud-controller-2'}
},
environment['parameter_defaults'])
self.assertEqual({'compute-0.example.com': 'Compute',
'compute-1.example.com': 'Compute',
'controller-X.example.com': 'Controller',
'node-0': 'Controller',
'overcloud-controller-2': 'Controller'},
hostname_role_map)
def test_unprovisioned(self):
roles = [{
@ -524,7 +556,7 @@ class TestExpandRoles(base.TestCase):
'provisioned': False
}]
}]
instances, environment, role_net_map = bd.expand(
instances, environment, role_net_map, hostname_role_map = bd.expand(
roles, 'overcloud', True, self.default_image
)
self.assertEqual([
@ -552,8 +584,11 @@ class TestExpandRoles(base.TestCase):
'overcloud-controller-3': 'overcloud-controller-3'}
},
environment['parameter_defaults'])
self.assertEqual({'overcloud-controller-0': 'Controller',
'overcloud-controller-3': 'Controller'},
hostname_role_map)
instances, environment, role_net_map = bd.expand(
instances, environment, role_net_map, hostname_role_map = bd.expand(
roles, 'overcloud', False, self.default_image
)
self.assertEqual([
@ -568,6 +603,9 @@ class TestExpandRoles(base.TestCase):
}],
instances)
self.assertEqual({}, environment)
self.assertEqual({'overcloud-controller-1': 'Controller',
'overcloud-controller-2': 'Controller'},
hostname_role_map)
def test_reprovisioned(self):
roles = [{
@ -584,7 +622,7 @@ class TestExpandRoles(base.TestCase):
'provisioned': False
}]
}]
instances, environment, role_net_map = bd.expand(
instances, environment, role_net_map, hostname_role_map = bd.expand(
roles, 'overcloud', True, self.default_image
)
self.assertEqual([
@ -622,8 +660,13 @@ class TestExpandRoles(base.TestCase):
'overcloud-controller-5': 'overcloud-controller-5'}
},
environment['parameter_defaults'])
self.assertEqual({'overcloud-controller-0': 'Controller',
'overcloud-controller-3': 'Controller',
'overcloud-controller-4': 'Controller',
'overcloud-controller-5': 'Controller'},
hostname_role_map)
instances, environment, role_net_map = bd.expand(
instances, environment, role_net_map, hostname_role_map = bd.expand(
roles, 'overcloud', False, self.default_image
)
self.assertEqual([
@ -663,7 +706,7 @@ class TestExpandRoles(base.TestCase):
'provisioned': True
}]
}]
instances, environment, role_net_map = bd.expand(
instances, environment, role_net_map, hostname_role_map = bd.expand(
roles, 'overcloud', True, self.default_image
)
self.assertEqual([
@ -694,7 +737,7 @@ class TestExpandRoles(base.TestCase):
},
environment['parameter_defaults'])
instances, environment, role_net_map = bd.expand(
instances, environment, role_net_map, hostname_role_map = bd.expand(
roles, 'overcloud', False, self.default_image
)
self.assertEqual([
@ -711,6 +754,8 @@ class TestExpandRoles(base.TestCase):
}],
instances)
self.assertEqual({}, environment)
self.assertEqual({'controller-1': 'Controller',
'controller-2': 'Controller'}, hostname_role_map)
def test_unprovisioned_no_hostname(self):
roles = [{
@ -732,7 +777,7 @@ class TestExpandRoles(base.TestCase):
'provisioned': True
}]
}]
instances, environment, role_net_map = bd.expand(
instances, environment, role_net_map, hostname_role_map = bd.expand(
roles, 'overcloud', True, self.default_image
)
self.assertEqual([
@ -762,8 +807,10 @@ class TestExpandRoles(base.TestCase):
'overcloud-controller-3': 'node-3'}
},
environment['parameter_defaults'])
self.assertEqual({'node-0': 'Controller', 'node-3': 'Controller'},
hostname_role_map)
instances, environment, role_net_map = bd.expand(
instances, environment, role_net_map, hostname_role_map = bd.expand(
roles, 'overcloud', False, self.default_image
)
self.assertEqual([
@ -780,6 +827,8 @@ class TestExpandRoles(base.TestCase):
}],
instances)
self.assertEqual({}, environment)
self.assertEqual({'node-1': 'Controller', 'node-2': 'Controller'},
hostname_role_map)
def test_name_in_defaults(self):
roles = [{
@ -836,7 +885,7 @@ class TestExpandRoles(base.TestCase):
'nics': [{'subnet': 'leaf-2'}]},
]},
]
instances, environment, role_net_map = bd.expand(
instances, environment, role_net_map, hostname_role_map = bd.expand(
roles, 'overcloud', True, self.default_image
)
self.assertEqual(
@ -854,6 +903,11 @@ class TestExpandRoles(base.TestCase):
'traits': ['CUSTOM_FOO'], 'nics': [{'subnet': 'leaf-2'}]},
],
instances)
self.assertEqual({'compute-0.example.com': 'Compute',
'compute-1.example.com': 'Compute',
'node-0': 'Controller',
'overcloud-controller-0': 'Controller'},
hostname_role_map)
def test_more_instances_than_count(self):
roles = [{