nixosTests.udisks2: port to python

This commit is contained in:
worldofpeace 2019-11-05 16:13:16 -05:00
parent 7d331eae2e
commit ceec19f682

View File

@ -1,4 +1,4 @@
import ./make-test.nix ({ pkgs, ... }: import ./make-test-python.nix ({ pkgs, ... }:
let let
@ -30,32 +30,40 @@ in
testScript = testScript =
'' ''
my $stick = $machine->stateDir . "/usbstick.img"; import lzma
system("xz -d < ${stick} > $stick") == 0 or die;
$machine->succeed("udisksctl info -b /dev/vda >&2"); with lzma.open(
$machine->fail("udisksctl info -b /dev/sda1"); "${stick}"
) as data, open(machine.state_dir + "/usbstick.img", "wb") as stick:
stick.write(data.read())
machine.succeed("udisksctl info -b /dev/vda >&2")
machine.fail("udisksctl info -b /dev/sda1")
# Attach a USB stick and wait for it to show up. # Attach a USB stick and wait for it to show up.
$machine->sendMonitorCommand("drive_add 0 id=stick,if=none,file=$stick,format=raw"); machine.send_monitor_command(
$machine->sendMonitorCommand("device_add usb-storage,id=stick,drive=stick"); f"drive_add 0 id=stick,if=none,file={stick.name},format=raw"
$machine->waitUntilSucceeds("udisksctl info -b /dev/sda1"); )
$machine->succeed("udisksctl info -b /dev/sda1 | grep 'IdLabel:.*USBSTICK'"); machine.send_monitor_command("device_add usb-storage,id=stick,drive=stick")
machine.wait_until_succeeds("udisksctl info -b /dev/sda1")
machine.succeed("udisksctl info -b /dev/sda1 | grep 'IdLabel:.*USBSTICK'")
# Mount the stick as a non-root user and do some stuff with it. # Mount the stick as a non-root user and do some stuff with it.
$machine->succeed("su - alice -c 'udisksctl info -b /dev/sda1'"); machine.succeed("su - alice -c 'udisksctl info -b /dev/sda1'")
$machine->succeed("su - alice -c 'udisksctl mount -b /dev/sda1'"); machine.succeed("su - alice -c 'udisksctl mount -b /dev/sda1'")
$machine->succeed("su - alice -c 'cat /run/media/alice/USBSTICK/test.txt'") =~ /Hello World/ or die; machine.succeed(
$machine->succeed("su - alice -c 'echo foo > /run/media/alice/USBSTICK/bar.txt'"); "su - alice -c 'cat /run/media/alice/USBSTICK/test.txt' | grep -q 'Hello World'"
)
machine.succeed("su - alice -c 'echo foo > /run/media/alice/USBSTICK/bar.txt'")
# Unmounting the stick should make the mountpoint disappear. # Unmounting the stick should make the mountpoint disappear.
$machine->succeed("su - alice -c 'udisksctl unmount -b /dev/sda1'"); machine.succeed("su - alice -c 'udisksctl unmount -b /dev/sda1'")
$machine->fail("[ -d /run/media/alice/USBSTICK ]"); machine.fail("[ -d /run/media/alice/USBSTICK ]")
# Remove the USB stick. # Remove the USB stick.
$machine->sendMonitorCommand("device_del stick"); machine.send_monitor_command("device_del stick")
$machine->waitUntilFails("udisksctl info -b /dev/sda1"); machine.wait_until_fails("udisksctl info -b /dev/sda1")
$machine->fail("[ -e /dev/sda ]"); machine.fail("[ -e /dev/sda ]")
''; '';
}) })