Finding nodes via XPath

lib/webcat/culerity_driver.rb

@@ -3,7 +3,20 @@ require 'rack'
 require 'net/http'
 
 class Webcat::Driver::Culerity
-  Response = Struct.new(:body)
+  class Node < Struct.new(:node)
+    def text
+      node.text
+    end
+    
+    def attribute(name)
+      value = if name.to_sym == :class
+        node.class_name
+      else
+        node.send(name.to_sym)
+      end
+      return value if value and not value.empty?
+    end
+  end
   
   attr_reader :app, :rack_server
 
@@ -30,6 +43,10 @@ class Webcat::Driver::Culerity
   def body
     browser.html
   end
+  
+  def find(selector)
+    browser.elements_by_xpath(selector).map { |node| Node.new(node) }
+  end
 
 private
 

lib/webcat/rack_test_driver.rb

@@ -1,6 +1,18 @@
 require 'rack/test'
+require 'nokogiri'
 
 class Webcat::Driver::RackTest
+  class Node < Struct.new(:node)
+    def text
+      node.text
+    end
+    
+    def attribute(name)
+      value = node.attributes[name.to_s]
+      return value.to_s if value
+    end
+  end
+  
   include ::Rack::Test::Methods
   attr_reader :app
 
@@ -18,4 +30,14 @@ class Webcat::Driver::RackTest
   def body
     response.body
   end
+  
+  def find(selector)
+    html.xpath(selector).map { |node| Node.new(node) }
+  end
+  
+private
+
+  def html
+    Nokogiri::HTML(body)
+  end
 end

spec/drivers_spec.rb

@@ -1,7 +1,7 @@
 require File.expand_path('spec_helper', File.dirname(__FILE__))
 
 shared_examples_for 'driver' do
-  
+
   describe '#visit' do
     it "should move to another page" do
       @driver.visit('/')
@@ -10,17 +10,42 @@ shared_examples_for 'driver' do
       @driver.body.should == 'Another World'
     end
   end
-  
+
   describe '#body' do
     it "should return text reponses" do
       @driver.visit('/')
       @driver.body.should == 'Hello world!'
     end
-    
+
     it "should return the full response html" do
       @driver.visit('/with_simple_html')
       @driver.body.should == '<h1>Bar</h1>'
     end
   end
-  
-end
+
+  describe '#find' do
+    context "with xpath selector" do
+      before do
+        @driver.visit('/with_html')
+      end
+
+      it "should find the correct number of elements" do
+        @driver.find('//a').size.should == 2
+      end
+
+      it "should extract node texts" do
+        @driver.find('//a')[0].text.should == 'labore'
+        @driver.find('//a')[1].text.should == 'ullamco'
+      end
+      
+      it "should extract node attributes" do
+        @driver.find('//a')[0].attribute(:href).should == '/with_simple_html'
+        @driver.find('//a')[0].attribute(:class).should == 'simple'
+        @driver.find('//a')[1].attribute(:href).should == '/foo'
+        @driver.find('//a')[1].attribute(:id).should == 'foo'
+        @driver.find('//a')[1].attribute(:rel).should be_nil
+      end
+    end
+  end
+
+end

spec/test_app.rb

@@ -8,6 +8,10 @@ class TestApp < Sinatra::Base
   get '/foo' do
     'Another World'
   end
+
+  get '/with_html' do
+    erb :with_html
+  end
   
   get '/with_simple_html' do
     erb :with_simple_html

spec/views/with_html.erb

@@ -0,0 +1,15 @@
+<h1>This is a test</h1>
+
+<p>
+  Lorem ipsum dolor sit amet, consectetur adipisicing elit, sed do eiusmod
+  tempor incididunt ut <a href="/with_simple_html" class="simple">labore</a>
+  et dolore magna aliqua. Ut enim ad minim veniam,
+  quis nostrud exercitation <a href="/foo" id="foo">ullamco</a> laboris nisi
+  ut aliquip ex ea commodo consequat.
+</p>
+
+<p>
+  Duis aute irure dolor in reprehenderit in voluptate velit esse cillum
+  dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident,
+  sunt in culpa qui officia deserunt mollit anim id est laborum.
+</p>