test: extend Nokogiri decorator tests to NodeSets

Author: flavorjones

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