pr0d1r2-capybara 0.3.7

data/lib/capybara.rb

@@ -0,0 +1,57 @@
+require 'timeout'
+require 'nokogiri'
+
+module Capybara
+  class CapybaraError < StandardError; end
+  class DriverNotFoundError < CapybaraError; end
+  class ElementNotFound < CapybaraError; end
+  class OptionNotFound < ElementNotFound; end
+  class UnselectNotAllowed < CapybaraError; end
+  class NotSupportedByDriverError < CapybaraError; end
+  class TimeoutError < CapybaraError; end
+  class LocateHiddenElementError < CapybaraError; end
+  class InfiniteRedirectError < TimeoutError; end
+  
+  class << self
+    attr_accessor :debug, :asset_root, :app_host, :run_server, :default_host
+    attr_accessor :default_selector, :default_wait_time, :default_timeout_in_follow_redirects, :ignore_hidden_elements
+
+    def default_selector
+      @default_selector ||= :xpath
+    end
+    
+    def default_wait_time
+      @default_wait_time ||= 2
+    end
+
+    def default_timeout_in_follow_redirects
+      @default_timeout_in_follow_redirects ||= 4
+    end
+
+    def log(message)
+      puts "[capybara] #{message}" if debug
+      true
+    end
+  end
+  
+  autoload :Server,     'capybara/server'
+  autoload :Session,    'capybara/session'
+  autoload :Node,       'capybara/node'
+  autoload :XPath,      'capybara/xpath'
+  autoload :Searchable, 'capybara/searchable'
+  autoload :VERSION,    'capybara/version'
+  
+  module Driver
+    autoload :Base,     'capybara/driver/base'
+    autoload :RackTest, 'capybara/driver/rack_test_driver'
+    autoload :Celerity, 'capybara/driver/celerity_driver'
+    autoload :Culerity, 'capybara/driver/culerity_driver'
+    autoload :Selenium, 'capybara/driver/selenium_driver'
+  end
+end
+
+Capybara.run_server = true
+Capybara.default_selector = :xpath
+Capybara.default_wait_time = 2
+Capybara.default_timeout_in_follow_redirects = 4
+Capybara.ignore_hidden_elements = false

data/lib/capybara/cucumber.rb

@@ -0,0 +1,32 @@
+require 'capybara'
+require 'capybara/dsl'
+
+World(Capybara)
+
+After do
+  Capybara.reset_sessions!
+end
+
+Before('@javascript') do
+  Capybara.current_driver = Capybara.javascript_driver
+end
+
+Before('@selenium') do
+  Capybara.current_driver = :selenium
+end
+
+Before('@celerity') do
+  Capybara.current_driver = :celerity
+end
+
+Before('@culerity') do
+  Capybara.current_driver = :culerity
+end
+
+Before('@rack_test') do
+  Capybara.current_driver = :rack_test
+end
+
+After do
+  Capybara.use_default_driver
+end

data/lib/capybara/driver/base.rb

@@ -0,0 +1,48 @@
+class Capybara::Driver::Base
+  def current_url
+    raise NotImplementedError
+  end
+
+  def visit(path)
+    raise NotImplementedError
+  end
+
+  def find(query)
+    raise NotImplementedError
+  end
+
+  def evaluate_script(script)
+    raise Capybara::NotSupportedByDriverError
+  end
+
+  def wait?
+    false
+  end
+
+  def wait_until *args
+  end
+
+  def response_headers
+    raise Capybara::NotSupportedByDriverError
+  end
+
+  def body
+    raise NotImplementedError
+  end
+
+  def within_frame(frame_id)
+    raise Capybara::NotSupportedByDriverError
+  end
+
+  def source
+    raise NotImplementedError
+  end
+
+  def cleanup!
+  end
+
+  def has_shortcircuit_timeout?
+    false
+  end
+
+end

data/lib/capybara/driver/celerity_driver.rb

