diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/btrfs-subvolumes.nix | 11 | ||||
-rw-r--r-- | tests/default.nix | 16 | ||||
-rw-r--r-- | tests/lib.nix | 48 | ||||
-rw-r--r-- | tests/luks-lvm.nix | 10 | ||||
-rw-r--r-- | tests/lvm-raid.nix | 12 | ||||
-rw-r--r-- | tests/mdadm.nix | 10 | ||||
-rw-r--r-- | tests/test.nix | 38 | ||||
-rw-r--r-- | tests/zfs-over-legacy.nix | 12 | ||||
-rw-r--r-- | tests/zfs.nix | 27 |
9 files changed, 146 insertions, 38 deletions
diff --git a/tests/btrfs-subvolumes.nix b/tests/btrfs-subvolumes.nix new file mode 100644 index 0000000..3ff4386 --- /dev/null +++ b/tests/btrfs-subvolumes.nix @@ -0,0 +1,11 @@ +{ pkgs ? (import <nixpkgs> { }) +, makeDiskoTest ? (pkgs.callPackage ./lib.nix { }).makeDiskoTest +}: +makeDiskoTest { + disko-config = import ../example/btrfs-subvolumes.nix; + extraTestScript = '' + machine.succeed("test -e /mnt/test"); + machine.succeed("btrfs subvolume list /mnt | grep -qs 'path test$'"); + ''; +} + diff --git a/tests/default.nix b/tests/default.nix new file mode 100644 index 0000000..525f936 --- /dev/null +++ b/tests/default.nix @@ -0,0 +1,16 @@ +{ makeTest ? import <nixpkgs/nixos/tests/make-test-python.nix> +, pkgs ? (import <nixpkgs> { }) +}@args: +let + lib = pkgs.lib; + makeDiskoTest = (pkgs.callPackage ./lib.nix { inherit makeTest; }).makeDiskoTest; + allTestFilenames = + builtins.map (lib.removeSuffix ".nix") ( + builtins.filter + (x: lib.hasSuffix ".nix" x && x != "default.nix" && x != "lib.nix") + (lib.attrNames (builtins.readDir ./.)) + ); + + allTests = lib.genAttrs (allTestFilenames) (test: import (./. + "/${test}.nix") { inherit makeDiskoTest; }); +in +allTests diff --git a/tests/lib.nix b/tests/lib.nix new file mode 100644 index 0000000..2ec42e6 --- /dev/null +++ b/tests/lib.nix @@ -0,0 +1,48 @@ +{ pkgs ? (import <nixpkgs> { }) +, makeTest ? import <nixpkgs/nixos/tests/make-test-python.nix> +, ... +}: +{ + makeDiskoTest = + { disko-config + , extraTestScript + , extraConfig ? { } + }: + let + lib = pkgs.lib; + makeTest' = args: + makeTest args { + inherit pkgs; + inherit (pkgs) system; + }; + tsp-create = pkgs.writeScript "create" ((pkgs.callPackage ../. { }).create disko-config); + tsp-mount = pkgs.writeScript "mount" ((pkgs.callPackage ../. { }).mount disko-config); + num-disks = builtins.length (builtins.filter (x: builtins.match "vd." x == [ ]) (lib.attrNames disko-config.content)); + in + makeTest' { + name = "disko"; + + nodes.machine = + { config, pkgs, modulesPath, ... }: + + { + imports = [ + (modulesPath + "/profiles/installation-device.nix") + (modulesPath + "/profiles/base.nix") + ]; + + # speed-up eval + documentation.enable = false; + + virtualisation.emptyDiskImages = builtins.genList (_: 512) num-disks; + } // extraConfig; + + testScript = '' + machine.succeed("echo 'secret' > /tmp/secret.key"); + machine.succeed("${tsp-create}"); + machine.succeed("${tsp-mount}"); + machine.succeed("${tsp-mount}"); # verify that the command is idempotent + ${extraTestScript} + ''; + }; +} diff --git a/tests/luks-lvm.nix b/tests/luks-lvm.nix new file mode 100644 index 0000000..d59f3f4 --- /dev/null +++ b/tests/luks-lvm.nix @@ -0,0 +1,10 @@ +{ pkgs ? (import <nixpkgs> { }) +, makeDiskoTest ? (pkgs.callPackage ./lib.nix { }).makeDiskoTest +}: +makeDiskoTest { + disko-config = import ../example/luks-lvm.nix; + extraTestScript = '' + machine.succeed("cryptsetup isLuks /dev/vdb2"); + machine.succeed("mountpoint /mnt/home"); + ''; +} diff --git a/tests/lvm-raid.nix b/tests/lvm-raid.nix new file mode 100644 index 0000000..102dcd7 --- /dev/null +++ b/tests/lvm-raid.nix @@ -0,0 +1,12 @@ +{ pkgs ? (import <nixpkgs> { }) +, makeDiskoTest ? (pkgs.callPackage ./lib.nix { }).makeDiskoTest +}: +makeDiskoTest { + disko-config = import ../example/lvm-raid.nix; + extraTestScript = '' + machine.succeed("mountpoint /mnt/home"); + ''; + extraConfig = { + boot.kernelModules = [ "dm-raid" "dm-mirror" ]; + }; +} diff --git a/tests/mdadm.nix b/tests/mdadm.nix new file mode 100644 index 0000000..bd57a00 --- /dev/null +++ b/tests/mdadm.nix @@ -0,0 +1,10 @@ +{ pkgs ? (import <nixpkgs> { }) +, makeDiskoTest ? (pkgs.callPackage ./lib.nix { }).makeDiskoTest +}: +makeDiskoTest { + disko-config = import ../example/mdadm.nix; + extraTestScript = '' + machine.succeed("test -b /dev/md/raid1"); + machine.succeed("mountpoint /mnt/raid"); + ''; +} diff --git a/tests/test.nix b/tests/test.nix deleted file mode 100644 index 1384590..0000000 --- a/tests/test.nix +++ /dev/null @@ -1,38 +0,0 @@ -{ makeTest ? import <nixpkgs/nixos/tests/make-test-python.nix> -, pkgs ? (import <nixpkgs> {}) -}: -let - makeTest' = args: - makeTest args { - inherit pkgs; - inherit (pkgs) system; - }; - disko-config = import ../example/raid.nix; - tsp-create = pkgs.writeScript "create" ((pkgs.callPackage ../. {}).create disko-config); - tsp-mount = pkgs.writeScript "mount" ((pkgs.callPackage ../. {}).mount disko-config); -in makeTest' { - name = "disko"; - - nodes.machine = - { config, pkgs, modulesPath, ... }: - - { - imports = [ - (modulesPath + "/profiles/installation-device.nix") - (modulesPath + "/profiles/base.nix") - ]; - - # speed-up eval - documentation.enable = false; - - virtualisation.emptyDiskImages = [ 512 512 ]; - }; - - testScript = '' - machine.succeed("echo 'secret' > /tmp/secret.key"); - machine.succeed("${tsp-create}"); - machine.succeed("${tsp-mount}"); - machine.succeed("${tsp-mount}"); # verify that the command is idempotent - machine.succeed("test -b /dev/md/raid1"); - ''; -} diff --git a/tests/zfs-over-legacy.nix b/tests/zfs-over-legacy.nix new file mode 100644 index 0000000..13f86a9 --- /dev/null +++ b/tests/zfs-over-legacy.nix @@ -0,0 +1,12 @@ +{ pkgs ? (import <nixpkgs> { }) +, makeDiskoTest ? (pkgs.callPackage ./lib.nix { }).makeDiskoTest +}: +makeDiskoTest { + disko-config = import ../example/zfs-over-legacy.nix; + extraTestScript = '' + machine.succeed("test -e /mnt/zfs_fs"); + machine.succeed("mountpoint /mnt"); + machine.succeed("mountpoint /mnt/zfs_fs"); + ''; +} + diff --git a/tests/zfs.nix b/tests/zfs.nix new file mode 100644 index 0000000..5940c4d --- /dev/null +++ b/tests/zfs.nix @@ -0,0 +1,27 @@ +{ pkgs ? (import <nixpkgs> { }) +, makeDiskoTest ? (pkgs.callPackage ./lib.nix { }).makeDiskoTest +}: +makeDiskoTest { + disko-config = import ../example/zfs.nix; + extraTestScript = '' + machine.succeed("test -b /dev/zvol/zroot/zfs_testvolume"); + + def assert_property(ds, property, expected_value): + out = machine.succeed(f"zfs get -H {property} {ds} -o value").rstrip() + assert ( + out == expected_value + ), f"Expected {property}={expected_value} on {ds}, got: {out}" + + assert_property("zroot", "compression", "lz4") + assert_property("zroot/zfs_fs", "compression", "lz4") + assert_property("zroot", "com.sun:auto-snapshot", "false") + assert_property("zroot/zfs_fs", "com.sun:auto-snapshot", "true") + assert_property("zroot/zfs_testvolume", "volsize", "10M") + assert_property("zroot/zfs_unmounted_fs", "mountpoint", "none") + + machine.succeed("mountpoint /mnt"); + machine.succeed("mountpoint /mnt/zfs_fs"); + machine.succeed("mountpoint /mnt/zfs_legacy_fs"); + machine.succeed("mountpoint /mnt/ext4onzfs"); + ''; +} |