From d64e5cdca35b5fad7c504f615357a7afe6d9c49e Mon Sep 17 00:00:00 2001
From: fidgetingbits <fidgetingbits@users.noreply.github.com>
Date: Wed, 10 Dec 2025 07:25:02 +0800
Subject: [PATCH] feat(mdadm): add test for mdadm with extraArgs

---
 example/mdadm-raid5-args.nix | 66 ++++++++++++++++++++++++++++++++++++
 tests/mdadm-raid5-args.nix   | 14 ++++++++
 2 files changed, 80 insertions(+)
 create mode 100644 example/mdadm-raid5-args.nix
 create mode 100644 tests/mdadm-raid5-args.nix

diff --git a/example/mdadm-raid5-args.nix b/example/mdadm-raid5-args.nix
new file mode 100644
index 0000000..ddda14e
--- /dev/null
+++ b/example/mdadm-raid5-args.nix
@@ -0,0 +1,66 @@
+{
+  disko.devices = {
+    disk = {
+      disk1 = {
+        type = "disk";
+        device = "/dev/my-disk";
+        content = {
+          type = "gpt";
+          partitions = {
+            boot = {
+              size = "1M";
+              type = "EF02"; # for grub MBR
+            };
+            mdadm = {
+              size = "100%";
+              content = {
+                type = "mdraid";
+                name = "raid5";
+              };
+            };
+          };
+        };
+      };
+      disk2 = {
+        type = "disk";
+        device = "/dev/my-disk2";
+        content = {
+          type = "gpt";
+          partitions = {
+            boot = {
+              size = "1M";
+              type = "EF02"; # for grub MBR
+            };
+            mdadm = {
+              size = "100%";
+              content = {
+                type = "mdraid";
+                name = "raid5";
+              };
+            };
+          };
+        };
+      };
+    };
+    mdadm = {
+      raid5 = {
+        type = "mdadm";
+        level = 5;
+        content = {
+          type = "gpt";
+          partitions = {
+            primary = {
+              size = "100%";
+              content = {
+                type = "filesystem";
+                format = "ext4";
+                mountpoint = "/";
+              };
+            };
+          };
+        };
+        extraArgs = [ "--assume-clean" ];
+      };
+    };
+  };
+}
diff --git a/tests/mdadm-raid5-args.nix b/tests/mdadm-raid5-args.nix
new file mode 100644
index 0000000..867fbe9
--- /dev/null
+++ b/tests/mdadm-raid5-args.nix
@@ -0,0 +1,14 @@
+{
+  pkgs ? import <nixpkgs> { },
+  diskoLib ? pkgs.callPackage ../lib { },
+}:
+diskoLib.testLib.makeDiskoTest {
+  inherit pkgs;
+  name = "mdadm-raid5-args";
+  disko-config = ../example/mdadm-raid5-args.nix;
+  extraTestScript = ''
+    machine.succeed("test -b /dev/md/raid5");
+    machine.succeed("mountpoint /");
+  '';
+  efi = false;
+}