about summary refs log tree commit diff
path: root/nixos/tests/grow-partition.nix
blob: 344910848dca80c7ba3b777712745fae244f7500 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
{ lib, ... }:

let
  rootFslabel = "external";
  rootFsDevice = "/dev/disk/by-label/${rootFslabel}";

  externalModule = partitionTableType: { config, lib, pkgs, ... }: {
    virtualisation.directBoot.enable = false;
    virtualisation.mountHostNixStore = false;
    virtualisation.useEFIBoot = partitionTableType == "efi";

    # This stops the qemu-vm module from overriding the fileSystems option
    # with virtualisation.fileSystems.
    virtualisation.fileSystems = lib.mkForce { };


    boot.loader.grub.enable = true;
    boot.loader.grub.efiSupport = partitionTableType == "efi";
    boot.loader.grub.efiInstallAsRemovable = partitionTableType == "efi";
    boot.loader.grub.device = if partitionTableType == "efi" then "nodev" else "/dev/vda";

    boot.growPartition = true;

    fileSystems = {
      "/".device = rootFsDevice;
    };

    system.build.diskImage = import ../lib/make-disk-image.nix {
      inherit config lib pkgs;
      label = rootFslabel;
      inherit partitionTableType;
      format = "raw";
      bootSize = "128M";
      additionalSpace = "0M";
      copyChannel = false;
    };
  };
in
{
  name = "grow-partition";

  meta.maintainers = with lib.maintainers; [ arianvp ];

  nodes = {
    efi = externalModule "efi";
    legacy = externalModule "legacy";
    legacyGPT = externalModule "legacy+gpt";
    hybrid = externalModule "hybrid";
  };


  testScript = { nodes, ... }:
    lib.concatLines (lib.mapAttrsToList (name: node: ''
    import os
    import subprocess
    import tempfile
    import shutil

    tmp_disk_image = tempfile.NamedTemporaryFile()

    shutil.copyfile("${node.system.build.diskImage}/nixos.img", tmp_disk_image.name)

    subprocess.run([
      "${node.virtualisation.qemu.package}/bin/qemu-img",
      "resize",
      "-f",
      "raw",
      tmp_disk_image.name,
      "+32M",
    ])

    # Set NIX_DISK_IMAGE so that the qemu script finds the right disk image.
    os.environ['NIX_DISK_IMAGE'] = tmp_disk_image.name

    ${name}.wait_for_unit("growpart.service")
    systemd_growpart_logs = ${name}.succeed("journalctl --boot --unit growpart.service")
    assert "CHANGED" in systemd_growpart_logs
    ${name}.succeed("systemctl restart growpart.service")
    systemd_growpart_logs = ${name}.succeed("journalctl --boot --unit growpart.service")
    assert "NOCHANGE" in systemd_growpart_logs

    '') nodes);
}