From 4f05c70ef142ecbbfdabcaa0c269f707b3d7911d Mon Sep 17 00:00:00 2001 From: Luciano Lo Giudice <luciano.logiudice@canonical.com> Date: Thu, 28 Nov 2024 17:38:42 -0300 Subject: [PATCH] Add a class to refresh all charms in a directory. --- zaza/charm_tests/lifecycle/refresh.py | 61 +++++++++++++++++++++++++++ 1 file changed, 61 insertions(+) create mode 100644 zaza/charm_tests/lifecycle/refresh.py diff --git a/zaza/charm_tests/lifecycle/refresh.py b/zaza/charm_tests/lifecycle/refresh.py new file mode 100644 index 00000000..b8597e9d --- /dev/null +++ b/zaza/charm_tests/lifecycle/refresh.py @@ -0,0 +1,61 @@ +#!/usr/bin/env python3 + +# Copyright 2024 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. + +"""Code for managing various lifecycle functionality via Zaza.""" + +import glob +import logging +import os + +import zaza.model as model +import subprocess + + +class CharmRefreshAll: + """Perform upgrades of all charms in a local file path.""" + + test_runner = 'direct_with_args' + + def run(self, *args, **kwargs): + """Upgrade all the charms found in a local path.""" + path = os.environ.get('CHARMS_ARTIFACT_DIR') + if not path: + logging.info('charms path not found') + return True + + path = os.path.realpath(path) + if path[-1] != '/': + path += '/' + + charms = glob.glob(path + '*.charm') + if not charms: + logging.info('no charms found in path') + return True + + for charm in charms: + app_name = charm[len(path):].replace('.charm', '') + try: + model.get_application(app_name) + except KeyError: + logging.info('charm %s not found in current model' % app_name) + continue + + logging.info('refreshing charm %s' % app_name) + subprocess.check_call([ + 'juju', 'refresh', '--path', charm, app_name + ]) + + return True