134 lines
3.9 KiB
Nix
134 lines
3.9 KiB
Nix
{ system ? builtins.currentSystem
|
|
, config ? { }
|
|
, pkgs ? import ../.. { inherit system config; }
|
|
}:
|
|
|
|
with import ../lib/testing-python.nix { inherit system pkgs; };
|
|
with pkgs.lib;
|
|
|
|
let
|
|
# A testScript fragment that prepares a disk with some empty, unpartitioned
|
|
# space. and uses it to boot the test with. Takes a single argument `machine`
|
|
# from which the diskImage is extraced.
|
|
useDiskImage = machine: ''
|
|
import os
|
|
import shutil
|
|
import subprocess
|
|
import tempfile
|
|
|
|
tmp_disk_image = tempfile.NamedTemporaryFile()
|
|
|
|
shutil.copyfile("${machine.system.build.diskImage}/nixos.img", tmp_disk_image.name)
|
|
|
|
subprocess.run([
|
|
"${pkgs.qemu}/bin/qemu-img",
|
|
"resize",
|
|
"-f",
|
|
"raw",
|
|
tmp_disk_image.name,
|
|
"+32M",
|
|
])
|
|
|
|
# Fix the GPT table by moving the backup table to the end of the enlarged
|
|
# disk image. This is necessary because we increased the size of the disk
|
|
# before. The disk needs to be a raw disk because sgdisk can only run on
|
|
# raw images.
|
|
subprocess.run([
|
|
"${pkgs.gptfdisk}/bin/sgdisk",
|
|
"--move-second-header",
|
|
tmp_disk_image.name,
|
|
])
|
|
|
|
# Set NIX_DISK_IMAGE so that the qemu script finds the right disk image.
|
|
os.environ['NIX_DISK_IMAGE'] = tmp_disk_image.name
|
|
'';
|
|
|
|
common = { config, pkgs, lib, ... }: {
|
|
virtualisation.useDefaultFilesystems = false;
|
|
virtualisation.fileSystems = {
|
|
"/" = {
|
|
device = "/dev/vda2";
|
|
fsType = "ext4";
|
|
};
|
|
};
|
|
|
|
# systemd-repart operates on disks with a partition table. The qemu module,
|
|
# however, creates separate filesystem images without a partition table, so
|
|
# we have to create a disk image manually.
|
|
#
|
|
# This creates two partitions, an ESP mounted on /dev/vda1 and the root
|
|
# partition mounted on /dev/vda2
|
|
system.build.diskImage = import ../lib/make-disk-image.nix {
|
|
inherit config pkgs lib;
|
|
# Use a raw format disk so that it can be resized before starting the
|
|
# test VM.
|
|
format = "raw";
|
|
# Keep the image as small as possible but leave some room for changes.
|
|
bootSize = "32M";
|
|
additionalSpace = "0M";
|
|
# GPT with an EFI System Partition is the typical use case for
|
|
# systemd-repart because it does not support MBR.
|
|
partitionTableType = "efi";
|
|
# We do not actually care much about the content of the partitions, so we
|
|
# do not need a bootloader installed.
|
|
installBootLoader = false;
|
|
# Improve determinism by not copying a channel.
|
|
copyChannel = false;
|
|
};
|
|
};
|
|
in
|
|
{
|
|
basic = makeTest {
|
|
name = "systemd-repart";
|
|
meta.maintainers = with maintainers; [ nikstur ];
|
|
|
|
nodes.machine = { config, pkgs, ... }: {
|
|
imports = [ common ];
|
|
|
|
boot.initrd.systemd.enable = true;
|
|
|
|
boot.initrd.systemd.repart.enable = true;
|
|
systemd.repart.partitions = {
|
|
"10-root" = {
|
|
Type = "linux-generic";
|
|
};
|
|
};
|
|
};
|
|
|
|
testScript = { nodes, ... }: ''
|
|
${useDiskImage nodes.machine}
|
|
|
|
machine.start()
|
|
machine.wait_for_unit("multi-user.target")
|
|
|
|
systemd_repart_logs = machine.succeed("journalctl --boot --unit systemd-repart.service")
|
|
assert "Growing existing partition 1." in systemd_repart_logs
|
|
'';
|
|
};
|
|
|
|
after-initrd = makeTest {
|
|
name = "systemd-repart-after-initrd";
|
|
meta.maintainers = with maintainers; [ nikstur ];
|
|
|
|
nodes.machine = { config, pkgs, ... }: {
|
|
imports = [ common ];
|
|
|
|
systemd.repart.enable = true;
|
|
systemd.repart.partitions = {
|
|
"10-root" = {
|
|
Type = "linux-generic";
|
|
};
|
|
};
|
|
};
|
|
|
|
testScript = { nodes, ... }: ''
|
|
${useDiskImage nodes.machine}
|
|
|
|
machine.start()
|
|
machine.wait_for_unit("multi-user.target")
|
|
|
|
systemd_repart_logs = machine.succeed("journalctl --unit systemd-repart.service")
|
|
assert "Growing existing partition 1." in systemd_repart_logs
|
|
'';
|
|
};
|
|
}
|