From 4346558fd4ce8ef2c3ce3ce690d6e4ca6846ce9b Mon Sep 17 00:00:00 2001
From: Michael Hoang <enzime@users.noreply.github.com>
Date: Tue, 19 Dec 2023 23:44:00 +1100
Subject: [PATCH] tests: add test for running ZFS as not the rootfs

---
 example/non-root-zfs.nix | 79 ++++++++++++++++++++++++++++++++++++++++
 tests/non-root-zfs.nix   | 26 +++++++++++++
 2 files changed, 105 insertions(+)
 create mode 100644 example/non-root-zfs.nix
 create mode 100644 tests/non-root-zfs.nix

diff --git a/example/non-root-zfs.nix b/example/non-root-zfs.nix
new file mode 100644
index 0000000..7434e62
--- /dev/null
+++ b/example/non-root-zfs.nix
@@ -0,0 +1,79 @@
+{
+  disko.devices = {
+    disk = {
+      x = {
+        type = "disk";
+        device = "/dev/sdx";
+        content = {
+          type = "gpt";
+          partitions = {
+            ESP = {
+              size = "64M";
+              type = "EF00";
+              content = {
+                type = "filesystem";
+                format = "vfat";
+                mountpoint = "/boot";
+              };
+            };
+            root = {
+              size = "100%";
+              content = {
+                type = "filesystem";
+                format = "ext4";
+                mountpoint = "/";
+              };
+            };
+          };
+        };
+      };
+      y = {
+        type = "disk";
+        device = "/dev/sdy";
+        content = {
+          type = "gpt";
+          partitions = {
+            zfs = {
+              size = "100%";
+              content = {
+                type = "zfs";
+                pool = "storage";
+              };
+            };
+          };
+        };
+      };
+      z = {
+        type = "disk";
+        device = "/dev/sdz";
+        content = {
+          type = "gpt";
+          partitions = {
+            zfs = {
+              size = "100%";
+              content = {
+                type = "zfs";
+                pool = "storage";
+              };
+            };
+          };
+        };
+      };
+    };
+    zpool = {
+      storage = {
+        type = "zpool";
+        mode = "mirror";
+        mountpoint = "/storage";
+
+        datasets = {
+          dataset = {
+            type = "zfs_fs";
+            mountpoint = "/storage/dataset";
+          };
+        };
+      };
+    };
+  };
+}
+
diff --git a/tests/non-root-zfs.nix b/tests/non-root-zfs.nix
new file mode 100644
index 0000000..e4fe24c
--- /dev/null
+++ b/tests/non-root-zfs.nix
@@ -0,0 +1,26 @@
+{ pkgs ? import <nixpkgs> { }
+, diskoLib ? pkgs.callPackage ../lib { }
+}:
+diskoLib.testLib.makeDiskoTest {
+  inherit pkgs;
+  name = "non-root-zfs";
+  disko-config = ../example/non-root-zfs.nix;
+  extraInstallerConfig.networking.hostId = "8425e349";
+  extraSystemConfig.networking.hostId = "8425e349";
+  postDisko = ''
+    machine.succeed("mountpoint /mnt/storage")
+    machine.succeed("mountpoint /mnt/storage/dataset")
+
+    filesystem = machine.execute("stat --file-system --format=%T /mnt/storage")[1].rstrip()
+    print(f"/mnt/storage {filesystem=}")
+    assert filesystem == "zfs", "/mnt/storage is not ZFS"
+  '';
+  extraTestScript = ''
+    machine.succeed("mountpoint /storage")
+    machine.succeed("mountpoint /storage/dataset")
+
+    filesystem = machine.execute("stat --file-system --format=%T /storage")[1].rstrip()
+    print(f"/mnt/storage {filesystem=}")
+    assert filesystem == "zfs", "/storage is not ZFS"
+  '';
+}