From c5adc83694838ecd0a042263949a9d5e950f5878 Mon Sep 17 00:00:00 2001
From: sinanmohd <sinan@sinanmohd.com>
Date: Wed, 28 Aug 2024 20:26:45 +0530
Subject: nixosTests/transitive-unbuilt-3: init

---
 nixos/tests/all-tests.nix | 34 ++++++++++++++++++++++++++++++++++
 nixos/tests/dsl.nix       |  2 +-
 2 files changed, 35 insertions(+), 1 deletion(-)

diff --git a/nixos/tests/all-tests.nix b/nixos/tests/all-tests.nix
index f7e8325..4825d4a 100644
--- a/nixos/tests/all-tests.nix
+++ b/nixos/tests/all-tests.nix
@@ -10,6 +10,20 @@ let
     nodes.d.inputs.c = { }; # d->c
   };
 
+  #  A
+  #  |
+  #  B
+  #  |
+  #  C
+  transitive.dag = {
+    nodes.a = {
+      goal = true;
+      inputs.b = { };
+    };
+    nodes.b.inputs.c = { };
+    nodes.c = {};
+  };
+
   #   A   B   C     D
   #   \   |  /      |
   #     U  V        W
@@ -89,6 +103,26 @@ builtins.mapAttrs
       ];
     };
 
+    transitive-unbuilt-3 = {
+      imports = [
+        {
+          dag = {
+            test.unconstrained.builds = 3;
+
+            constraints.builds = 2;
+            test.constrained.builds = 0;
+
+            nodes = {
+              a.test.needed = true;
+              b.test.needed = true;
+              c.test.needed = true;
+            };
+          };
+        }
+        transitive
+      ];
+    };
+
     sunset-unbuilt-7 = {
       imports = [
         {
diff --git a/nixos/tests/dsl.nix b/nixos/tests/dsl.nix
index 18d7ed4..1fc1fbc 100644
--- a/nixos/tests/dsl.nix
+++ b/nixos/tests/dsl.nix
@@ -249,7 +249,7 @@ in
         else:
           expect_n_chosen_nodes = None
 
-        if expect_chosen_nodes or expect_n_chosen_nodes is not None:
+        if expect_chosen_nodes or expect_n_chosen_builds is not None or expect_n_chosen_downloads is not None:
           evanix = subprocess.run(evanix_args, check=True, stdout=subprocess.PIPE)
           evanix_output = evanix.stdout.decode("utf-8")
           evanix_choices = parse_evanix_dry_run(evanix_output)
-- 
cgit v1.2.3