selenium-webdriver 0.0.1

data/common/src/rb/lib/selenium-webdriver.rb

@@ -0,0 +1 @@
+require "selenium/webdriver"

data/common/src/rb/lib/selenium/webdriver.rb

@@ -0,0 +1,52 @@
+require "tmpdir"
+require "fileutils"
+
+require "selenium/webdriver/core_ext/dir"
+require "selenium/webdriver/error"
+require "selenium/webdriver/platform"
+require "selenium/webdriver/child_process"
+require "selenium/webdriver/target_locator"
+require "selenium/webdriver/navigation"
+require "selenium/webdriver/options"
+require "selenium/webdriver/find"
+require "selenium/webdriver/bridge_helper"
+require "selenium/webdriver/driver"
+require "selenium/webdriver/element"
+
+begin
+  require "json" # gem dependency
+rescue LoadError => e
+  msg = Selenium::WebDriver::Platform.jruby? ? "jruby -S gem install json-jruby" : "gem install json"
+
+
+  raise LoadError, <<-END
+       #{e.message}
+
+       You need to install the json gem or (require rubygems):
+           #{msg}
+  END
+end
+
+
+module Selenium
+  module WebDriver
+    Point     = Struct.new(:x, :y)
+    Dimension = Struct.new(:width, :heigth)
+
+    autoload :IE,      'selenium/webdriver/ie'
+    autoload :Remote,  'selenium/webdriver/remote'
+    autoload :Chrome,  'selenium/webdriver/chrome'
+    autoload :Firefox, 'selenium/webdriver/firefox'
+
+    def self.root
+      @root ||= File.expand_path(File.join(File.dirname(__FILE__), "..", "..", "..", "..", ".."))
+    end
+
+    def self.for(*args)
+      WebDriver::Driver.for(*args)
+    end
+
+  end # WebDriver
+end # Selenium
+
+Thread.abort_on_exception = true

data/common/src/rb/lib/selenium/webdriver/bridge_helper.rb

@@ -0,0 +1,88 @@
+module Selenium
+  module WebDriver
+
+    #
+    # Shared across bridges
+    #
+
+    module BridgeHelper
+
+      def wrap_script_argument(arg)
+        case arg
+        when Integer, Float
+          { :type => "NUMBER", :value => arg }
+        when TrueClass, FalseClass, NilClass
+          { :type => "BOOLEAN", :value => !!arg }
+        when Element
+          { :type => "ELEMENT", :value => arg.ref }
+        when String
+          { :type => "STRING", :value => arg.to_s }
+        when Array # Enumerable?
+          arg.map { |e| wrap_script_argument(e) }
+        else
+          raise TypeError, "Parameter is not of recognized type: #{arg.inspect}:#{arg.class}"
+        end
+      end
+
+      def unwrap_script_argument(arg)
+        if arg.kind_of?(Array)
+          arg.map { |e| unwrap_script_argument(e) }
+        else
+          case arg["type"]
+          when "NULL"
+            nil
+          when "ELEMENT"
+            Element.new self, element_id_from(arg["value"])
+          when "POINT"
+            Point.new arg['x'], arg['y']
+          when "DIMENSION"
+            Dimension.new arg['width'], arg['height']
+          when "COOKIE"
+            {:name => arg['name'], :value => arg['value']}
+          else
+            arg["value"]
+          end
+        end
+      end
+
+      def element_id_from(arr)
+        arr.to_s.split("/").last
+      end
+
+      def parse_cookie_string(str)
+        result = {
+          'name'    => '',
+          'value'   => '',
+          'domain'  => '',
+          'path'    => '',
+          'expires' => '',
+          'secure'  => false
+        }
+
+        str.split(";").each do |attribute|
+          if attribute.include? "="
+            key, value = attribute.strip.split("=", 2)
+            if result['name'].empty?
+              result['name']  = key
+              result['value'] = value
+            elsif key == 'domain' && value.strip =~ /^\.(.+)/
+              result['domain'] = $1
+            elsif key && value
+              result[key] = value
+            end
+          elsif attribute == "secure"
+            result['secure'] = true
+          end
+
+          unless [nil, "", "0"].include?(result['expires'])
+            # firefox stores expiry as number of seconds
+            result['expires'] = Time.at(result['expires'].to_i)
+          end
+        end
+
+        result
+      end
+
+    end # BridgeHelper
+  end # WebDriver
+end # Selenium

