capybara 0.1

data/lib/capybara/driver/firewatir_driver.rb

@@ -0,0 +1,66 @@
+require 'watir'
+Watir::Browser.default = "firefox"
+
+class Capybara::Driver::FireWatir
+  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
+    
+    def click
+      node.click
+    end
+    
+    def tag_name
+      # FIXME: this might be the dumbest way ever of getting the tag name
+      # there has to be something better...
+      node.to_xml[/^\s*<([a-z0-9\-\:]+)/, 1]
+    end
+  end
+  
+  attr_reader :app, :rack_server
+
+  def initialize(app)
+    @app = app
+    @rack_server = Capybara::Server.new(@app)
+    @rack_server.boot
+  end
+  
+  def visit(path)
+    browser.goto(url(path))
+  end
+  
+  def body
+    browser.html
+  end
+  
+  def find(selector)
+    browser.elements_by_xpath(selector).map { |node| Node.new(node) }
+  end
+
+private
+
+  def url(path)
+    rack_server.url(path)
+  end
+  
+  def browser
+    unless @_browser
+      @_browser = Watir::Browser.new
+      at_exit do
+        @_browser.exit
+      end
+    end
+    @_browser
+  end
+
+end

data/lib/capybara/driver/rack_test_driver.rb

@@ -0,0 +1,151 @@
+require 'rack/test'
+require 'nokogiri'
+
+class Capybara::Driver::RackTest
+  class Node < Capybara::Node
+    def text
+      node.text
+    end
+    
+    def [](name)
+      value = node[name.to_s]
+      return value.to_s if value
+    end
+
+    def set(value)
+      if tag_name == 'input' and %w(text password hidden file).include?(type)
+        node['value'] = value.to_s
+      elsif tag_name == 'input' and type == 'radio'
+        driver.html.xpath("//input[@name='#{self[:name]}']").each { |node| node.remove_attribute("checked") }
+        node['checked'] = 'checked'
+      elsif tag_name == 'input' and type == 'checkbox'
+        if value
+          node['checked'] = 'checked'
+        else
+          node.remove_attribute('checked')
+        end
+      elsif tag_name == "textarea"
+        node.content = value.to_s
+      end
+    end
+    
+    def select(option)
+      node.xpath(".//option").each { |node| node.remove_attribute("selected") }
+      node.xpath(".//option[contains(.,'#{option}')]").first["selected"] = 'selected'
+    end
+
+    def click
+      if tag_name == 'a'
+        driver.visit(self[:href])
+      elsif tag_name == 'input' and %w(submit image).include?(type)
+        Form.new(driver, form).submit(self)
+      end
+    end
+    
+    def tag_name
+      node.node_name
+    end
+
+  private
+  
+    def type
+      self[:type]
+    end
+
+    def form
+      node.ancestors('form').first
+    end
+  end
+
+  class Form < Node
+    def params(button)
+      params = []
+      params += node.xpath(".//input[@type='text']", ".//input[@type='hidden']", ".//input[@type='password']").inject([]) do |agg, input|
+        agg << [input['name'].to_s, input['value'].to_s]
+        agg
+      end
+      params += node.xpath(".//textarea").inject([]) do |agg, textarea|
+        agg << [textarea['name'].to_s, textarea.text.to_s]
+        agg
+      end
+      params += node.xpath(".//input[@type='radio']").inject([]) do |agg, input|
+        agg << [input['name'].to_s, input['value'].to_s] if input['checked']
+        agg
+      end
+      params += node.xpath(".//input[@type='checkbox']").inject([]) do |agg, input|
+        agg << [input['name'].to_s, input['value'].to_s] if input['checked']
+        agg
+      end
+      params += node.xpath(".//select").inject([]) do |agg, select|
+        option = select.xpath(".//option[@selected]").first
+        option ||= select.xpath('.//option').first
+        agg << [select['name'].to_s, (option['value'] || option.text).to_s] if option 
+        agg
+      end
+      params += node.xpath(".//input[@type='file']").inject([]) do |agg, input|
+        if multipart?
+          agg << [input['name'].to_s, Rack::Test::UploadedFile.new(input['value'].to_s)]
+        else
+          agg << [input['name'].to_s, File.basename(input['value'].to_s)]
+        end
+        agg
+      end
+      params.push [button[:name], button[:value]] if button[:name]
+      if multipart?
+        params.inject({}) { |agg, (key, value)| agg[key] = value; agg }
+      else
+        params.map { |key, value| "#{key}=#{value}" }.join('&')
+      end
+    end
+
+    def submit(button)
+      if post?
+        driver.submit(node['action'].to_s, params(button)) 
+      else
+        driver.visit(node['action'].to_s.split('?').first + '?' + params(button)) 
+      end
+    end
+
+    def multipart?
+      self[:enctype] == "multipart/form-data"
+    end
+    
+    def post?
+      self[:method] =~ /post/i
+    end
+  end
+  
+  include ::Rack::Test::Methods
+  attr_reader :app, :html, :body
+
+  alias_method :response, :last_response
+  alias_method :request, :last_request
+  
+  def initialize(app)
+    @app = app
+  end
+  
+  def visit(path)
+    get(path)
+    follow_redirect! while response.redirect?
+    cache_body
+  end
+
+  def submit(path, attributes)
+    post(path, attributes)
+    follow_redirect! while response.redirect?
+    cache_body  
+  end
+  
+  def find(selector)
+    html.xpath(selector).map { |node| Node.new(self, node) }
+  end
+
+  private
+
+  def cache_body
+    @body = response.body
+    @html = Nokogiri::HTML(body)
+  end
+
+end

