From ca46628491546f2e7e846c114e748bae7b527595 Mon Sep 17 00:00:00 2001
From: sinanmohd <sinan@sinanmohd.com>
Date: Mon, 12 Aug 2024 23:17:33 +0530
Subject: nixosTests/dsl: make sure cache value is valid

---
 nixos/tests/dsl.nix | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

(limited to 'nixos')

diff --git a/nixos/tests/dsl.nix b/nixos/tests/dsl.nix
index 341207c..4d59bcb 100644
--- a/nixos/tests/dsl.nix
+++ b/nixos/tests/dsl.nix
@@ -30,14 +30,14 @@ let
       };
       options.cache = lib.mkOption {
         type = lib.types.enum [
-          "none"
+          "unbuilt"
           "remote"
           "local"
         ];
         description = ''
           Whether the dependency is pre-built and available in the local /nix/store ("local"), can be substituted ("remote"), or has to be built ("none")
         '';
-        default = "none";
+        default = "unbuilt";
       };
       options.inputs = lib.mkOption {
         type = lib.types.attrsOf (lib.types.submodule Dependency);
@@ -175,12 +175,15 @@ in
 
         for name, node in nodes.items():
           error_msg = f"Wrong plan for {name}"
+          action = drv_to_action.get(name, None)
           if node["cache"] == "local":
-            assert name not in drv_to_action, error_msg
+            assert action is None, error_msg
           elif node["cache"] == "remote":
-            assert drv_to_action.get(name, None) == "fetch", error_msg
+            assert action == "fetch", error_msg
           elif node["cache"] == "unbuilt":
-            assert drv_to_action.get(name, None) == "build", error_msg
+            assert action == "build", error_msg
+          else:
+            raise AssertionError('cache is not in [ "local", "remote", "unbuilt" ]')
 
         need_dls, need_builds = set(), set()
         for name, node in nodes.items():
-- 
cgit v1.2.3