test: extend Nokogiri decorator tests to NodeSets

flavorjones · Jan 17, 2021 · 07e89a5 · 07e89a5
1 parent 82f0c9e
commit 07e89a5
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/test/unit/test_api.rb b/test/unit/test_api.rb
@@ -55,6 +55,13 @@ class UnitTestApi < Loofah::TestCase
       node_set.scrub!(:strip)
     end
 
+    it "scrubs fragment nodesets" do
+      doc = Loofah.fragment(HTML)
+      assert(node_set = doc.css("div"))
+      assert_instance_of Nokogiri::XML::NodeSet, node_set
+      node_set.scrub!(:strip)
+    end
+
     it "exposes serialize_root on HTML::DocumentFragment" do
       doc = Loofah.fragment(HTML)
       assert_equal HTML, doc.serialize_root.to_html
@@ -110,6 +117,13 @@ class UnitTestApi < Loofah::TestCase
       assert(node = doc.at_css("div"))
       node.scrub!(:strip)
     end
+
+    it "scrubs document nodesets" do
+      doc = Loofah.xml_document(XML)
+      assert(node_set = doc.css("div"))
+      assert_instance_of Nokogiri::XML::NodeSet, node_set
+      node_set.scrub!(:strip)
+    end
   end
 
   private