def test_get_upgrade_levels(mocker, levels, version, result, error): mocker.patch.dict("octane.magic_consts.UPGRADE_LEVELS", levels, clear=True) if error: msg = ("Could not find suitable upgrade_levels for the " "{version} release.".format(version=version)) with pytest.raises(KeyError, message=msg): nova.get_upgrade_levels(version) else: assert nova.get_upgrade_levels(version) == result
def set_upgrade_levels(env): version = env.data["fuel_version"] openstack_release = nova.get_upgrade_levels(version) for node in get_nodes(env, ["controller", "compute"]): node_util.add_compute_upgrade_levels(node, openstack_release) node_util.restart_nova_services(node)