depot/third_party/nixpkgs/nixos/tests/drawterm.nix

{ system, pkgs }:
let
  tests = {
    xorg = {
      node = { pkgs, ... }: {
        imports = [ ./common/user-account.nix ./common/x11.nix ];
        services.xserver.enable = true;
        services.xserver.displayManager.sessionCommands = ''
          ${pkgs.drawterm}/bin/drawterm -g 1024x768 &
        '';
        test-support.displayManager.auto.user = "alice";
      };
      systems = [ "x86_64-linux" "aarch64-linux" ];
    };
    wayland = {
      node = { pkgs, ... }: {
        imports = [ ./common/wayland-cage.nix ];
        services.cage.program = "${pkgs.drawterm-wayland}/bin/drawterm";
      };
      systems = [ "x86_64-linux" ];
    };
  };

  mkTest = name: machine:
    import ./make-test-python.nix ({ pkgs, ... }: {
      inherit name;

      nodes = { "${name}" = machine; };

      meta = with pkgs.lib.maintainers; {
        maintainers = [ moody ];
      };

      enableOCR = true;

      testScript = ''
        @polling_condition
        def drawterm_running():
            machine.succeed("pgrep drawterm")

        # cage is a bit wonky here.
        # it seems to lag behind drawing
        # and somehow needs a single input character
        # in order to get the first prompt to show up.
        # This is not present in any other compositor
        # as far as I know, and after spending a couple
        # hours with the upstream source trying to deduce
        # how to perhaps fix it, I figured just polling is OK.
        @polling_condition
        def cpu_shown_up():
            machine.send_chars(".")
            machine.wait_for_text("cpu", 1)

        start_all()

        machine.wait_for_unit("graphical.target")
        drawterm_running.wait() # type: ignore[union-attr]
        cpu_shown_up.wait() # type: ignore[union-attr]
        machine.send_chars("cpu\n")
        machine.wait_for_text("auth")
        machine.send_chars("cpu\n")
        machine.wait_for_text("ending")
        machine.screenshot("out.png")
      '';

    });
  mkTestOn = systems: name: machine:
    if pkgs.lib.elem system systems then mkTest name machine
    else { ... }: { };
in
builtins.mapAttrs (k: v: mkTestOn v.systems k v.node { inherit system; }) tests