diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 17:32:43 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 17:32:43 +0000 |
commit | 6bf0a5cb5034a7e684dcc3500e841785237ce2dd (patch) | |
tree | a68f146d7fa01f0134297619fbe7e33db084e0aa /testing/condprofile/condprof/desktop.py | |
parent | Initial commit. (diff) | |
download | thunderbird-6bf0a5cb5034a7e684dcc3500e841785237ce2dd.tar.xz thunderbird-6bf0a5cb5034a7e684dcc3500e841785237ce2dd.zip |
Adding upstream version 1:115.7.0.upstream/1%115.7.0upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to '')
-rw-r--r-- | testing/condprofile/condprof/desktop.py | 90 |
1 files changed, 90 insertions, 0 deletions
diff --git a/testing/condprofile/condprof/desktop.py b/testing/condprofile/condprof/desktop.py new file mode 100644 index 0000000000..7f8fe4e309 --- /dev/null +++ b/testing/condprofile/condprof/desktop.py @@ -0,0 +1,90 @@ +# This Source Code Form is subject to the terms of the Mozilla Public +# License, v. 2.0. If a copy of the MPL was not distributed with this +# file, You can obtain one at http://mozilla.org/MPL/2.0/. +import os +import contextlib + +import attr +from arsenic.services import Geckodriver, free_port, subprocess_based_service + +from condprof.util import ( + BaseEnv, + latest_nightly, + logger, + get_version, + get_current_platform, + DEFAULT_PREFS, +) + + +@attr.s +class DesktopGeckodriver(Geckodriver): + async def start(self): + port = free_port() + await self._check_version() + logger.info("Running Webdriver on port %d" % port) + logger.info("Running Marionette on port 2828") + pargs = [ + self.binary, + "--log", + "trace", + "--port", + str(port), + "--marionette-port", + "2828", + ] + try: + return await subprocess_based_service( + pargs, f"http://localhost:{port}", self.log_file + ) + except ProcessLookupError as e: + return await subprocess_based_service( + pargs.extend(["--host", "127.0.0.1"]), + f"http://localhost:{port}", + self.log_file, + ) + + +@contextlib.contextmanager +def dummy_device(*args, **kw): + yield None + + +class DesktopEnv(BaseEnv): + def get_target_platform(self): + return get_current_platform() + + def get_device(self, *args, **kw): + return dummy_device(*args, **kw) + + @contextlib.contextmanager + def get_browser(self): + with latest_nightly(self.firefox) as binary: + self.firefox = os.path.abspath(binary) + if not os.path.exists(self.firefox): + raise IOError(self.firefox) + yield + + def get_browser_args(self, headless, prefs=None): + final_prefs = dict(DEFAULT_PREFS) + if prefs is not None: + final_prefs.update(prefs) + options = ["--allow-downgrade", "-profile", self.profile] + if headless: + options.append("-headless") + args = {"moz:firefoxOptions": {"args": options}} + if self.firefox is not None: + args["moz:firefoxOptions"]["binary"] = self.firefox + args["moz:firefoxOptions"]["prefs"] = final_prefs + args["moz:firefoxOptions"]["log"] = {"level": "trace"} + return args + + def get_browser_version(self): + try: + return get_version(self.firefox) + except Exception: + logger.error("Could not get Firefox version", exc_info=True) + return "unknown" + + def get_geckodriver(self, log_file): + return DesktopGeckodriver(binary=self.geckodriver, log_file=log_file) |