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

migrate more APP_HARNESS env vars to new config system

Browse Source
This commit is contained in:
Benedikt Bartscher 2024-10-28 22:58:35 +01:00
parent f16273fa11
commit e1ec51a589
No known key found for this signature in database
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
reflex/config.py
View File

@ -494,6 +494,12 @@ class EnvironmentVariables:
# Whether to run app harness tests in headless mode.
APP_HARNESS_HEADLESS: EnvVar[bool] = env_var(False)
# Which app harness driver to use.
APP_HARNESS_DRIVER: EnvVar[str] = env_var("Chrome")
# Arguments to pass to the app harness driver.
APP_HARNESS_DRIVER_ARGS: EnvVar[str] = env_var("")
environment = EnvironmentVariables()

4
reflex/testing.py
View File

@ -620,7 +620,7 @@ class AppHarness:
if environment.APP_HARNESS_HEADLESS.get:
want_headless = True
if driver_clz is None:
requested_driver = os.environ.get("APP_HARNESS_DRIVER", "Chrome")
requested_driver = environment.APP_HARNESS_DRIVER.get
driver_clz = getattr(webdriver, requested_driver)
if driver_options is None:
driver_options = getattr(webdriver, f"{requested_driver}Options")()
@ -642,7 +642,7 @@ class AppHarness:
driver_options.add_argument("headless")
if driver_options is None:
raise RuntimeError(f"Could not determine options for {driver_clz}")
if args := os.environ.get("APP_HARNESS_DRIVER_ARGS"):
if args := environment.APP_HARNESS_DRIVER_ARGS.get:
for arg in args.split(","):
driver_options.add_argument(arg)
if driver_option_args is not None: