fix last tests

gosub-io · Jan 14, 2025 · 4199050 · 4199050
1 parent fc43016
commit 4199050
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/benches/tree_iterator.rs b/benches/tree_iterator.rs
@@ -38,12 +38,12 @@ fn wikipedia_main_page(c: &mut Criterion) {
     let mut stream = ByteStream::new(Encoding::UTF8, None);
     let _ = stream.read_from_file(html_file);
 
-    let doc_handle: DocumentHandle<Config> = DocumentBuilderImpl::new_document(None);
-    let _ = Html5Parser::parse_document(&mut stream, doc_handle.clone(), None);
+    let mut doc = <DocumentBuilderImpl as DocumentBuilder<Config>>::new_document(None);
+    let _ = Html5Parser::<Config>::parse_document(&mut stream, &mut doc, None);
 
     group.bench_function("wikipedia main page", |b| {
         b.iter(|| {
-            let tree_iterator = TreeIterator::new(doc_handle.clone());
+            let tree_iterator = TreeIterator::<Config>::new(&doc);
             let _ = tree_iterator.collect::<Vec<NodeId>>();
         })
     });
@@ -63,12 +63,12 @@ fn stackoverflow_home(c: &mut Criterion) {
     let mut bytestream = ByteStream::new(Encoding::UTF8, None);
     let _ = bytestream.read_from_file(html_file);
 
-    let doc_handle: DocumentHandle<Config> = DocumentBuilderImpl::new_document(None);
-    let _ = Html5Parser::parse_document(&mut bytestream, doc_handle.clone(), None);
+    let mut doc = <DocumentBuilderImpl as DocumentBuilder<Config>>::new_document(None);
+    let _ = Html5Parser::<Config>::parse_document(&mut bytestream, &mut doc, None);
 
     group.bench_function("stackoverflow home", |b| {
         b.iter(|| {
-            let tree_iterator = TreeIterator::new(doc_handle.clone());
+            let tree_iterator = TreeIterator::<Config>::new(&doc);
             let _ = tree_iterator.collect::<Vec<NodeId>>();
         })
     });