mirror of
https://github.com/pgadmin-org/pgadmin4.git
synced 2024-11-29 12:03:52 -06:00
81 lines
2.7 KiB
Python
81 lines
2.7 KiB
Python
##########################################################################
|
|
#
|
|
# pgAdmin 4 - PostgreSQL Tools
|
|
#
|
|
# Copyright (C) 2013 - 2021, The pgAdmin Development Team
|
|
# This software is released under the PostgreSQL Licence
|
|
#
|
|
##########################################################################
|
|
import os
|
|
import subprocess
|
|
import signal
|
|
import random
|
|
|
|
import time
|
|
from selenium.common.exceptions import WebDriverException
|
|
|
|
|
|
class AppStarter:
|
|
""" Helper for starting the full pgadmin4 app and loading the page via
|
|
selenium
|
|
"""
|
|
|
|
def __init__(self, driver, app_config):
|
|
self.driver = driver
|
|
self.app_config = app_config
|
|
|
|
def start_app(self):
|
|
""" This function start the subprocess to start pgAdmin app """
|
|
random_server_port = str(random.randint(10000, 65535))
|
|
env = {
|
|
"PGADMIN_INT_PORT": random_server_port,
|
|
"SQLITE_PATH": str(self.app_config.TEST_SQLITE_PATH)
|
|
}
|
|
env.update(os.environ)
|
|
|
|
# Add OS check for pass value for 'preexec_fn'
|
|
self.pgadmin_process = subprocess.Popen(
|
|
["python", "pgAdmin4.py"],
|
|
shell=False,
|
|
preexec_fn=None if os.name == 'nt' else os.setsid,
|
|
stderr=open(os.devnull, 'w'),
|
|
env=env
|
|
)
|
|
|
|
def launch_browser(retry_count):
|
|
try:
|
|
self.driver.get(
|
|
"http://" + self.app_config.DEFAULT_SERVER + ":" +
|
|
random_server_port
|
|
)
|
|
|
|
except WebDriverException:
|
|
# In case of WebDriverException sleep for 1 second and retry
|
|
# again. Retry 10 times and if still app will not start then
|
|
# raise exception.
|
|
time.sleep(1)
|
|
if retry_count < 60:
|
|
retry_count = retry_count + 1
|
|
launch_browser(retry_count)
|
|
else:
|
|
raise RuntimeError('Unable to start python server even '
|
|
'after retrying 60 times.')
|
|
|
|
if self.driver is not None:
|
|
launch_browser(0)
|
|
else:
|
|
return "http://" + self.app_config.DEFAULT_SERVER + ":" \
|
|
+ random_server_port
|
|
|
|
def stop_app(self):
|
|
""" This function stop the started app by killing process """
|
|
if self.driver is not None:
|
|
self.driver.quit()
|
|
# os.killpg supported in Mac and Unix as this function not supported in
|
|
# Windows
|
|
try:
|
|
os.killpg(os.getpgid(self.pgadmin_process.pid), signal.SIGTERM)
|
|
except AttributeError:
|
|
# os.kill is supported by Windows
|
|
os.kill(self.pgadmin_process.pid, signal.SIGTERM)
|