about summary refs log tree commit diff
path: root/nixos/tests/ydotool.nix
blob: 818ac6f2d50de82cba71a1c6b1d6354acd295376 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
import ./make-test-python.nix (
  { pkgs, lib, ... }:
  let
    textInput = "This works.";
    inputBoxText = "Enter input";
    inputBox = pkgs.writeShellScript "zenity-input" ''
      ${lib.getExe pkgs.gnome.zenity} --entry --text '${inputBoxText}:' > /tmp/output &
    '';
  in
  {
    name = "ydotool";

    meta = {
      maintainers = with lib.maintainers; [
        OPNA2608
        quantenzitrone
      ];
    };

    nodes = {
      headless =
        { config, ... }:
        {
          imports = [ ./common/user-account.nix ];

          users.users.alice.extraGroups = [ "ydotool" ];

          programs.ydotool.enable = true;

          services.getty.autologinUser = "alice";
        };

      x11 =
        { config, ... }:
        {
          imports = [
            ./common/user-account.nix
            ./common/auto.nix
            ./common/x11.nix
          ];

          users.users.alice.extraGroups = [ "ydotool" ];

          programs.ydotool.enable = true;

          test-support.displayManager.auto = {
            enable = true;
            user = "alice";
          };

          services.xserver.windowManager.dwm.enable = true;
          services.displayManager.defaultSession = lib.mkForce "none+dwm";
        };

      wayland =
        { config, ... }:
        {
          imports = [ ./common/user-account.nix ];

          services.cage = {
            enable = true;
            user = "alice";
          };

          programs.ydotool.enable = true;

          services.cage.program = inputBox;
        };
    };

    enableOCR = true;

    testScript =
      { nodes, ... }:
      ''
        def as_user(cmd: str):
          """
          Return a shell command for running a shell command as a specific user.
          """
          return f"sudo -u alice -i {cmd}"

        start_all()

        # Headless
        headless.wait_for_unit("multi-user.target")
        headless.wait_for_text("alice")
        headless.succeed(as_user("ydotool type 'echo ${textInput} > /tmp/output'")) # text input
        headless.succeed(as_user("ydotool key 28:1 28:0")) # text input
        headless.screenshot("headless_input")
        headless.wait_for_file("/tmp/output")
        headless.wait_until_succeeds("grep '${textInput}' /tmp/output") # text input

        # X11
        x11.wait_for_x()
        x11.execute(as_user("${inputBox}"))
        x11.wait_for_text("${inputBoxText}")
        x11.succeed(as_user("ydotool type '${textInput}'")) # text input
        x11.screenshot("x11_input")
        x11.succeed(as_user("ydotool mousemove -a 400 110")) # mouse input
        x11.succeed(as_user("ydotool click 0xC0")) # mouse input
        x11.wait_for_file("/tmp/output")
        x11.wait_until_succeeds("grep '${textInput}' /tmp/output") # text input

        # Wayland
        wayland.wait_for_unit("graphical.target")
        wayland.wait_for_text("${inputBoxText}")
        wayland.succeed("ydotool type '${textInput}'") # text input
        wayland.screenshot("wayland_input")
        wayland.succeed("ydotool mousemove -a 100 100") # mouse input
        wayland.succeed("ydotool click 0xC0") # mouse input
        wayland.wait_for_file("/tmp/output")
        wayland.wait_until_succeeds("grep '${textInput}' /tmp/output") # text input
      '';
  }
)