data/common/src/rb/lib/selenium/webdriver/child_process.rb

@@ -0,0 +1,85 @@
+module Selenium
+  module WebDriver
+
+    #
+    # Cross platform child process launcher
+    #
+
+    class ChildProcess
+      def initialize(*args)
+        @args = args
+
+        if Platform.jruby?
+          extend JRubyProcess
+        elsif Platform.os == :windows
+          extend WindowsProcess
+        end
+      end
+
+      def alive?
+        return false unless @pid
+
+        # TODO: check if this works on windows
+        Process.kill 0, @pid
+        true
+      rescue Errno::ESRCH
+        false
+      end
+
+      def start
+        @pid = fork { exec(*@args) }
+
+        self
+      end
+
+      def wait
+        raise "called wait with no pid" unless @pid
+        Process.waitpid @pid
+      end
+
+      def kill
+        Process.kill("TERM", @pid) if @pid
+      end
+
+      def kill!
+        Process.kill("KILL", @pid) if @pid
+      end
+
+      module WindowsProcess
+        def start
+          require "win32/process" # adds a dependency on windows
+          @pid = Process.create(:app_name        => @args.join(" "),
+                                :process_inherit => true,
+                                :thread_inherit  => true,
+                                :inherit         => true).process_id
+
+          self
+        end
+      end
+
+      module JRubyProcess
+        def start
+          pb = java.lang.ProcessBuilder.new(@args)
+
+          # this isn't good
+          env = pb.environment
+          ENV.each { |k,v| env.put(k, v) }
+
+          @process = pb.start
+
+          self
+        end
+
+        def kill
+          @process.destroy if @process
+        end
+        alias_method :kill!, :kill
+
+        def wait
+          @process.waitFor
+        end
+      end
+
+    end # ChildProcess
+  end # WebDriver
+end # Selenium

data/common/src/rb/lib/selenium/webdriver/core_ext/dir.rb

@@ -0,0 +1,41 @@
+class Dir
+  def self.mktmpdir(prefix_suffix=nil, tmpdir=nil)
+    case prefix_suffix
+    when nil
+      prefix = "d"
+      suffix = ""
+    when String
+      prefix = prefix_suffix
+      suffix = ""
+    when Array
+      prefix = prefix_suffix[0]
+      suffix = prefix_suffix[1]
+    else
+      raise ArgumentError, "unexpected prefix_suffix: #{prefix_suffix.inspect}"
+    end
+    tmpdir ||= Dir.tmpdir
+    t = Time.now.strftime("%Y%m%d")
+    n = nil
+    begin
+      path = "#{tmpdir}/#{prefix}#{t}-#{$$}-#{rand(0x100000000).to_s(36)}"
+      path << "-#{n}" if n
+      path << suffix
+      Dir.mkdir(path, 0700)
+    rescue Errno::EEXIST
+      n ||= 0
+      n += 1
+      retry
+    end
+
+    if block_given?
+      begin
+        yield path
+      ensure
+        FileUtils.remove_entry_secure path
+      end
+    else
+      path
+    end
+  end unless respond_to? :mktmpdir # > 1.8.7
+end
+

data/common/src/rb/lib/selenium/webdriver/driver.rb

