Fix the probe_raw test on no_std

cuviper · Mar 28, 2024 · e01e120 · e01e120
1 parent e0a2240
commit e01e120
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/src/tests.rs b/src/tests.rs
@@ -136,13 +136,17 @@ fn probe_constant() {
 #[test]
 fn probe_raw() {
     let ac = AutoCfg::for_test().unwrap();
+    let prefix = if ac.no_std { "#![no_std]\n" } else { "" };
+    let f = |s| format!("{}{}", prefix, s);
 
     // This attribute **must** be used at the crate level.
-    assert!(ac.probe_raw("#![no_builtins]").is_ok());
+    assert!(ac.probe_raw(&f("#![no_builtins]")).is_ok());
 
-    assert!(ac.probe_raw("#![deny(dead_code)] fn x() {}").is_err());
-    assert!(ac.probe_raw("#![allow(dead_code)] fn x() {}").is_ok());
-    assert!(ac.probe_raw("#![deny(dead_code)] pub fn x() {}").is_ok());
+    assert!(ac.probe_raw(&f("#![deny(dead_code)] fn x() {}")).is_err());
+    assert!(ac.probe_raw(&f("#![allow(dead_code)] fn x() {}")).is_ok());
+    assert!(ac
+        .probe_raw(&f("#![deny(dead_code)] pub fn x() {}"))
+        .is_ok());
 }
 
 #[test]