nixosTests: allow extending testScript without overriding it

1 files changed, 13 insertions, 6 deletions

diff --git a/nixos/tests/all-tests.nix b/nixos/tests/all-tests.nix
index adeaf65..b79b198 100644
--- a/nixos/tests/all-tests.nix
+++ b/nixos/tests/all-tests.nix
@@ -42,13 +42,20 @@ builtins.mapAttrs
       {
         inherit name;
         imports = value.imports ++ [ dsl ];
-        testScript = ''
-          start_all()
-          substituter.wait_for_unit("nix-serve.service")
-          builder.succeed("dag-test")
-        '';
+        testScript =
+          value.testScriptPre or ""
+          + ''
+            start_all()
+            substituter.wait_for_unit("nix-serve.service")
+            builder.succeed("dag-test")
+          ''
+          + value.testScriptPost or "";
       }
-      // builtins.removeAttrs value [ "imports" ]
+      // builtins.removeAttrs value [
+        "imports"
+        "testScriptPre"
+        "testScriptPost"
+      ]
     )
   )
   {