@@ -0,0 +1,128 @@
+module Selenium
+  module WebDriver
+    class Driver
+      include Find
+
+      attr_reader :bridge
+
+      class << self
+        def for(driver, *args)
+          case driver
+          when :ie, :internet_explorer
+            new WebDriver::IE::Bridge.new(*args)
+          when :remote
+            new WebDriver::Remote::Bridge.new(*args)
+          when :chrome
+            new WebDriver::Chrome::Bridge.new(*args)
+          when :firefox, :ff
+            new WebDriver::Firefox::Bridge.new(*args)
+          else
+            raise ArgumentError, "unknown driver: #{driver.inspect}"
+          end
+        end
+      end
+
+      def initialize(bridge)
+        @bridge = bridge
+      end
+
+      def navigate
+        @navigate ||= WebDriver::Navigation.new(self)
+      end
+
+      def switch_to
+        @switch_to ||= WebDriver::TargetLocator.new(self)
+      end
+
+      def manage
+        @manage ||= WebDriver::Options.new(self)
+      end
+
+      def current_url
+        bridge.getCurrentUrl
+      end
+
+      def title
+        bridge.getTitle
+      end
+
+      def page_source
+        bridge.getPageSource
+      end
+
+      def visible?
+        bridge.getBrowserVisible
+      end
+
+      def visible=(bool)
+        bridge.setBrowserVisible bool
+      end
+
+      def quit
+        bridge.quit
+      end
+
+      def close
+        bridge.close
+      end
+
+      def window_handles
+        bridge.getWindowHandles
+      end
+
+      def window_handle
+        bridge.getCurrentWindowHandle
+      end
+
+      def execute_script(script, *args)
+        bridge.executeScript(script, *args)
+      end
+
+      #-------------------------------- sugar  --------------------------------
+
+      #
+      # driver.first(:id, 'foo')
+      #
+
+      alias_method :first, :find_element
+
+      #
+      # driver.all(:class, 'bar') #=> [#<WebDriver::Element:0x1011c3b88, ...]
+      #
+
+      alias_method :all, :find_elements
+
+      #
+      # opens the specified URL in the browser.
+      #
+
+      def get(url)
+        navigate.to(url)
+      end
+
+      #
+      # driver.script('function() { ... };')
+      #
+
+      alias_method :script, :execute_script
+
+      #
+      # driver['someElementId'] #=> #<WebDriver::Element:0x1011c3b88>
+      #
+
+      def [](id)
+        find_element :id, id
+      end
+
+
+      #
+      # for Find
+      #
+
+      def ref
+        nil
+      end
+
+    end # Driver
+  end # WebDriver
+end # Selenium

data/common/src/rb/lib/selenium/webdriver/element.rb

@@ -0,0 +1,126 @@
+module Selenium
+  module WebDriver
+    class Element
+      include Find
+
+      attr_reader :bridge
+
+      def initialize(bridge, id)
+        @bridge, @id = bridge, id
+      end
+
+      def click
+        bridge.clickElement @id
+      end
+
+      def tag_name
+        bridge.getElementTagName @id
+      end
+
+      def value
+        bridge.getElementValue @id
+      end
+
+      def attribute(name)
+        bridge.getElementAttribute @id, name
+      end
+
+      def text
+        bridge.getElementText @id
+      end
+
+      # FIXME: send_keys args
+      def send_keys(*args)
+        args.each { |str| bridge.sendKeysToElement(@id, str) }
+      end
+
+      def clear
+        bridge.clearElement @id
+      end
+
+      def enabled?
+        bridge.isElementEnabled @id
+      end
+
+      def selected?
+        bridge.isElementSelected @id
+      end
+
+      def displayed?
+        bridge.isElementDisplayed @id
+      end
+
+      def select
+        bridge.setElementSelected @id
+      end
+
+      def submit
+        bridge.submitElement @id
+      end
+
+      def toggle
+        bridge.toggleElement @id
+      end
+
+      def style(prop)
+        bridge.getElementValueOfCssProperty @id, prop
+      end
+
+      def hover
+        bridge.hoverOverElement @id
+      end
+
+      def location
+        bridge.getElementLocation @id
+      end
+
+      def size
+        bridge.getElementSize @id
+      end
+
+      def drag_and_drop_by(right_by, down_by)
+        bridge.dragElement @id, right_by, down_by
+      end
+
+      def drag_and_drop_on(other)
+        current_location = location()
+        destination      = other.location
+
+        right = destination.x - current_location.x
+        down  = destination.y - current_location.y
+
+        drag_and_drop_by right, down
+      end
+
+      #-------------------------------- sugar  --------------------------------
+
+      #
+      # element.first(:id, 'foo')
+      #
+
+      alias_method :first, :find_element
+
+      #
+      # element.all(:class, 'bar')
+      #
+
+      alias_method :all, :find_elements
+
+      #
+      # element['class'] or element[:class] #=> "someclass"
+      #
+      alias_method :[], :attribute
+
+      #
+      # for Find and execute_script
+      #
+
+      def ref
+        @id
+      end
+
+
+
+    end # Element
+  end # WebDriver
+end # Selenium