157 lines
5.3 KiB
Python
157 lines
5.3 KiB
Python
import sys
|
|
|
|
from core.input import trigger, LEFT_MOUSE, RIGHT_MOUSE
|
|
from core.config import Config
|
|
from modules.core.xrandr import Module
|
|
|
|
|
|
MOCK_EXECUTE = "modules.core.xrandr.util.cli.execute"
|
|
|
|
|
|
def mock_xrandr(mocker, xrandr_output):
|
|
return mocker.patch(MOCK_EXECUTE, return_value=xrandr_output)
|
|
|
|
|
|
def assert_widgets(module, *expected_widgets):
|
|
assert len(module.widgets()) == len(expected_widgets)
|
|
|
|
for widget, (name, state, pos) in zip(module.widgets(), expected_widgets):
|
|
assert widget.name == name
|
|
assert widget.get("state") == state
|
|
assert widget.get("pos") == pos
|
|
|
|
|
|
def assert_trigger(mocker, module, widget_index, button, expected_command):
|
|
xrandr_cli = mock_xrandr(mocker, "")
|
|
widget = module.widgets()[widget_index]
|
|
trigger({"button": button, "instance": widget.id, "name": module.id})
|
|
|
|
if expected_command is None:
|
|
xrandr_cli.assert_not_called()
|
|
else:
|
|
xrandr_cli.assert_called_once_with(expected_command)
|
|
|
|
|
|
def test_autoupdate(mocker):
|
|
xrandr_cli = mock_xrandr(mocker, FULL_OUTPUT_TWO_DISPLAYS)
|
|
module = Module(Config([]), theme=None)
|
|
module.update()
|
|
xrandr_cli.assert_called_once_with("xrandr -q")
|
|
|
|
assert_widgets(module, ("eDP-1-1", "on", 0), ("HDMI-1-1", "on", 1920))
|
|
|
|
assert_trigger(mocker, module, 0, LEFT_MOUSE, "xrandr --output eDP-1-1 --off")
|
|
assert_trigger(mocker, module, 0, RIGHT_MOUSE, "xrandr --output eDP-1-1 --off")
|
|
assert_trigger(mocker, module, 1, LEFT_MOUSE, "xrandr --output HDMI-1-1 --off")
|
|
assert_trigger(mocker, module, 1, RIGHT_MOUSE, "xrandr --output HDMI-1-1 --off")
|
|
|
|
|
|
def test_display_off(mocker):
|
|
xrandr_cli = mock_xrandr(mocker, TRUNCATED_OUTPUT_DISPLAY_OFF)
|
|
module = Module(Config([]), theme=None)
|
|
module.update()
|
|
xrandr_cli.assert_called_once_with("xrandr -q")
|
|
|
|
assert_widgets(module, ("eDP-1-1", "on", 0), ("HDMI-1-1", "off", sys.maxsize))
|
|
|
|
assert_trigger(mocker, module, 0, LEFT_MOUSE, None)
|
|
assert_trigger(mocker, module, 0, RIGHT_MOUSE, None)
|
|
assert_trigger(mocker, module, 1, LEFT_MOUSE, "xrandr --output HDMI-1-1 --auto --left-of eDP-1-1")
|
|
assert_trigger(mocker, module, 1, RIGHT_MOUSE, "xrandr --output HDMI-1-1 --auto --right-of eDP-1-1")
|
|
|
|
|
|
def test_no_autoupdate(mocker):
|
|
xrandr_cli = mock_xrandr(mocker, FULL_OUTPUT_TWO_DISPLAYS)
|
|
module = Module(Config(["-p", "xrandr.autoupdate=false"]), theme=None)
|
|
module.update()
|
|
xrandr_cli.assert_called_once_with("xrandr -q")
|
|
|
|
assert_widgets(
|
|
module, ("eDP-1-1", "on", 0), ("HDMI-1-1", "on", 1920), (None, "refresh", None)
|
|
)
|
|
|
|
assert_trigger(mocker, module, 0, LEFT_MOUSE, "xrandr --output eDP-1-1 --off")
|
|
assert_trigger(mocker, module, 0, RIGHT_MOUSE, "xrandr --output eDP-1-1 --off")
|
|
assert_trigger(mocker, module, 1, LEFT_MOUSE, "xrandr --output HDMI-1-1 --off")
|
|
assert_trigger(mocker, module, 1, RIGHT_MOUSE, "xrandr --output HDMI-1-1 --off")
|
|
|
|
|
|
def test_exclude(mocker):
|
|
xrandr_cli = mock_xrandr(mocker, FULL_OUTPUT_TWO_DISPLAYS)
|
|
module = Module(Config(["-p", "xrandr.exclude=eDP"]), theme=None)
|
|
module.update()
|
|
xrandr_cli.assert_called_once_with("xrandr -q")
|
|
|
|
assert_widgets(module, ("HDMI-1-1", "on", 1920))
|
|
|
|
assert_trigger(mocker, module, 0, LEFT_MOUSE, "xrandr --output HDMI-1-1 --off")
|
|
assert_trigger(mocker, module, 0, RIGHT_MOUSE, "xrandr --output HDMI-1-1 --off")
|
|
|
|
|
|
def test_exclude_off(mocker):
|
|
xrandr_cli = mock_xrandr(mocker, TRUNCATED_OUTPUT_DISPLAY_OFF)
|
|
module = Module(Config(["-p", "xrandr.exclude=eDP"]), theme=None)
|
|
module.update()
|
|
xrandr_cli.assert_called_once_with("xrandr -q")
|
|
|
|
assert_widgets(module, ("HDMI-1-1", "off", sys.maxsize))
|
|
|
|
assert_trigger(mocker, module, 0, LEFT_MOUSE, "xrandr --output HDMI-1-1 --auto --left-of eDP-1-1")
|
|
assert_trigger(mocker, module, 0, RIGHT_MOUSE, "xrandr --output HDMI-1-1 --auto --right-of eDP-1-1")
|
|
|
|
|
|
# xrandr sample data
|
|
|
|
FULL_OUTPUT_TWO_DISPLAYS = """Screen 0: minimum 8 x 8, current 4480 x 1440, maximum 32767 x 32767
|
|
eDP-1-1 connected primary 1920x1080+0+0 344mm x 193mm
|
|
1920x1080 60.00*+ 59.93 48.00
|
|
1680x1050 59.95 59.88
|
|
1600x1024 60.17
|
|
1400x1050 59.98
|
|
1280x1024 60.02
|
|
1440x900 59.89
|
|
1280x960 60.00
|
|
1360x768 59.80 59.96
|
|
1152x864 60.00
|
|
1024x768 60.04 60.00
|
|
960x720 60.00
|
|
928x696 60.05
|
|
896x672 60.01
|
|
960x600 60.00
|
|
960x540 59.99
|
|
800x600 60.00 60.32 56.25
|
|
840x525 60.01 59.88
|
|
800x512 60.17
|
|
700x525 59.98
|
|
640x512 60.02
|
|
720x450 59.89
|
|
640x480 60.00 59.94
|
|
680x384 59.80 59.96
|
|
576x432 60.06
|
|
512x384 60.00
|
|
400x300 60.32 56.34
|
|
320x240 60.05
|
|
HDMI-1-1 connected 2560x1440+1920+0 596mm x 335mm
|
|
2560x1440 59.95*+
|
|
1920x1080 60.00 50.00 59.94
|
|
1920x1080i 60.00 50.00 59.94
|
|
1680x1050 59.88
|
|
1600x900 60.00
|
|
1280x1024 60.02
|
|
1280x800 59.91
|
|
1280x720 60.00 50.00 59.94
|
|
1024x768 60.00
|
|
800x600 60.32
|
|
720x576 50.00
|
|
720x576i 50.00
|
|
720x480 60.00 59.94
|
|
720x480i 60.00 59.94
|
|
640x480 60.00 59.94
|
|
"""
|
|
|
|
|
|
TRUNCATED_OUTPUT_DISPLAY_OFF = """eDP-1-1 connected primary 1920x1080+0+0 344mm x 193mm
|
|
1920x1080 60.00*+ 59.93 48.00
|
|
HDMI-1-1 connected
|
|
2560x1440 59.95 +
|
|
"""
|