kolla-cli/tests/client_upgrade.py
Steve Noyes ed7b0253cc unit test updates for mitaka changes
- these directories are gone and should not be
collected by the log collector:
  - /etc/kolla/config
  - /usr/share/kolla/docs

- handle ceilometer-compute container properly
in the upgrade test

Change-Id: I83a2deffe64f22c9bf0776cb6768a32846ed198e
Jira-Issue: None
2016-09-15 18:40:36 +02:00

174 lines
6.5 KiB
Python

# Copyright(c) 2016, Oracle and/or its affiliates. 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.
#
from common import KollaCliTest
import logging
import os
import shutil
import unittest
from kollacli.api.client import ClientApi
from kollacli.common.allinone import AllInOne
from kollacli.common.utils import get_kollacli_etc
INV_NAME = 'inventory.json'
CLIENT = ClientApi()
LOG = logging.getLogger(__name__)
class TestFunctional(KollaCliTest):
"""Test description
This test will look for old version inventory files in the local current
working directory. If none are found, it will look in the user's home
directory. If none are found there too, it will print a warning and skip
the test.
Old version inventory files must be named inventory.json.v1,
inventory.json.v2, etc.
An upgrade test will be run on each old version inventory file that is
found.
"""
def test_upgrade(self):
inv_fpaths = self._find_inv_fpaths()
if not inv_fpaths:
LOG.warning('No old version inventory files were found. '
'Skipping test.')
for inv_fpath in inv_fpaths:
version = self._get_version(inv_fpath)
self._replace_inventory(inv_fpath)
self._test_upgrade(version)
def _get_version(self, inv_fpath):
try:
version = int(inv_fpath.split('.v')[1])
except Exception:
raise Exception('Invalid version number on old inventory file: %s'
% inv_fpath)
return version
def _find_inv_fpaths(self):
"""find old version inventories
Look in order at these locations:
- current working directory
- home directory
"""
fpaths = []
search_dirs = [os.getcwd(), os.path.expanduser('~')]
for search_dir in search_dirs:
fpaths = self._get_inv_fpaths(search_dir)
if fpaths:
break
return fpaths
def _get_inv_fpaths(self, inv_dir):
upg_inventory_paths = []
fnames = os.listdir(inv_dir)
for fname in fnames:
if fname.startswith(INV_NAME + '.v'):
path = os.path.join(inv_dir, fname)
upg_inventory_paths.append(path)
return upg_inventory_paths
def _replace_inventory(self, old_version_inv_path):
inv_path = os.path.join(get_kollacli_etc(),
'ansible', 'inventory.json')
shutil.copyfile(old_version_inv_path, inv_path)
def _test_upgrade(self, version):
hostname = 'test_host_upg'
# This host add will cause the inventory to be upgraded
CLIENT.host_add([hostname])
CLIENT.host_remove([hostname])
# run tests for each version:
if version <= 1:
self._test_v1_upgrade()
if version <= 2:
self._test_v2_upgrade()
def _test_v1_upgrade(self):
# this is a v1 inventory
# in v1 > v2, ceilometer was added, check that it's there
# and verify that all ceilometer groups are in the same groups
# as heat.
allinone = AllInOne()
heat = CLIENT.service_get(['heat'])[0]
expected_groups = sorted(heat.get_groups())
ceilometer = allinone.services['ceilometer']
expected_services = ceilometer.get_sub_servicenames()
expected_services.append('ceilometer')
expected_services = sorted(expected_services)
services = CLIENT.service_get_all()
services_found = []
for service in services:
servicename = service.get_name()
if servicename.startswith('ceilometer'):
groups = sorted(service.get_groups())
if servicename == 'ceilometer':
self.assertEqual(expected_groups, groups,
'groups mismatch between '
'ceilometer '
'and %s' % servicename)
elif servicename == 'ceilometer-compute':
self.assertEqual(['compute'], groups,
'groups mismatch between '
'ceilometer-compute '
'and %s' % servicename)
else:
# sub-services should have no groups (they inherit)
self.assertEqual([], groups,
'%s has unexpected groups'
% servicename)
services_found.append(servicename)
services_found = sorted(services_found)
self.assertEqual(expected_services, services_found,
'ceilometer subservices mismatch')
def _test_v2_upgrade(self):
# this is a v2 inventory. In the v2 to v3 upgrade, all subservices were
# fixed up to have a parent service
services = CLIENT.service_get_all()
servicenames = []
for service in services:
servicenames.append(service.name)
if '-' in service.name:
# this is a subservice
parent = service.get_parent()
self.assertIsNotNone(parent,
'subservice: %s, is missing its parent'
% service.name)
# ceilometer-alarms were removed in kolla v3.0.1,
# check that they're gone
self.assertNotIn('ceilometer-alarm-evaluator', servicenames,
'ceilometer-alarm-evaluator still exists.')
self.assertNotIn('ceilometer-alarm-notifier', servicenames,
'ceilometer-alarm-notifier still exists.')
# aodh and ceph were added in 3.0.1
self.assertIn('aodh', servicenames, 'aodh not in inventory')
self.assertIn('ceph', servicenames, 'ceph not in inventory')
if __name__ == '__main__':
unittest.main()