Nicolas Pochet
141 lines
4.7 KiB
Python
141 lines
4.7 KiB
Python
# Copyright 2018 Canonical Ltd.
|
|
#
|
|
# 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.
|
|
|
|
"""Run full test lifecycle."""
|
|
import argparse
|
|
import asyncio
|
|
import logging
|
|
import os
|
|
import sys
|
|
import uuid
|
|
|
|
import zaza.charm_lifecycle.configure as configure
|
|
import zaza.charm_lifecycle.destroy as destroy
|
|
import zaza.charm_lifecycle.utils as utils
|
|
import zaza.charm_lifecycle.prepare as prepare
|
|
import zaza.charm_lifecycle.deploy as deploy
|
|
import zaza.charm_lifecycle.test as test
|
|
|
|
|
|
def generate_model_name():
|
|
"""Generate a unique model name.
|
|
|
|
:returns: Model name
|
|
:rtype: str
|
|
"""
|
|
return 'zaza-{}'.format(str(uuid.uuid4())[-12:])
|
|
|
|
|
|
def func_test_runner(keep_model=False, smoke=False, dev=False, bundle=None):
|
|
"""Deploy the bundles and run the tests as defined by the charms tests.yaml.
|
|
|
|
:param keep_model: Whether to destroy model at end of run
|
|
:type keep_model: boolean
|
|
:param smoke: Whether to just run smoke test.
|
|
:param dev: Whether to just run dev test.
|
|
:type smoke: boolean
|
|
:type dev: boolean
|
|
"""
|
|
test_config = utils.get_charm_config()
|
|
if bundle:
|
|
bundles = [bundle]
|
|
else:
|
|
if smoke:
|
|
bundle_key = 'smoke_bundles'
|
|
elif dev:
|
|
bundle_key = 'dev_bundles'
|
|
else:
|
|
bundle_key = 'gate_bundles'
|
|
bundles = test_config[bundle_key]
|
|
last_test = bundles[-1]
|
|
for t in bundles:
|
|
model_name = generate_model_name()
|
|
# Prepare
|
|
prepare.prepare(model_name)
|
|
# Deploy
|
|
deploy.deploy(
|
|
os.path.join(utils.BUNDLE_DIR, '{}.yaml'.format(t)),
|
|
model_name)
|
|
# Configure
|
|
configure.configure(model_name, test_config['configure'])
|
|
# Test
|
|
test.test(model_name, test_config['tests'])
|
|
# Destroy
|
|
# Keep the model from the last run if keep_model is true, this is to
|
|
# maintian compat with osci and should change when the zaza collect
|
|
# functions take over from osci for artifact collection.
|
|
if keep_model and t == last_test:
|
|
pass
|
|
else:
|
|
destroy.destroy(model_name)
|
|
|
|
|
|
def parse_args(args):
|
|
"""Parse command line arguments.
|
|
|
|
:param args: List of configure functions functions
|
|
:type list: [str1, str2,...] List of command line arguments
|
|
:returns: Parsed arguments
|
|
:rtype: Namespace
|
|
"""
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--keep-model', dest='keep_model',
|
|
help='Keep model at the end of the run',
|
|
action='store_true')
|
|
parser.add_argument('--smoke', dest='smoke',
|
|
help='Just run smoke test(s)',
|
|
action='store_true')
|
|
parser.add_argument('--dev', dest='dev',
|
|
help='Just run dev test(s)',
|
|
action='store_true')
|
|
parser.add_argument('-b', '--bundle', dest='bundle',
|
|
help='Override the bundle to be run',
|
|
required=False)
|
|
parser.add_argument('--log', dest='loglevel',
|
|
help='Loglevel [DEBUG|INFO|WARN|ERROR|CRITICAL]')
|
|
parser.set_defaults(keep_model=False,
|
|
smoke=False,
|
|
dev=False,
|
|
loglevel='INFO')
|
|
return parser.parse_args(args)
|
|
|
|
|
|
def main():
|
|
"""Execute full test run."""
|
|
args = parse_args(sys.argv[1:])
|
|
|
|
level = getattr(logging, args.loglevel.upper(), None)
|
|
if not isinstance(level, int):
|
|
raise ValueError('Invalid log level: "{}"'.format(args.loglevel))
|
|
logging.basicConfig(level=level)
|
|
|
|
if args.dev and args.smoke:
|
|
raise ValueError('Ambiguous arguments: --smoke and '
|
|
'--dev cannot be used together')
|
|
|
|
if args.dev and args.bundle:
|
|
raise ValueError('Ambiguous arguments: --bundle and '
|
|
'--dev cannot be used together')
|
|
|
|
if args.smoke and args.bundle:
|
|
raise ValueError('Ambiguous arguments: --bundle and '
|
|
'--smoke cannot be used together')
|
|
|
|
func_test_runner(
|
|
keep_model=args.keep_model,
|
|
smoke=args.smoke,
|
|
dev=args.dev,
|
|
bundle=args.bundle)
|
|
asyncio.get_event_loop().close()
|