@@ -0,0 +1,143 @@
+class Capybara::Driver::Celerity < Capybara::Driver::Base
+  class Node < Capybara::Node
+    def text
+      node.text
+    end
+
+    def [](name)
+      value = if name.to_sym == :class
+        node.class_name
+      else
+        node.send(name.to_sym)
+      end
+      return value if value and not value.to_s.empty?
+    end
+
+    def value
+      if tag_name == "select" and node.multiple?
+        node.selected_options
+      else
+        super
+      end
+    end
+
+    def set(value)
+      node.set(value)
+    end
+
+    def select(option)
+      node.select(option)
+    rescue
+      options = all(:xpath, "//option").map { |o| "'#{o.text}'" }.join(', ')
+      raise Capybara::OptionNotFound, "No such option '#{option}' in this select box. Available options: #{options}"
+    end
+
+    def unselect(option)
+      unless node.multiple?
+        raise Capybara::UnselectNotAllowed, "Cannot unselect option '#{option}' from single select box."
+      end
+
+      # FIXME: couldn't find a clean way to unselect, so clear and reselect
+      selected_options = node.selected_options
+      if unselect_option  = selected_options.detect { |value| value == option } ||
+                            selected_options.detect { |value| value.index(option) }
+        node.clear
+        (selected_options - [unselect_option]).each { |value| node.select_value(value) }
+      else
+        options = all(:xpath, "//option").map { |o| "'#{o.text}'" }.join(', ')
+        raise Capybara::OptionNotFound, "No such option '#{option}' in this select box. Available options: #{options}"
+      end
+    end
+
+    def click
+      node.click
+    end
+
+    def drag_to(element)
+      node.fire_event('mousedown')
+      element.node.fire_event('mousemove')
+      element.node.fire_event('mouseup')
+    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
+
+    def visible?
+      node.visible?
+    end
+
+    def path
+      node.xpath
+    end
+    
+    def trigger(event)
+      node.fire_event(event.to_s)
+    end
+
+  private
+
+    def all_unfiltered(locator)
+      noko_node = Nokogiri::HTML(driver.body).xpath(node.xpath).first
+      all_nodes = noko_node.xpath(locator).map { |n| n.path }.join(' | ')
+      driver.find(all_nodes)
+    end
+
+  end
+
+  attr_reader :app, :rack_server
+
+  def initialize(app)
+    @app = app
+    @rack_server = Capybara::Server.new(@app)
+    @rack_server.boot if Capybara.run_server
+  end
+
+  def visit(path)
+    browser.goto(url(path))
+  end
+
+  def current_url
+    browser.url
+  end
+  
+  def source
+    browser.html
+  end
+
+  def body
+    browser.document.as_xml
+  end
+
+  def response_headers
+    browser.response_headers
+  end
+
+  def find(selector)
+    browser.elements_by_xpath(selector).map { |node| Node.new(self, node) }
+  end
+
+  def wait?; true; end
+
+  def evaluate_script(script)
+    browser.execute_script "#{script}"
+  end
+
+  def browser
+    unless @_browser
+      require 'celerity'
+      @_browser = ::Celerity::Browser.new(:browser => :firefox, :log_level => :off)
+    end
+
+    @_browser
+  end
+
+private
+
+  def url(path)
+    rack_server.url(path)
+  end
+
+end

data/lib/capybara/driver/culerity_driver.rb

@@ -0,0 +1,25 @@
+require 'culerity'
+
+class Capybara::Driver::Culerity < Capybara::Driver::Celerity
+
+  def self.server
+    unless @_server
+      @_server = ::Culerity::run_server
+      at_exit do
+        @_server.close
+      end
+    end
+    @_server
+  end
+
+  def browser
+    unless @_browser
+      @_browser = ::Culerity::RemoteBrowserProxy.new self.class.server, {:browser => :firefox, :log_level => :off}
+      at_exit do
+        @_browser.exit
+      end
+    end
+    @_browser
+  end
+
+end

data/lib/capybara/driver/rack_test_driver.rb

