murano-dashboard/muranodashboard/tests/functional/utils.py

140 lines
4.6 KiB
Python

# 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.
import json
import logging
import os
import yaml
import zipfile
from oslo_log import log
from selenium.webdriver.support import wait
from muranodashboard.tests.functional.config import config as cfg
from muranodashboard.tests.functional import consts
log = log.getLogger(__name__).logger
log.setLevel(logging.DEBUG)
log.addHandler(logging.StreamHandler())
MANIFEST = {'Format': 'MuranoPL/1.0',
'Type': 'Application',
'Description': 'MockApp for webUI tests',
'Author': 'Mirantis, Inc',
'UI': 'mock_ui.yaml'}
class ImageException(Exception):
message = "Image doesn't exist"
def __init__(self, im_type):
self._error_string = (self.message + '\nDetails: {0} image is '
'not found,'.format(im_type))
def __str__(self):
return self._error_string
def upload_app_package(client, app_name, data, hot=False,
package_dir=consts.PackageDir, **manifest_kwargs):
try:
if not hot:
manifest = os.path.join(package_dir, 'manifest.yaml')
archive = compose_package(app_name, manifest, package_dir,
**manifest_kwargs)
else:
manifest = os.path.join(consts.HotPackageDir, 'manifest.yaml')
archive = compose_package(app_name, manifest,
consts.HotPackageDir, hot=True,
**manifest_kwargs)
package = client.packages.create(data, {app_name: open(archive, 'rb')})
return package.id
finally:
os.remove(archive)
os.remove(manifest)
def compose_package(app_name, manifest, package_dir,
require=None, archive_dir=None, hot=False, **kwargs):
"""Composes a murano package
Composes package `app_name` with `manifest` file as a template for the
manifest and files from `package_dir`.
Includes `require` section if any in the manifest file.
Puts the resulting .zip file into `acrhive_dir` if present or in the
`package_dir`.
"""
with open(manifest, 'w') as f:
fqn = 'io.murano.apps.' + app_name
mfest_copy = MANIFEST.copy()
mfest_copy['FullName'] = fqn
mfest_copy['Name'] = app_name
if hot:
mfest_copy['Format'] = 'Heat.HOT/1.0'
else:
mfest_copy['Format'] = '1.0'
mfest_copy['Classes'] = {fqn: 'mock_muranopl.yaml'}
if require:
mfest_copy['Require'] = require
if kwargs:
mfest_copy.update(kwargs)
f.write(yaml.dump(mfest_copy, default_flow_style=False))
name = app_name + '.zip'
if not archive_dir:
archive_dir = os.path.dirname(os.path.abspath(__file__))
archive_path = os.path.join(archive_dir, name)
with zipfile.ZipFile(archive_path, 'w') as zip_file:
for root, dirs, files in os.walk(package_dir):
for f in files:
zip_file.write(
os.path.join(root, f),
arcname=os.path.join(os.path.relpath(root, package_dir), f)
)
return archive_path
def compose_bundle(bundle_path, app_names):
"""Composes a murano bundle. """
bundle = {'Packages': []}
for app_name in app_names:
bundle['Packages'].append({'Name': app_name})
with open(bundle_path, 'w') as f:
f.write(json.dumps(bundle))
def glare_enabled():
return cfg.common.packages_service == "glare"
class wait_for_page_change(object):
"""Waits until current page change in a browser."""
def __init__(self, driver, timeout=30):
self.driver = driver
self.timeout = timeout
def __enter__(self):
self.old_page = self.driver.find_element_by_tag_name('html')
def page_has_loaded(self):
new_page = self.driver.find_element_by_tag_name('html')
return new_page.id != self.old_page.id
def __exit__(self, *_):
wait.WebDriverWait(self.driver, timeout=self.timeout).until(
lambda _: self.page_has_loaded()
)