nixos/tests/udisks2.nix

   1 import ./make-test-python.nix ({ pkgs, ... }:
   2
   3 let
   4
   5   # FIXME: 404s
   6   stick = pkgs.fetchurl {
   7     url = "https://nixos.org/~eelco/nix/udisks-test.img.xz";
   8     sha256 = "0was1xgjkjad91nipzclaz5biv3m4b2nk029ga6nk7iklwi19l8b";
   9   };
  10
  11 in
  12
  13 {
  14   name = "udisks2";
  15   meta = with pkgs.lib.maintainers; {
  16     maintainers = [ ];
  17   };
  18
  19   nodes.machine =
  20     { ... }:
  21     { services.udisks2.enable = true;
  22       imports = [ ./common/user-account.nix ];
  23
  24       security.polkit.extraConfig =
  25         ''
  26           polkit.addRule(function(action, subject) {
  27             if (subject.user == "alice") return "yes";
  28           });
  29         '';
  30     };
  31
  32   testScript =
  33     ''
  34       import lzma
  35
  36       machine.systemctl("start udisks2")
  37       machine.wait_for_unit("udisks2.service")
  38
  39       with lzma.open(
  40           "${stick}"
  41       ) as data, open(machine.state_dir / "usbstick.img", "wb") as stick:
  42           stick.write(data.read())
  43
  44       machine.succeed("udisksctl info -b /dev/vda >&2")
  45       machine.fail("udisksctl info -b /dev/sda1")
  46
  47       # Attach a USB stick and wait for it to show up.
  48       machine.send_monitor_command(
  49           f"drive_add 0 id=stick,if=none,file={stick.name},format=raw"
  50       )
  51       machine.send_monitor_command("device_add usb-storage,id=stick,drive=stick")
  52       machine.wait_until_succeeds("udisksctl info -b /dev/sda1")
  53       machine.succeed("udisksctl info -b /dev/sda1 | grep 'IdLabel:.*USBSTICK'")
  54
  55       # Mount the stick as a non-root user and do some stuff with it.
  56       machine.succeed("su - alice -c 'udisksctl info -b /dev/sda1'")
  57       machine.succeed("su - alice -c 'udisksctl mount -b /dev/sda1'")
  58       machine.succeed(
  59           "su - alice -c 'cat /run/media/alice/USBSTICK/test.txt' | grep -q 'Hello World'"
  60       )
  61       machine.succeed("su - alice -c 'echo foo > /run/media/alice/USBSTICK/bar.txt'")
  62
  63       # Unmounting the stick should make the mountpoint disappear.
  64       machine.succeed("su - alice -c 'udisksctl unmount -b /dev/sda1'")
  65       machine.fail("[ -d /run/media/alice/USBSTICK ]")
  66
  67       # Remove the USB stick.
  68       machine.send_monitor_command("device_del stick")
  69       machine.wait_until_fails("udisksctl info -b /dev/sda1")
  70       machine.fail("[ -e /dev/sda ]")
  71     '';
  72
  73 })