@@ -0,0 +1,271 @@
+require 'rack/test'
+require 'mime/types'
+require 'nokogiri'
+require 'cgi'
+
+class Capybara::Driver::RackTest < Capybara::Driver::Base
+  class Node < Capybara::Node
+    def text
+      node.text
+    end
+
+    def [](name)
+      attr_name = name.to_s
+      case
+      when 'select' == tag_name && 'value' == attr_name
+        if node['multiple'] == 'multiple'
+          node.xpath(".//option[@selected='selected']").map { |option| option.content  }
+        else
+          option = node.xpath(".//option[@selected='selected']").first || node.xpath(".//option").first
+          option.content if option
+        end
+      when 'input' == tag_name && 'checkbox' == type && 'checked' == attr_name
+        node[attr_name] == 'checked' ? true : false
+      else
+        node[attr_name]
+      end
+    end
+
+
+    def set(value)
+      if tag_name == 'input' and type == 'radio'
+        driver.html.xpath("//input[@name=#{Capybara::XPath.escape(self[:name])}]").each { |node| node.remove_attribute("checked") }
+        node['checked'] = 'checked'
+      elsif tag_name == 'input' and type == 'checkbox'
+        if value && !node['checked']
+          node['checked'] = 'checked'
+        elsif !value && node['checked']
+          node.remove_attribute('checked')
+        end
+      elsif tag_name == 'input'
+        node['value'] = value.to_s
+      elsif tag_name == "textarea"
+        node.content = value.to_s
+      end
+    end
+
+    def select(option)
+      if node['multiple'] != 'multiple'
+        node.xpath(".//option[@selected]").each { |node| node.remove_attribute("selected") }
+      end
+
+      if option_node = node.xpath(".//option[text()=#{Capybara::XPath.escape(option)}]").first ||
+                       node.xpath(".//option[contains(.,#{Capybara::XPath.escape(option)})]").first
+        option_node["selected"] = 'selected'
+      else
+        options = node.xpath(".//option").map { |o| "'#{o.text}'" }.join(', ')
+        raise Capybara::OptionNotFound, "No such option '#{option}' in this select box. Available options: #{options}"
+      end
+    end
+
+    def unselect(option)
+      if node['multiple'] != 'multiple'
+        raise Capybara::UnselectNotAllowed, "Cannot unselect option '#{option}' from single select box."
+      end
+
+      if option_node = node.xpath(".//option[text()=#{Capybara::XPath.escape(option)}]").first ||
+                       node.xpath(".//option[contains(.,#{Capybara::XPath.escape(option)})]").first
+        option_node.remove_attribute('selected')
+      else
+        options = node.xpath(".//option").map { |o| "'#{o.text}'" }.join(', ')
+        raise Capybara::OptionNotFound, "No such option '#{option}' in this select box. Available options: #{options}"
+      end
+    end
+
+    def click
+      if tag_name == 'a'
+        method = self["data-method"] || :get
+        driver.process(method, self[:href].to_s)
+      elsif (tag_name == 'input' or tag_name == 'button') and %w(submit image).include?(type)
+        Form.new(driver, form).submit(self)
+      end
+    end
+
+    def tag_name
+      node.node_name
+    end
+
+    def visible?
+      node.xpath("./ancestor-or-self::*[contains(@style, 'display:none') or contains(@style, 'display: none')]").size == 0
+    end
+
+    def path
+      node.path
+    end
+
+  private
+
+    def all_unfiltered(locator)
+      node.xpath(locator).map { |n| self.class.new(driver, n) }
+    end
+
+    def type
+      node[:type]
+    end
+
+    def form
+      node.ancestors('form').first
+    end
+  end
+
+  class Form < Node
+    def params(button)
+      params = {}
+
+      node.xpath(".//input[@type!='radio' and @type!='checkbox' and @type!='submit']").map do |input|
+        merge_param!(params, input['name'].to_s, input['value'].to_s)
+      end
+      node.xpath(".//textarea").map do |textarea|
+        merge_param!(params, textarea['name'].to_s, textarea.text.to_s)
+      end
+      node.xpath(".//input[@type='radio' or @type='checkbox']").map do |input|
+        merge_param!(params, input['name'].to_s, input['value'].to_s) if input['checked']
+      end
+      node.xpath(".//select").map do |select|
+        if select['multiple'] == 'multiple'
+          options = select.xpath(".//option[@selected]")
+          options.each do |option|
+            merge_param!(params, select['name'].to_s, (option['value'] || option.text).to_s)
+          end
+        else
+          option = select.xpath(".//option[@selected]").first
+          option ||= select.xpath('.//option').first
+          merge_param!(params, select['name'].to_s, (option['value'] || option.text).to_s) if option
+        end
+      end
+      node.xpath(".//input[@type='file']").map do |input|
+        unless input['value'].to_s.empty?
+          if multipart?
+            content_type = MIME::Types.type_for(input['value'].to_s).first.to_s
+            file = Rack::Test::UploadedFile.new(input['value'].to_s, content_type)
+            merge_param!(params, input['name'].to_s, file)
+          else
+            merge_param!(params, input['name'].to_s, File.basename(input['value'].to_s))
+          end
+        end
+      end
+      merge_param!(params, button[:name], button[:value] || "") if button[:name]
+      params
+    end
+
+    def submit(button)
+      driver.submit(method, node['action'].to_s, params(button))
+    end
+
+    def multipart?
+      self[:enctype] == "multipart/form-data"
+    end
+
+  private
+
+    def method
+      self[:method] =~ /post/i ? :post : :get
+    end
+
+    def merge_param!(params, key, value)
+      collection = key.sub!(/\[\]$/, '')
+      if collection
+        if params[key]
+          params[key] << value
+        else
+          params[key] = [value]
+        end
+      else
+        params[key] = value
+      end
+    end
+  end
+
+  include ::Rack::Test::Methods
+  attr_reader :app
+
+  alias_method :response, :last_response
+  alias_method :request, :last_request
+
+  def initialize(app)
+    @app = app
+  end
+
+  def visit(path, attributes = {})
+    process(:get, path, attributes)
+  end
+
+  def process(method, path, attributes = {})
+    return if path.gsub(/^#{current_path}/, '') =~ /^#/
+    send(method, path, attributes, env)
+    follow_redirects!
+  end
+
+  def current_url
+    request.url rescue ""
+  end
+
+  def response_headers
+    response.headers
+  end
+
+  def submit(method, path, attributes)
+    path = current_path if not path or path.empty?
+    send(method, path, attributes, env)
+    follow_redirects!
+  end
+
+  def find(selector)
+    html.xpath(selector).map { |node| Node.new(self, node) }
+  end
+  
+  def body
+    @body ||= response.body
+  end
+  
+  def html
+    @html ||= Nokogiri::HTML(body)
+  end
+  alias_method :source, :body
+
+  def get(*args, &block); reset_cache; super; end
+  def post(*args, &block); reset_cache; super; end
+  def put(*args, &block); reset_cache; super; end
+  def delete(*args, &block); reset_cache; super; end
+  
+private
+
+  def reset_cache
+    @body = nil
+    @html = nil
+  end
+
+  def build_rack_mock_session # :nodoc:
+    Rack::MockSession.new(app, Capybara.default_host || "www.example.com")
+  end
+
+  def current_path
+    request.path rescue ""
+  end
+
+  def follow_redirects!
+    Capybara::WaitUntil.timeout(Capybara.default_timeout_in_follow_redirects) do
+      redirect = response.redirect?
+      follow_redirect! if redirect
+      not redirect
+    end
+  rescue Capybara::TimeoutError
+    raise Capybara::InfiniteRedirectError, "infinite redirect detected!"
+  end
+
+  def env
+    env = {}
+    begin
+      env["HTTP_REFERER"] = request.url
+    rescue Rack::Test::Error
+      # no request yet
+    end
+    env
+  end
+
+  def reset_cache
+    @body = nil
+    @html = nil
+  end
+
+end