test/automated-pytest-suite/testfixtures/horizon.py

88 lines
2.3 KiB
Python

#
# Copyright (c) 2019 Wind River Systems, Inc.
#
# SPDX-License-Identifier: Apache-2.0
#
import os
import datetime
from pytest import fixture
from utils.horizon.pages import loginpage
from utils.horizon import video_recorder
from utils.horizon.helper import HorizonDriver
from utils.tis_log import LOG
from consts import horizon
from consts.auth import Tenant
from consts.proj_vars import ProjVar
@fixture(scope="session")
def driver(request):
driver_ = HorizonDriver.get_driver()
def teardown():
HorizonDriver.quit_driver()
request.addfinalizer(teardown)
return driver_
@fixture(scope='function')
def admin_home_pg(driver, request):
return __login_base(request=request, driver=driver,
auth_info=Tenant.get('admin_platform'))
@fixture(scope='function')
def admin_home_pg_container(driver, request):
return __login_base(request=request, driver=driver,
auth_info=Tenant.get('admin'))
@fixture(scope='function')
def tenant_home_pg_container(driver, request):
return __login_base(request=request, driver=driver,
auth_info=Tenant.get_primary())
def __login_base(request, driver, auth_info, port=None):
horizon.test_result = False
if not auth_info:
auth_info = Tenant.get_primary()
user = auth_info['user']
password = auth_info['password']
project = auth_info['tenant']
if not port and not auth_info.get('platform'):
port = 31000
gmttime = datetime.datetime.utcnow().strftime("%Y%m%d%H%M%S")
video_path = ProjVar.get_var('LOG_DIR') + '/horizon/' + \
str(gmttime) + '.mp4'
recorder = video_recorder.VideoRecorder(1920, 1080, os.environ['DISPLAY'],
video_path)
recorder.start()
home_pg = None
try:
LOG.fixture_step('Login as {}'.format(user))
login_pg = loginpage.LoginPage(driver, port=port)
login_pg.go_to_target_page()
home_pg = login_pg.login(user=user, password=password)
home_pg.change_project(name=project)
finally:
def teardown():
if home_pg:
LOG.fixture_step('Logout')
home_pg.log_out()
recorder.stop()
if horizon.test_result:
recorder.clear()
request.addfinalizer(teardown)
return home_pg