test(fuzz): Verify error rendering doesn't panic

This will ensure the span we return is within valid UTF-8 byte offsets
toml-rs · Apr 19, 2024 · 73cbee7 · 73cbee7
1 parent ed26084
commit 73cbee7
Showing 1 changed file with 18 additions and 14 deletions.
diff --git a/crates/toml_edit_fuzz/parse_document.rs b/crates/toml_edit_fuzz/parse_document.rs
@@ -5,19 +5,23 @@ use toml_edit::DocumentMut;
 libfuzzer_sys::fuzz_target!(|data| {
     if let Ok(data) = std::str::from_utf8(data) {
         println!("parsing: {data:?}");
-        let doc = data.parse::<DocumentMut>();
-        if let Ok(doc) = doc {
-            let toml = doc.to_string();
-            println!("parsing: {toml:?}");
-            let doc = toml.parse::<DocumentMut>();
-            assert!(
-                doc.is_ok(),
-                "Failed to parse `doc.to_string()`: {}\n```\n{}\n```",
-                doc.unwrap_err(),
-                toml
-            );
-            let doc = doc.unwrap();
-            assert_eq!(doc.to_string(), toml);
-        }
+        let doc = match data.parse::<DocumentMut>() {
+            Ok(doc) => doc,
+            Err(err) => {
+                println!("{err}");
+                return;
+            }
+        };
+        let toml = doc.to_string();
+        println!("parsing: {toml:?}");
+        let doc = toml.parse::<DocumentMut>();
+        assert!(
+            doc.is_ok(),
+            "Failed to parse `doc.to_string()`: {}\n```\n{}\n```",
+            doc.unwrap_err(),
+            toml
+        );
+        let doc = doc.unwrap();
+        assert_eq!(doc.to_string(), toml);
     }
 });