depot/third_party/nixpkgs/nixos/tests/swap-file-btrfs.nix

import ./make-test-python.nix (
  { lib, ... }:
  {
    name = "swap-file-btrfs";

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

    nodes.machine =
      { pkgs, ... }:
      {
        virtualisation.useDefaultFilesystems = false;

        virtualisation.rootDevice = "/dev/vda";

        boot.initrd.postDeviceCommands = ''
          ${pkgs.btrfs-progs}/bin/mkfs.btrfs --label root /dev/vda
        '';

        virtualisation.fileSystems = {
          "/" = {
            device = "/dev/disk/by-label/root";
            fsType = "btrfs";
          };
        };

        swapDevices = [
          {
            device = "/var/swapfile";
            size = 1; # 1MiB.
          }
        ];
      };

    testScript = ''
      machine.wait_for_unit('var-swapfile.swap')
      # Ensure the swap file creation script ran to completion without failing when creating the swap file
      machine.fail("systemctl is-failed --quiet mkswap-var-swapfile.service")
      machine.succeed("stat --file-system --format=%T /var/swapfile | grep btrfs")
      # First run. Auto creation.
      machine.succeed("swapon --show | grep /var/swapfile")

      machine.shutdown()
      machine.start()

      # Second run. Use it as-is.
      machine.wait_for_unit('var-swapfile.swap')
      # Ensure the swap file creation script ran to completion without failing when the swap file already exists
      machine.fail("systemctl is-failed --quiet mkswap-var-swapfile.service")
      machine.succeed("swapon --show | grep /var/swapfile")
    '';
  }
)