64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
import logging
|
|
import time
|
|
import typing
|
|
|
|
import configs.testpath
|
|
from scripts.utils import local_system
|
|
|
|
_PROCESS_NAME = '_squishserver'
|
|
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
|
|
class SquishServer:
|
|
|
|
def __init__(
|
|
self,
|
|
host: str = '127.0.0.1',
|
|
port: int = 4322
|
|
):
|
|
self.path = configs.testpath.SQUISH_DIR / 'bin' / 'squishserver'
|
|
self.config = configs.testpath.ROOT / 'squish_server.ini'
|
|
self.host = host
|
|
self.port = port
|
|
|
|
def start(self):
|
|
cmd = [
|
|
f'"{self.path}"',
|
|
'--configfile', str(self.config),
|
|
f'--host={self.host}',
|
|
f'--port={self.port}',
|
|
]
|
|
local_system.execute(cmd)
|
|
try:
|
|
local_system.wait_for_started(_PROCESS_NAME)
|
|
except AssertionError as err:
|
|
_logger.info(err)
|
|
local_system.execute(cmd, check=True)
|
|
|
|
def stop(self, attempt: int = 2):
|
|
local_system.run([self.path, '--stop'])
|
|
try:
|
|
local_system.wait_for_close(_PROCESS_NAME, 2)
|
|
except AssertionError as err:
|
|
_logger.debug(err)
|
|
if attempt:
|
|
time.sleep(1)
|
|
self.stop(attempt-1)
|
|
else:
|
|
raise err
|
|
|
|
# https://doc-snapshots.qt.io/squish/cli-squishserver.html
|
|
def configuring(self, action: str, options: typing.Union[int, str, list]):
|
|
local_system.run(
|
|
[f'"{self.path}"', '--configfile', str(self.config), '--config', action, ' '.join(options)])
|
|
|
|
def add_executable_aut(self, aut_id, app_dir):
|
|
self.configuring('addAUT', [aut_id, f'"{app_dir}"'])
|
|
|
|
def add_attachable_aut(self, aut_id: str, port: int):
|
|
self.configuring('addAttachableAUT', [aut_id, f'localhost:{port}'])
|
|
|
|
def set_aut_timeout(self, value: int = configs.timeouts.PROCESS_TIMEOUT_SEC):
|
|
self.configuring('setAUTTimeout', [str(value)])
|