23 lines
770 B
Python
23 lines
770 B
Python
from unittest.mock import patch
|
|
|
|
from tools.computer_use import cua_backend
|
|
|
|
|
|
_VAR = "CUA_DRIVER_RS_ENABLE_WAYLAND"
|
|
|
|
|
|
def _child_env(base_env, native_wayland):
|
|
config = {"computer_use": {"native_wayland": native_wayland}}
|
|
with patch("hermes_cli.config.load_config", return_value=config), \
|
|
patch.object(cua_backend.sys, "platform", "linux"):
|
|
return cua_backend.cua_driver_child_env(base_env)
|
|
|
|
|
|
def test_configured_native_wayland_reaches_linux_wayland_child():
|
|
assert _child_env({"WAYLAND_DISPLAY": "wayland-1"}, True)[_VAR] == "1"
|
|
|
|
|
|
def test_native_wayland_not_injected_without_wayland_display_or_opt_in():
|
|
assert _VAR not in _child_env({"DISPLAY": ":0"}, True)
|
|
assert _VAR not in _child_env({"WAYLAND_DISPLAY": "wayland-1"}, False)
|