feat(bolero-generator): add trace utility (#240)

camshaft · Jul 2, 2024 · 429a416 · 429a416
1 parent 8ee8a20
commit 429a416
Show file tree

Hide file tree

Showing 3 changed files with 529 additions and 0 deletions.
diff --git a/lib/bolero-generator/Cargo.toml b/lib/bolero-generator/Cargo.toml
@@ -24,3 +24,14 @@ rand_core = { version = "0.6", default-features = false }
 
 [dev-dependencies]
 rand = "0.8"
+
+[lints.rust.unexpected_cfgs]
+level = "warn"
+check-cfg = [
+  'cfg(kani)',
+  'cfg(fuzzing)',
+  'cfg(fuzzing_afl)',
+  'cfg(fuzzing_libfuzzer)',
+  'cfg(fuzzing_honggfuzz)',
+  'cfg(fuzzing_random)'
+]
diff --git a/lib/bolero-generator/src/lib.rs b/lib/bolero-generator/src/lib.rs
@@ -43,6 +43,8 @@ pub mod one_of;
 pub mod range;
 pub mod result;
 pub mod time;
+#[cfg(feature = "std")]
+pub mod trace;
 pub mod tuple;
 
 #[cfg(feature = "arbitrary")]
@@ -143,6 +145,13 @@ pub trait ValueGenerator: Sized {
             filter_map,
         }
     }
+
+    /// Traces generated values to `stderr`
+    #[cfg(feature = "std")]
+    #[inline]
+    fn trace(self) -> trace::Trace<Self> {
+        trace::Trace::new(self)
+    }
 }
 
 impl<'a, T: ValueGenerator> ValueGenerator for &'a T {