laipeng/reflex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

AppHarness: support headless mode and driver selection (#1963)

Browse Source
This commit is contained in:
Masen Furer 2023-10-13 12:53:20 -07:00 committed by GitHub
parent 2c4ffc879f
commit d0cb5b07e7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
reflex/testing.py
View File

@ -397,7 +397,23 @@ class AppHarness:
)
if self.frontend_url is None:
raise RuntimeError("Frontend is not running.")
driver = driver_clz() if driver_clz is not None else webdriver.Chrome()
want_headless = False
options = None
if os.environ.get("APP_HARNESS_HEADLESS"):
want_headless = True
if driver_clz is None:
requested_driver = os.environ.get("APP_HARNESS_DRIVER", "Chrome")
driver_clz = getattr(webdriver, requested_driver)
if driver_clz is webdriver.Chrome and want_headless:
options = webdriver.ChromeOptions()
options.add_argument("--headless=new")
elif driver_clz is webdriver.Firefox and want_headless:
options = webdriver.FirefoxOptions()
options.add_argument("-headless")
elif driver_clz is webdriver.Edge and want_headless:
options = webdriver.EdgeOptions()
options.add_argument("headless")
driver = driver_clz(options=options) # type: ignore
driver.get(self.frontend_url)
self._frontends.append(driver)
return driver