Fix resizing selector set

Author: kornelski

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "lol_html"
-version = "2.8.0"
+version = "2.8.1"
 authors = ["Ivan Nikulin <inikulin@cloudflare.com, ifaaan@gmail.com>"]
 license = "BSD-3-Clause"
 description = "Streaming HTML rewriter/parser with CSS selector-based API"

src/selectors_vm/match_info.rs

@@ -79,7 +79,7 @@ impl DenseHashSet {
     fn resize(&mut self, new_len: usize) -> &mut [u32] {
         let bits = self.slice_mut();
         let mut new = Vec::with_capacity(new_len);
-        new.copy_from_slice(bits);
+        new.extend_from_slice(bits);
         let cap = new.capacity();
         new.resize(cap, 0);
         *self = Self::Heap(new.into_boxed_slice());

src/selectors_vm/tests.rs

@@ -111,7 +111,7 @@ mod vm_tests {
         ($selectors:expr) => {{
             let mut ast = Ast::default();
 
-            for (selector, match_id) in $selectors.iter().zip(0..) {
+            for (selector, match_id) in $selectors.into_iter().zip(0..) {
                 ast.add_selector(&selector.parse().unwrap(), match_id);
             }
 
@@ -133,7 +133,7 @@ mod vm_tests {
 
                         {
                             let mut match_handler = |m: MatchInfo| {
-                                assert_eq!(m.with_content, $expectation.should_match_with_content);
+                                assert_eq!(m.with_content, $expectation.should_match_with_content, "with_content");
                                 matched_ids.insert(m.match_id);
                             };
 
@@ -168,7 +168,7 @@ mod vm_tests {
                             }
                         }
 
-                        assert_eq!(matched_ids, $expectation.matched_ids);
+                        assert_eq!(matched_ids, $expectation.matched_ids, "{:?}", matched_ids.iter().collect::<Vec<_>>());
                     }
                     _ => panic!("Start tag expected"),
                 }
@@ -1219,6 +1219,26 @@ mod vm_tests {
         );
     }
 
+    #[test]
+    fn lots_of_selectors() {
+        let mut vm = create_vm!(
+            ('a'..='z').map(|c| c.to_string())
+            .chain(('A'..='Z').map(|c| format!("#{c}")))
+            .chain(('0'..='9').map(|c| format!("[data{c}]")))
+        );
+
+        exec_for_start_tag_and_assert!(
+            vm,
+            "<a id=Z>",
+            Namespace::Html,
+            Expectation {
+                should_bailout: true,
+                should_match_with_content: true,
+                matched_ids: DenseHashSet::from([0, 51]),
+            }
+        );
+    }
+
     /// Test that match_handler is called exactly once per matched id,
     /// even when the same match_ids can be reached through multiple instruction paths.
     /// For example, `"span, [foo$=bar]"` is one selector (match_id 0) with two