test(core): add Spark CDI policy baseline

Signed-off-by: Evan Lezar <elezar@nvidia.com>

Author: elezar

crates/openshell-core/src/cdi_linux.rs

@@ -455,14 +455,9 @@ mod tests {
         additional_gids: &'a [u32],
     }
 
-    #[test]
-    fn orin_cdi_spec_matches_generated_policy_baseline() {
+    fn assert_cdi_spec_matches_generated_policy_baseline(spec: &str, expected: &str) {
         let dir = tempfile::tempdir().unwrap();
-        write_spec(
-            dir.path(),
-            "nvidia.yaml",
-            include_str!("../testdata/orin-nvidia.yaml"),
-        );
+        write_spec(dir.path(), "nvidia.yaml", spec);
 
         let requirements = resolve_with_kind(
             &context(dir.path(), &["nvidia.com/gpu=all"]),
@@ -478,9 +473,22 @@ mod tests {
         };
         let generated = serde_json::to_string_pretty(&baseline).unwrap() + "\n";
 
-        assert_eq!(
-            generated,
-            include_str!("../testdata/orin-nvidia-policy-baseline.json")
+        assert_eq!(generated, expected);
+    }
+
+    #[test]
+    fn orin_cdi_spec_matches_generated_policy_baseline() {
+        assert_cdi_spec_matches_generated_policy_baseline(
+            include_str!("../testdata/orin-nvidia.yaml"),
+            include_str!("../testdata/orin-nvidia-policy-baseline.json"),
+        );
+    }
+
+    #[test]
+    fn spark_cdi_spec_matches_generated_policy_baseline() {
+        assert_cdi_spec_matches_generated_policy_baseline(
+            include_str!("../testdata/spark-nvidia.yaml"),
+            include_str!("../testdata/spark-nvidia-policy-baseline.json"),
         );
     }
 

crates/openshell-core/testdata/spark-nvidia-policy-baseline.json

@@ -0,0 +1,21 @@
+{
+  "device_node_paths": [
+    "/dev/dri/card0",
+    "/dev/dri/renderD128",
+    "/dev/nvidia-modeset",
+    "/dev/nvidia-uvm",
+    "/dev/nvidia-uvm-tools",
+    "/dev/nvidia0",
+    "/dev/nvidiactl"
+  ],
+  "library_directories": [
+    "/usr/lib/aarch64-linux-gnu",
+    "/usr/lib/aarch64-linux-gnu/nvidia/xorg",
+    "/usr/lib/aarch64-linux-gnu/vdpau"
+  ],
+  "read_write_mount_paths": [],
+  "additional_gids": [
+    44,
+    993
+  ]
+}