test: support cargo bench

umijs · Jul 16, 2024 · d448586 · d448586
1 parent ffd5763
commit d448586
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/rust-toolchain.toml b/rust-toolchain.toml
@@ -0,0 +1,2 @@
+[toolchain]
+channel = "nightly-2024-07-16"
diff --git a/src/lib.rs b/src/lib.rs
@@ -1,6 +1,10 @@
+#![feature(test)]
+
 #[macro_use]
 extern crate napi_derive;
 
+extern crate test;
+
 use kuchikiki::{parse_html, traits::*};
 use node_repr::NodeRepr;
 
@@ -16,3 +20,15 @@ pub fn parse(html: String) -> NodeRepr {
     node_ref: document_node,
   }
 }
+
+#[cfg(test)]
+mod tests {
+  use super::parse;
+  use test::Bencher;
+
+  #[bench]
+  fn bench_parse(b: &mut Bencher) {
+    let html = include_str!("../test/jquery.html").to_string();
+    b.iter(|| parse(html.clone()));
+  }
+}