data/lib/capybara/driver/safariwatir_driver.rb

@@ -0,0 +1,67 @@
+require 'safariwatir'
+
+class Capybara::Driver::SafariWatir
+  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
+    
+    def click
+      node.click
+    end
+    
+    def tag_name
+      # FIXME: this might be the dumbest way ever of getting the tag name
+      # there has to be something better...
+      node.to_xml[/^\s*<([a-z0-9\-\:]+)/, 1]
+    end
+  end
+  
+  attr_reader :app, :rack_server
+
+  def initialize(app)
+    @app = app
+    @rack_server = Capybara::Server.new(@app)
+    @rack_server.boot
+  end
+  
+  def visit(path)
+    browser.goto(url(path))
+  end
+  
+  def body
+    browser.html
+  end
+  
+  def find(selector)
+    foo = Struct.new(:what).new
+    foo.what = selector
+    browser.send(:scripter).operate_by_xpath(foo){}.map { |node| Node.new(node) }
+  end
+
+private
+
+  def url(path)
+    rack_server.url(path)
+  end
+  
+  def browser
+    unless @_browser
+      @_browser = Watir::Safari.new
+      at_exit do
+        @_browser.exit
+      end
+    end
+    @_browser
+  end
+
+end

data/lib/capybara/driver/selenium_driver.rb

@@ -0,0 +1,93 @@
+require 'selenium-webdriver'
+
+class Capybara::Driver::Selenium
+  class Node < Capybara::Node
+    def text
+      node.text
+    end
+
+    def [](name)
+      if name == :value
+        node.value
+      else
+        node.attribute(name)
+      end
+    rescue Selenium::WebDriver::Error::WebDriverError
+      nil
+    end
+
+    def set(value)
+      if tag_name == 'input' and %w(text password hidden file).include?(type)
+        node.clear
+        node.send_keys(value.to_s)
+      elsif tag_name == 'input' and type == 'radio'
+        node.select
+      elsif tag_name == 'input' and type == 'checkbox'
+        node.toggle
+      elsif tag_name == "textarea"
+        node.clear
+        node.send_keys(value.to_s)
+      end
+    end
+
+    def select(option)
+      node.find_element(:xpath, ".//option[contains(.,'#{option}')]").select
+    end
+
+    def click
+      node.click
+    end
+
+    def tag_name
+      node.tag_name
+    end
+
+  private
+
+    def type
+      self[:type]
+    end
+
+  end
+
+  attr_reader :app, :rack_server
+
+  def self.driver
+    unless @driver
+      @driver = Selenium::WebDriver.for :firefox
+      at_exit do
+        @driver.quit
+      end
+    end
+    @driver
+  end
+
+  def initialize(app)
+    @app = app
+    @rack_server = Capybara::Server.new(@app)
+    @rack_server.boot
+  end
+
+  def visit(path)
+    driver.navigate.to(url(path))
+  end
+
+  def body
+    driver.page_source
+  end
+
+  def find(selector)
+    driver.find_elements(:xpath, selector).map { |node| Node.new(self, node) }
+  end
+
+private
+
+  def url(path)
+    rack_server.url(path)
+  end
+
+  def driver
+    self.class.driver
+  end
+
+end

data/lib/capybara/dsl.rb

@@ -0,0 +1,58 @@
+module Capybara
+  class << self
+    attr_writer :default_driver, :current_driver, :javascript_driver
+
+    attr_accessor :app
+
+    def default_driver
+      @default_driver || :rack_test
+    end
+
+    def current_driver
+      @current_driver || default_driver 
+    end
+    alias_method :mode, :current_driver
+    
+    def javascript_driver
+      @javascript_driver || :selenium
+    end
+
+    def use_default_driver
+      @current_driver = nil 
+    end
+
+    def current_session
+      session_pool["#{current_driver}#{app.object_id}"] ||= Capybara::Session.new(current_driver, app)
+    end
+    
+    def reset_sessions!
+      @session_pool = nil
+    end
+
+  private
+
+    def session_pool
+      @session_pool ||= {}
+    end
+  end
+
+  extend(self)
+
+  def page
+    Capybara.current_session
+  end
+
+  SESSION_METHODS = [
+    :visit, :body, :click_link, :click_button, :fill_in, :choose, :has_xpath?, :has_css?,
+    :check, :uncheck, :attach_file, :select, :has_content?, :within, :save_and_open_page,
+    :find_field, :find_link, :find_button, :field_labeled
+  ]
+  SESSION_METHODS.each do |method|
+    class_eval <<-RUBY, __FILE__, __LINE__+1
+      def #{method}(*args, &block)
+        page.#{method}(*args, &block)
+      end
+    RUBY
+  end
+
+end

data/lib/capybara/node.rb

@@ -0,0 +1,36 @@
+class Capybara::Node
+  attr_reader :driver, :node
+
+  def initialize(driver, node)
+    @driver = driver
+    @node = node
+  end
+
+  def text
+    raise "Not implemented"
+  end
+  
+  def [](name)
+    raise "Not implemented"
+  end
+  
+  def value
+    self[:value]
+  end
+
+  def set(value)
+    raise "Not implemented"
+  end
+  
+  def select(option)
+    raise "Not implemented"
+  end
+
+  def click
+    raise "Not implemented"
+  end
+  
+  def tag_name
+    raise "Not implemented"
+  end
+end