spec/support/shared/element/capabilities/with_node.rb in arachni-1.4 vs spec/support/shared/element/capabilities/with_node.rb in arachni-1.5

@@ -6,11 +6,11 @@
     end
 
     describe '#node' do
         it 'returns the set node' do
             node = with_node.node
-            expect(node.is_a?( Nokogiri::XML::Element )).to be_truthy
-            expect(node.to_s).to eq(Nokogiri::HTML.fragment( with_node.source.dup ).to_s)
+            expect(node.is_a?( Arachni::Parser::Nodes::Element )).to be_truthy
+            expect(node.to_s).to eq(Arachni::Parser.parse_fragment( with_node.source ).to_s)
         end
     end
 
     describe '#dup' do
         let(:dupped) { with_node.dup }