nixos/tests/startx: test imperative setup

nixos/tests/all-tests.nix

@@ -1153,7 +1153,7 @@ in {
   systemd-userdbd = handleTest ./systemd-userdbd.nix {};
   systemd-homed = handleTest ./systemd-homed.nix {};
   systemtap = handleTest ./systemtap.nix {};
-  startx = runTest ./startx.nix;
+  startx = import ./startx.nix { inherit pkgs runTest; };
   taler = handleTest ./taler {};
   tandoor-recipes = handleTest ./tandoor-recipes.nix {};
   tandoor-recipes-script-name = handleTest ./tandoor-recipes-script-name.nix {};

nixos/tests/startx.nix

@@ -1,8 +1,10 @@
-{ lib, ... }:
+{ pkgs, runTest }:
 
 {
+
+  declarative = runTest {
     name = "startx";
-  meta.maintainers = with lib.maintainers; [ rnhmjoj ];
+    meta.maintainers = with pkgs.lib.maintainers; [ rnhmjoj ];
 
     nodes.machine =
       { pkgs, ... }:
@@ -105,4 +107,94 @@
           machine.fail(f"{sysu} is-active urxvtd")
           machine.fail(f"{sysu} is-active xss-lock")
     '';
+  };
+
+  imperative = runTest {
+    name = "startx-imperative";
+    meta.maintainers = with pkgs.lib.maintainers; [ rnhmjoj ];
+
+    nodes.machine =
+      { pkgs, ... }:
+      {
+        services.getty.autologinUser = "root";
+        programs.bash.promptInit = "PS1='# '";
+
+        # startx+twm setup
+        services.xserver = {
+          enable = true;
+          windowManager.twm.enable = true;
+          displayManager.startx.enable = true;
+          displayManager.startx.generateScript = false;
+        };
+
+        # enable some user services
+        security.polkit.enable = true;
+        services.urxvtd.enable = true;
+        programs.xss-lock.enable = true;
+      };
+
+    testScript = ''
+      import textwrap
+
+      sysu = "env XDG_RUNTIME_DIR=/run/user/0 systemctl --user";
+      prompt = "# "
+
+      with subtest("Wait for the autologin"):
+          machine.wait_until_tty_matches("1", prompt)
+
+      with subtest("Setup dotfiles"):
+          machine.execute(textwrap.dedent("""
+            cat <<EOF > ~/.Xresources
+            urxvt*foreground: #9b9081
+            urxvt*background: #181b20
+            urxvt*scrollBar:  false
+            urxvt*title:      myterm
+            urxvt*geometry:   20x20+40+40
+            EOF
+            cat <<EOF > ~/.twmrc
+            "Return" = meta : all : f.exec "urxvtc"
+            "Escape" = meta : all : f.quit
+            EOF
+            cat <<EOF > ~/.xinitrc
+            xrdb -load ~/.Xresources
+            xsetroot -solid '#343d46'
+            xsetroot -cursor_name trek
+            # start user services
+            systemctl --user import-environment DISPLAY XDG_SESSION_ID
+            systemctl --user start nixos-fake-graphical-session.target
+            # run the window manager
+            twm
+            # stop services and all subprocesses
+            systemctl --user stop nixos-fake-graphical-session.target
+            EOF
+          """))
+
+      with subtest("Can start the X server"):
+          machine.send_chars("startx\n")
+          machine.wait_for_x()
+
+      with subtest("Graphical services are running"):
+          machine.succeed(f"{sysu} is-active graphical-session.target")
+          machine.succeed(f"{sysu} is-active urxvtd")
+          machine.succeed(f"{sysu} is-active xss-lock")
+
+      with subtest("Can interact with the WM"):
+          machine.wait_until_succeeds("pgrep twm")
+          # spawn a terminal
+          machine.send_key("alt-ret")
+          machine.wait_for_window("myterm")
+          machine.screenshot("screenshot.png")
+
+      with subtest("Can stop the X server"):
+          # kill the WM
+          machine.send_key("alt-esc")
+          machine.wait_until_tty_matches("1", prompt)
+
+      with subtest("Graphical session has stopped"):
+          machine.fail(f"{sysu} is-active graphical-session.target")
+          machine.fail(f"{sysu} is-active urxvtd")
+          machine.fail(f"{sysu} is-active xss-lock")
+    '';
+  };
+
 }