Fix tests failing with missing /lib(64)

phylum-dev · Sep 26, 2023 · d03c9ba · d03c9ba
1 parent 9e98f7a
commit d03c9ba
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tests/exec.rs b/tests/exec.rs
@@ -1,4 +1,5 @@
 use std::fs;
+use std::path::PathBuf;
 use std::process::Command;
 
 use birdcage::{Birdcage, Exception, Sandbox};
@@ -7,8 +8,12 @@ fn main() {
     let mut birdcage = Birdcage::new().unwrap();
     birdcage.add_exception(Exception::ExecuteAndRead("/usr/bin/true".into())).unwrap();
     birdcage.add_exception(Exception::ExecuteAndRead("/usr/lib".into())).unwrap();
-    birdcage.add_exception(Exception::ExecuteAndRead("/lib64".into())).unwrap();
-    birdcage.add_exception(Exception::ExecuteAndRead("/lib".into())).unwrap();
+    if PathBuf::from("/lib64").exists() {
+        birdcage.add_exception(Exception::ExecuteAndRead("/lib64".into())).unwrap();
+    }
+    if PathBuf::from("/lib64").exists() {
+        birdcage.add_exception(Exception::ExecuteAndRead("/lib".into())).unwrap();
+    }
     birdcage.lock().unwrap();
 
     // Check for success when executing `true`.