disko/tests/lib.nix

{ 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}
    '';
  };
}