mirror of
https://github.com/pgadmin-org/pgadmin4.git
synced 2025-01-08 15:13:10 -06:00
8857f0d179
1) String literals should not be duplicated. 2) Prefer using an optional chain expression instead, as it's more concise and easier to read. 3) Expected the Promise rejection reason to be an Error.
116 lines
3.5 KiB
Python
116 lines
3.5 KiB
Python
##########################################################################
|
|
#
|
|
# pgAdmin 4 - PostgreSQL Tools
|
|
#
|
|
# Copyright (C) 2013 - 2024, The pgAdmin Development Team
|
|
# This software is released under the PostgreSQL Licence
|
|
#
|
|
##########################################################################
|
|
|
|
import errno
|
|
import sys
|
|
import os
|
|
|
|
from datetime import datetime
|
|
|
|
from copy import deepcopy
|
|
|
|
import config as app_config
|
|
from pgadmin.utils.route import BaseTestGenerator
|
|
from regression.feature_utils.pgadmin_page import PgadminPage
|
|
from regression.python_test_utils import test_utils
|
|
from pgadmin.utils.constants import TWO_PARAM_STRING
|
|
|
|
|
|
class BaseFeatureTest(BaseTestGenerator):
|
|
CURRENT_PATH = os.path.dirname(os.path.realpath(__file__))
|
|
|
|
def setUp(self):
|
|
self.server = deepcopy(self.server)
|
|
self.server['name'] += ' Feature Tests'
|
|
if app_config.SERVER_MODE and not self.parallel_ui_tests:
|
|
self.skipTest(
|
|
"Currently, config is set to start pgadmin in server mode. "
|
|
"This test doesn't know username and password so doesn't work "
|
|
"in server mode"
|
|
)
|
|
|
|
self.page = PgadminPage(self.driver, app_config)
|
|
try:
|
|
if self.parallel_ui_tests:
|
|
self.page.login_to_app(self.server['login_details'])
|
|
test_utils.reset_layout_db()
|
|
self.page.driver.switch_to.default_content()
|
|
self.page.wait_for_app()
|
|
self.page.wait_for_spinner_to_disappear()
|
|
self.page.refresh_page()
|
|
self.page.wait_for_spinner_to_disappear()
|
|
self.before()
|
|
except Exception:
|
|
self._screenshot()
|
|
raise
|
|
|
|
def runTest(self):
|
|
# To be implemented by child classes
|
|
pass
|
|
|
|
def before(self):
|
|
# To be implemented by child classes
|
|
pass
|
|
|
|
def after(self):
|
|
# To be implemented by child classes
|
|
pass
|
|
|
|
def tearDown(self):
|
|
python3_failures = hasattr(self, '_outcome') and self.any_step_failed()
|
|
|
|
if python3_failures:
|
|
self._screenshot()
|
|
|
|
self.after()
|
|
|
|
def any_step_failed(self):
|
|
if hasattr(self._outcome, 'errors'):
|
|
for step in self._outcome.errors:
|
|
if step[1] is not None:
|
|
return True
|
|
return False
|
|
|
|
def current_test_failed(self):
|
|
all_failures = self._resultForDoCleanups.errors + \
|
|
self._resultForDoCleanups.failures
|
|
for failure in all_failures:
|
|
if failure[0] == self:
|
|
return True
|
|
return False
|
|
|
|
def _screenshot(self):
|
|
screenshots_directory = '{0}/../screenshots'.format(self.CURRENT_PATH)
|
|
screenshots_server_directory = TWO_PARAM_STRING.format(
|
|
screenshots_directory,
|
|
self.server["name"].replace(" ", "_")
|
|
)
|
|
|
|
self.ensure_directory_exists(screenshots_directory)
|
|
self.ensure_directory_exists(screenshots_server_directory)
|
|
|
|
date = datetime.now().strftime("%Y.%m.%d_%H.%M.%S")
|
|
python_version = sys.version.split(" ", maxsplit=1)[0]
|
|
|
|
self.page.driver.save_screenshot(
|
|
'{0}/{1}-{2}-Python-{3}.png'.format(
|
|
screenshots_server_directory,
|
|
self.__class__.__name__,
|
|
date,
|
|
python_version
|
|
)
|
|
)
|
|
|
|
def ensure_directory_exists(self, path):
|
|
try:
|
|
os.mkdir(path)
|
|
except OSError as e:
|
|
if e.errno != errno.EEXIST:
|
|
raise
|