davidtrogers-selenium-grid 0.0.1

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/.gitignore

@@ -0,0 +1,21 @@
+## MAC OS
+.DS_Store
+
+## TEXTMATE
+*.tmproj
+tmtags
+
+## EMACS
+*~
+\#*
+.\#*
+
+## VIM
+*.swp
+
+## PROJECT::GENERAL
+coverage
+rdoc
+pkg
+
+## PROJECT::SPECIFIC

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Dave Rogers
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README

@@ -0,0 +1,10 @@
+Thank you for using Selenium Grid
+
+Have a look at :
+
+ * Website content : ./doc/website/index.html
+ * Developer documentation: ./doc directory
+
+If you have any problem contact Philippe Hanrigou:
+http://ph7spot.com/about/contact_me
+

data/README.rdoc

@@ -0,0 +1,17 @@
+= davidtrogers-selenium-grid
+
+Description goes here.
+
+== Note on Patches/Pull Requests
+ 
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+== Copyright
+
+Copyright (c) 2010 Dave Rogers. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,243 @@
+require 'rubygems'
+require 'rake'
+#
+# Rakefile managing Selenium Grid components
+#
+require "net/http"
+require "yaml"
+require File.dirname(__FILE__) + '/lib/ruby/tcp_socket_extensions'
+require File.dirname(__FILE__) + '/lib/ruby/file_extensions'
+require File.dirname(__FILE__) + '/lib/ruby/java/classpath'
+require File.dirname(__FILE__) + '/lib/ruby/java/vm'
+require File.dirname(__FILE__) + '/lib/ruby/s_grid/hub'
+require File.dirname(__FILE__) + '/lib/ruby/s_grid/remote_control'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "davidtrogers-selenium-grid"
+    gem.summary = %Q{selenium grid packaged as a gem}
+    gem.description = %Q{}
+    gem.email = "david.t.rogers@gmail.com"
+    # gem.homepage = "http://github.com/davidtrogers/davidtrogers-selenium-grid"
+    gem.authors = ["Dave Rogers"]
+    gem.add_development_dependency "thoughtbot-shoulda", ">= 0"
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: gem install jeweler"
+end
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end
+
+begin
+  require 'rcov/rcovtask'
+  Rcov::RcovTask.new do |test|
+    test.libs << 'test'
+    test.pattern = 'test/**/test_*.rb'
+    test.verbose = true
+  end
+rescue LoadError
+  task :rcov do
+    abort "RCov is not available. In order to run rcov, you must: sudo gem install spicycode-rcov"
+  end
+end
+
+task :test => :check_dependencies
+
+task :default => :test
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  version = File.exist?('VERSION') ? File.read('VERSION') : ""
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "davidtrogers-selenium-grid #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end
+
+desc "Print help"
+task :help do
+  puts <<-EOS
+  
+ Rake tasks to manage a Selenium Grid on a *single* machine.
+
+   NOTE: To manage a Grid accross *multiple* machines use the 
+         Capistrano recipes provided with Selenium Grid distribution
+ 
+ 
+ Start the Hub and 15 Remote Controls
+ ------------------------------------
+ 
+    rake all:start
+   
+ Stop the Hub and 15 Remote Controls
+ -----------------------------------
+ 
+    rake all:stop
+   
+ Restart the Hub and 15 Remote Controls
+ --------------------------------------
+
+    rake all:restart
+
+ Start the Hub with only 10 Remote Controls, ports 5000 to 5010
+ --------------------------------------------------------------
+
+    rake all:restart PORTS=5000-5010
+
+  Stop them with
+
+    rake all:stop PORTS=5000-5010
+  
+ Start 5 Remote Controls with the Hub, ports 6500 to 6505
+ --------------------------------------------------------
+
+   rake rc:start_all PORTS=6500-6505
+
+ Start a single Remote Control on port 6666
+-------------------------------------------
+
+    rake rc:start PORT=6666
+
+   
+  EOS
+end
+
+desc "Restart all  services (Hub a Remote Controls"
+task :'all:restart' do
+  Rake::Task[:'all:stop'].execute([]) rescue nil
+  Rake::Task[:'all:start'].execute []
+end
+
+desc "Launch all  services (Hub a Remote Controls"
+task :'all:start' do
+  ENV['BACKGROUND'] = "true"
+  Rake::Task[:'hub:start'].execute []
+  puts "Waiting for Hub to come up..."
+  hub.wait_until_up_and_running
+  Rake::Task[:'rc:start_all'].execute []
+end
+
+desc "Stop all  services (Hub a Remote Controls"
+task :'all:stop' do
+  puts "Stopping all Selenium Grid servers..."
+  Rake::Task[:'rc:stop_all'].execute []
+  Rake::Task[:'hub:stop'].execute [] rescue nil
+end
+
+desc "Launch Hub"
+task :'hub:start' do
+  SGrid::Hub.new.start \
+      :background => ("true" == ENV['BACKGROUND']),
+      :log_file => File.native_path(File.dirname(__FILE__) + "/log/hub.log")
+end
+
+desc "Stop Hub"
+task :'hub:stop' do
+  puts "Shutting down Selenium Grid hub..."
+  hub.shutdown rescue EOFError
+end
+
+desc "Launch Remote Control"
+task :'rc:start' do
+  port = ENV['PORT'] || 5555  
+  remote_control(:port => port).start \
+      :background => ENV['BACKGROUND'],
+      :log_file => File.join(File.dirname(__FILE__), "log", "rc-#{port}.log")  
+end
+
+desc "Stop Remote Control"
+task :'rc:stop' do
+  remote_control.stop rescue EOFError
+end
+
+desc "Launch Remote Control"
+task :'rc:start_all' do
+  ports = ENV['PORTS'] || "5000-5015"
+  port_range = Range.new(*ports.split("-"))
+  port_range.each do |port|
+    remote_control(:port => port, :background => true).start \
+        :background => true,
+        :log_file => File.join(File.dirname(__FILE__), "log", "rc-#{port}.log")
+  end
+end
+
+desc"Stop Remote Controls. Usage rake rc:stop PORTS=5555-5560"
+task :'rc:stop_all' do
+  ports = ENV['PORTS'] || "5000-5015"
+  port_range = Range.new(*ports.split("-"))
+  port_range.each do |port|
+    begin
+      puts "Stopping Remote Control on port #{port}"
+      rc = SGrid::RemoteControl.new :host => ENV['HOST'] || "localhost",      
+                                    :port => port
+      rc.shutdown
+    rescue Exception => e
+      STDERR.puts "Could not properly shutdown remote control #{port} : #{e}"
+    end
+  end
+end
+
+desc "Launch a Xvnc server"
+task :'xvnc:start' do
+  sh "Xvnc :1 -alwaysshared -geometry 1280x1024 -depth 24 -desktop 'Selenium Grid' 2>&1 >#{File.join(File.dirname(__FILE__), "log", "X.log")} &"
+end
+
+desc "Fully Automated Demo (for continuous integration)"
+task :'ci:demo' do
+  begin
+    Rake::Task["all:restart"].execute []
+    sh "ant run-demo-in-parallel" unless running_on_openqa_servers
+    sh "ant run-demo-in-sequence"
+  ensure
+    Rake::Task["all:stop"].execute []
+  end
+end
+
+desc "Fully Automated Ruby Example (for continuous integration)"
+task :'ci:ruby-example' do
+  begin
+    Rake::Task["all:restart"].execute []
+    sh "(cd  examples/ruby; rake tests:run_in_parallel)" unless running_on_openqa_servers
+    sh "(cd  examples/ruby; rake tests:run_in_sequence)"
+  ensure
+    Rake::Task["all:stop"].execute []
+  end
+end
+
+def config
+  @config ||= YAML.load(File.read(File.dirname(__FILE__) + "/grid_configuration.yml"))
+end
+
+def hub_port
+  (ENV["HUB_PORT"] || config["hub"]["port"]).to_i
+end
+
+def hub
+  SGrid::Hub.new :port => hub_port
+end
+
+def remote_control(options={})
+  SGrid::RemoteControl.new(
+      :host => ENV['HOST'] || "localhost",
+      :port => (options[:port] || ENV['PORT'] || 5555),
+      :hub_url => (ENV['HUB_URL'] || "http://localhost:#{hub_port}"),
+      :shutdown_command => ENV['RC_SHUTDOWN_COMMAND'])
+end
+
+# Network is slow as hell on these VMs
+def running_on_openqa_servers
+  begin
+    not `hostname -s`.grep(/(xserve|osxvm)/).empty?
+  rescue Exception
+    false
+  end
+end

data/VERSION

@@ -0,0 +1 @@
+0.0.1

data/examples/ec2/lib/selenium_grid/aws/cloud.rb

@@ -0,0 +1,35 @@
+module SeleniumGrid
+  module AWS
+  
+    class Cloud
+      FILE = "cloud.yml"
+      attr_accessor :hub, :farms
+            
+      def self.load
+        begin
+          YAML.load(File.read(FILE))
+        rescue Errno::ENOENT
+          new
+        end
+      end
+
+      def self.update
+        cloud = self.load
+        yield cloud
+      ensure
+        cloud.write unless cloud.nil?
+      end
+      
+      def write 
+        File.open(FILE, "w") {|file| file.write(self.to_yaml)}
+      end
+      
+      def farms
+        @farms ||= []
+      end
+            
+    end
+
+  end
+end
+

data/examples/ec2/lib/selenium_grid/aws/ec2.rb

@@ -0,0 +1,7 @@
+require 'rubygems'
+require "yaml"
+require File.dirname(__FILE__) + '/cloud'
+require File.dirname(__FILE__) + '/ec2_client'
+require File.dirname(__FILE__) + '/server'
+require File.dirname(__FILE__) + '/hub'
+require File.dirname(__FILE__) + '/remote_command'

data/examples/ec2/lib/selenium_grid/aws/ec2_client.rb

@@ -0,0 +1,54 @@
+module SeleniumGrid
+  module AWS
+    module Ec2Client
+      
+      def describe(instance_id)
+        output = ec2_shell "ec2-describe-instances #{instance_id}"
+        output =~ /INSTANCE\s+(i-.*)$/
+        fields = $1.split(/\s+/)
+        if output =~ /running/
+          {:instance_id => fields[0],
+           :ami => fields[1],
+           :public_dns => fields[2],
+           :private_dns => fields[3],
+           :status => fields[4] }
+        else 
+          {:instance_id => fields[0],
+           :ami => fields[1],
+           :status => fields[2] }
+        end
+      end
+            
+      def launch(ami, options ={})
+        output = ec2_shell "ec2-run-instances #{ami} -k #{options[:keypair]}"
+        output =~ /INSTANCE\s+(i-\S+)\s+ami-/
+        $1
+      end
+
+      def shutdown(instance_id)
+        ec2_shell "ec2-terminate-instances #{instance_id}"
+      end
+
+      def version
+        ec2_shell "ec2-version"
+      end
+            
+      def authorize_port(port)
+        puts "Opening port #{port}..."        
+        ec2_shell "ec2-authorize default -p #{port}"
+      end
+            
+      def ec2_shell(command)
+        puts "[EC2] '#{command}'" if tracing?
+        output = `${EC2_HOME}/bin/#{command}`
+        puts "[EC2] #{output}" if tracing?
+        output
+      end
+
+      def tracing?
+        ENV['TRACE_EC2_COMMANDS']
+      end
+              
+    end
+  end
+end

data/examples/ec2/lib/selenium_grid/aws/hub.rb

@@ -0,0 +1,22 @@
+module SeleniumGrid
+  module AWS
+  
+    class Hub < Server
+      
+      def url
+        "http://#{public_dns}:4444"
+      end
+
+      def private_url
+        "http://#{private_dns}:4444"
+      end
+
+      def console_url
+        "#{url}/console"
+      end
+      
+    end
+    
+  end
+end
+

data/examples/ec2/lib/selenium_grid/aws/remote_command.rb

@@ -0,0 +1,47 @@
+module SeleniumGrid
+  module AWS
+    
+    class RemoteCommand
+      attr_accessor :options
+      
+      def initialize(command, options={})
+        @command, @options = command, options
+      end
+      
+      def execute
+        puts full_command
+        system full_command
+        raise "Error with #{full_command}" if 0 != $?        
+      end
+      
+      def full_command
+        cmd = "#{ssh_command} " 
+        cmd << "\"su -l #{options[:su]} -c " if options[:su]
+        cmd << "'#{remote_command}'"
+        cmd << '"' if options[:su]
+        cmd
+      end
+      
+      def ssh_command
+        shell_command = [ "ssh" ]
+        shell_command << "-i '#{options[:keypair]}'" if options[:keypair]
+        shell_command << "root@#{options[:host]}"
+        
+        shell_command.join " "
+      end
+
+      def remote_command
+        shell_command = []
+        shell_command << "PATH=#{options[:path]}:${PATH}; export PATH;" if options[:path]
+        shell_command << "DISPLAY=#{options[:display]}; export DISPLAY;" if options[:display]
+        shell_command << "cd '#{options[:pwd]}';" if options[:pwd]
+        shell_command << @command
+        
+        shell_command.join " "
+      end
+      
+    end  
+
+  end
+end
+

data/examples/ec2/lib/selenium_grid/aws/server.rb

@@ -0,0 +1,56 @@
+module SeleniumGrid
+  module AWS
+  
+    class Server
+      extend Ec2Client
+      
+      attr_accessor :instance_id, :public_dns, :private_dns
+    
+      def initialize(instance_id)
+        self.instance_id = instance_id
+      end       
+
+      def self.boot(ami, options = {})
+        new launch(ami, options)
+      end
+
+      def self.boot_and_acquire_dns(ami, options)
+        server = boot(ami, options)
+        server.wait_for_dns
+      end
+
+      def wait_for_dns
+        puts "Fetching DNS Info..."
+        until dns_allocated?
+          sleep 2
+          putc "."
+          refresh_status
+        end
+        puts
+        self
+      end
+      
+      def dns_allocated?
+        public_dns != nil && public_dns != "" &&
+        private_dns != nil && private_dns != ""
+      end
+      
+      def refresh_status
+        info = self.class.describe instance_id
+        @public_dns = info[:public_dns]
+        @private_dns = info[:private_dns]
+      end
+
+      def shutdown
+        self.class.shutdown instance_id
+      end
+            
+      def run(command, options)
+        command = RemoteCommand.new command, options.merge(:host => public_dns)
+        command.execute
+      end         
+    end
+
+  end
+end
+

data/examples/ec2/test/unit/all_test.rb

@@ -0,0 +1,2 @@
+#!/usr/bin/env ruby
+Dir['**/*_test.rb'].each {|test_file| require test_file }

data/examples/ec2/test/unit/cloud_test.rb

@@ -0,0 +1,27 @@
+require File.dirname(__FILE__) + "/test_helper"
+
+unit_tests do
+
+  test "hub is nil on a fresh cloud" do
+    assert_nil SeleniumGrid::AWS::Cloud.new.hub
+  end
+
+  test "hub return the latest assigned hub instance" do
+    cloud = SeleniumGrid::AWS::Cloud.new
+    cloud.hub = :old_hub
+    cloud.hub = :new_hub
+    assert_equal :new_hub, cloud.hub
+  end
+
+  test "remote_control_farms is empty on a fresh cloud" do
+    assert_equal [], SeleniumGrid::AWS::Cloud.new.farms
+  end
+
+  test "remote_control_farms returns all added farms" do
+    cloud = SeleniumGrid::AWS::Cloud.new
+    cloud.farms << :a_farm
+    cloud.farms << :another_farm
+    assert_equal [:a_farm, :another_farm], cloud.farms
+  end
+
+end

data/examples/ec2/test/unit/ec2_client_test.rb

@@ -0,0 +1,137 @@
+require File.dirname(__FILE__) + "/test_helper"
+
+unit_tests do
+
+  test "run_instance launch a new AMI using ec2-run-instances script" do
+    client = Class.new.extend SeleniumGrid::AWS::Ec2Client
+    client.expects(:ec2_shell).with("ec2-run-instances TheAMI -k TheKeyPair")
+    client.launch "TheAMI", :keypair => "TheKeyPair"
+  end
+
+  test "run_instance returns the instance id when launch is successful" do
+    client = Class.new.extend SeleniumGrid::AWS::Ec2Client
+    client.stubs(:ec2_shell).returns(<<-EOS)
+      RESERVATION	r-ee629587	069216575636	default
+      INSTANCE	i-6fef1006	ami-d306e3ba			pending	grid-keypair	0		m1.small	2008-02-17T20:49:08+0000
+    EOS
+    assert_equal "i-6fef1006", client.launch(:an_ami)
+  end
+
+  test "authorize launch ec2-authorize script for default group" do
+    client = Class.new.extend SeleniumGrid::AWS::Ec2Client
+    client.expects(:ec2_shell).with("ec2-authorize default -p ThePort")
+    client.authorize_port "ThePort"
+  end
+
+  test "describe launch ec2-describe script for a particular AMI" do
+    client = Class.new.extend SeleniumGrid::AWS::Ec2Client
+    client.expects(:ec2_shell).with("ec2-describe-instances TheAMI").returns("INSTANCE i-")
+    client.describe "TheAMI"
+  end
+
+  test "describe returns the instance id when running" do
+    client = Class.new.extend SeleniumGrid::AWS::Ec2Client
+    client.stubs(:ec2_shell).returns(<<-EOS)
+      RESERVATION	r-ee629587	069216575636	default
+      INSTANCE	i-6fef1006	ami-d306e3ba	ec2-67-202-19-143.compute-1.amazonaws.com	domU-12-31-38-00-3D-E6.compute-1.internal	running	grid-keypair	0		m1.small	2008-02-17T20:49:08+0000      
+    EOS
+    assert_equal "i-6fef1006", client.describe(:an_ami)[:instance_id]
+  end
+
+  test "describe returns the ami when running" do
+    client = Class.new.extend SeleniumGrid::AWS::Ec2Client
+    client.stubs(:ec2_shell).returns(<<-EOS)
+      RESERVATION	r-ee629587	069216575636	default
+      INSTANCE	i-6fef1006	ami-d306e3ba	ec2-67-202-19-143.compute-1.amazonaws.com	domU-12-31-38-00-3D-E6.compute-1.internal	running	grid-keypair	0		m1.small	2008-02-17T20:49:08+0000      
+    EOS
+    assert_equal "ami-d306e3ba", client.describe(:an_ami)[:ami]
+  end
+
+  test "describe returns the public_dns when running" do
+    client = Class.new.extend SeleniumGrid::AWS::Ec2Client
+    client.stubs(:ec2_shell).returns(<<-EOS)
+      RESERVATION	r-ee629587	069216575636	default
+      INSTANCE	i-6fef1006	ami-d306e3ba	ec2-67-202-19-143.compute-1.amazonaws.com	domU-12-31-38-00-3D-E6.compute-1.internal	running	grid-keypair	0		m1.small	2008-02-17T20:49:08+0000      
+    EOS
+    assert_equal "ec2-67-202-19-143.compute-1.amazonaws.com", 
+                 client.describe(:an_ami)[:public_dns]
+  end
+
+  test "describe returns the private_dns when running" do
+    client = Class.new.extend SeleniumGrid::AWS::Ec2Client
+    client.stubs(:ec2_shell).returns(<<-EOS)
+      RESERVATION	r-ee629587	069216575636	default
+      INSTANCE	i-6fef1006	ami-d306e3ba	ec2-67-202-19-143.compute-1.amazonaws.com	domU-12-31-38-00-3D-E6.compute-1.internal	running	grid-keypair	0		m1.small	2008-02-17T20:49:08+0000      
+    EOS
+    assert_equal "domU-12-31-38-00-3D-E6.compute-1.internal", 
+                 client.describe(:an_ami)[:private_dns]
+  end
+
+  test "describe returns the status when running" do
+    client = Class.new.extend SeleniumGrid::AWS::Ec2Client
+    client.stubs(:ec2_shell).returns(<<-EOS)
+      RESERVATION	r-ee629587	069216575636	default
+      INSTANCE	i-6fef1006	ami-d306e3ba	ec2-67-202-19-143.compute-1.amazonaws.com	domU-12-31-38-00-3D-E6.compute-1.internal	running	grid-keypair	0		m1.small	2008-02-17T20:49:08+0000      
+    EOS
+    assert_equal "running", client.describe(:an_ami)[:status]
+  end
+
+  test "describe returns the instance id when terminated" do
+    client = Class.new.extend SeleniumGrid::AWS::Ec2Client
+    client.stubs(:ec2_shell).returns(<<-EOS)
+      RESERVATION	r-ee629587	069216575636	default
+      INSTANCE	i-6fef1006	ami-d306e3ba			terminated	grid-keypair	0		m1.small	2008-02-17T20:49:08+0000
+    EOS
+    assert_equal "i-6fef1006", client.describe(:an_ami)[:instance_id]
+  end
+
+  test "describe returns the status when pending" do
+    client = Class.new.extend SeleniumGrid::AWS::Ec2Client
+    client.stubs(:ec2_shell).returns(<<-EOS)
+      RESERVATION	r-40609729	069216575636	default
+      INSTANCE	i-afee11c6	ami-6801e401			pending	grid-keypair	0		m1.small	2008-02-17T22:33:38+0000
+    EOS
+    assert_equal "pending", client.describe(:an_ami)[:status]
+  end
+
+  test "describe returns a nil public dns the status when pending" do
+    client = Class.new.extend SeleniumGrid::AWS::Ec2Client
+    client.stubs(:ec2_shell).returns(<<-EOS)
+      RESERVATION	r-40609729	069216575636	default
+      INSTANCE	i-afee11c6	ami-6801e401			pending	grid-keypair	0		m1.small	2008-02-17T22:33:38+0000
+    EOS
+    assert_nil client.describe(:an_ami)[:public_dns]
+  end
+
+  test "describe returns a nil private dns the status when pending" do
+    client = Class.new.extend SeleniumGrid::AWS::Ec2Client
+    client.stubs(:ec2_shell).returns(<<-EOS)
+      RESERVATION	r-40609729	069216575636	default
+      INSTANCE	i-afee11c6	ami-6801e401			pending	grid-keypair	0		m1.small	2008-02-17T22:33:38+0000
+    EOS
+    assert_nil client.describe(:an_ami)[:private_dns]
+  end
+
+  test "describe returns the status when terminated" do
+    client = Class.new.extend SeleniumGrid::AWS::Ec2Client
+    client.stubs(:ec2_shell).returns(<<-EOS)
+      RESERVATION	r-ee629587	069216575636	default
+      INSTANCE	i-6fef1006	ami-d306e3ba			terminated	grid-keypair	0		m1.small	2008-02-17T20:49:08+0000
+    EOS
+    assert_equal "terminated", client.describe(:an_ami)[:status]
+  end
+
+  test "shutdown terminates an instance" do
+    client = Class.new.extend SeleniumGrid::AWS::Ec2Client
+    client.expects(:ec2_shell).with("ec2-terminate-instances The-Instance-ID")
+    client.shutdown("The-Instance-ID")
+  end
+
+  test "version returns EC2 version using defined keypair" do
+    client = Class.new.extend SeleniumGrid::AWS::Ec2Client
+    client.expects(:ec2_shell).with("ec2-version").returns(:ec2_version)
+    assert_equal :ec2_version, client.version
+  end
+
+# ec2-version -K "${EC2_PRIVATE_KEY}"
+end

data/examples/ec2/test/unit/hub_test.rb

@@ -0,0 +1,24 @@
+require File.dirname(__FILE__) + "/test_helper"
+
+unit_tests do
+
+  test "url return the public url where the hub can be contacted" do
+    hub = SeleniumGrid::AWS::Hub.new nil
+    hub.public_dns = "public.dns"
+    assert_equal "http://public.dns:4444", hub.url
+  end
+
+  test "private_url return the private url where the hub can be contacted" do
+    hub = SeleniumGrid::AWS::Hub.new nil
+    hub.private_dns = "private.dns"
+    assert_equal "http://private.dns:4444", hub.private_url
+  end
+
+  test "console_url return the public url of the hub console" do
+    hub = SeleniumGrid::AWS::Hub.new nil
+    hub.public_dns = "public.dns"
+    assert_equal "http://public.dns:4444/console", hub.console_url
+  end
+
+
+end

data/examples/ec2/test/unit/remote_command_test.rb

@@ -0,0 +1,46 @@
+require File.dirname(__FILE__) + "/test_helper"
+
+unit_tests do
+
+  test "ssh_command builds ssh based command targetting the host" do
+    command = SeleniumGrid::AWS::RemoteCommand.new "ls", :host => "the.host"
+    assert_equal "ssh root@the.host", command.ssh_command
+  end
+
+  test "ssh_command use custom key when keypair options is provided" do
+    command = SeleniumGrid::AWS::RemoteCommand.new "ls", 
+                :host => "the.host", :keypair => "/the/key.pair"
+    assert_equal "ssh -i '/the/key.pair' root@the.host", command.ssh_command
+  end
+
+  test "remote_command" do
+    command = SeleniumGrid::AWS::RemoteCommand.new "ls", :pwd => "/a/directory"
+    assert_equal "cd '/a/directory'; ls", command.remote_command
+  end
+
+  test "remote_command set path when path is provided as an option" do
+    command = SeleniumGrid::AWS::RemoteCommand.new "ls", :path => "/a/directory:and/another"
+    assert_equal "PATH=/a/directory:and/another:${PATH}; export PATH; ls", command.remote_command
+  end
+
+  test "remote_command set display when display is provided as an option" do
+    command = SeleniumGrid::AWS::RemoteCommand.new "ls", :display => ":0"
+    assert_equal "DISPLAY=:0; export DISPLAY; ls", command.remote_command
+  end
+
+  test "full_command execute the remote command using ssh_command" do
+    command = SeleniumGrid::AWS::RemoteCommand.new nil
+    command.stubs(:ssh_command).returns("the_ssh_command")
+    command.stubs(:remote_command).returns("the remote command")
+    assert_equal "the_ssh_command 'the remote command'", command.full_command
+  end
+
+  test "full_command wraps remote_command with 'su user -c' when su option is set" do
+    command = SeleniumGrid::AWS::RemoteCommand.new nil, :su => "a-user"
+    command.stubs(:ssh_command).returns("the_ssh_command")
+    command.stubs(:remote_command).returns("the remote command")
+    assert_equal "the_ssh_command \"su -l a-user -c 'the remote command'\"", 
+                 command.full_command
+  end
+  
+end

data/examples/ec2/test/unit/server_test.rb

@@ -0,0 +1,22 @@
+require File.dirname(__FILE__) + "/test_helper"
+
+unit_tests do
+
+  test "refresh_status updates public dns" do
+    server = SeleniumGrid::AWS::Server.new :an_instance_id
+    SeleniumGrid::AWS::Server.expects(:describe).with(:an_instance_id).
+                              returns(:public_dns => :new_public_dns)
+    server.refresh_status
+    assert_equal :new_public_dns, server.public_dns
+  end
+
+  test "refresh_status updates private dns" do
+    server = SeleniumGrid::AWS::Server.new :an_instance_id
+    SeleniumGrid::AWS::Server.expects(:describe).with(:an_instance_id).
+                              returns(:private_dns => :new_private_dns)
+    server.refresh_status
+    assert_equal :new_private_dns, server.private_dns
+  end
+
+  
+end

data/examples/ec2/test/unit/test_helper.rb

@@ -0,0 +1,4 @@
+require File.dirname(__FILE__) + "/../../lib/selenium_grid/aws/ec2"
+require "rubygems"
+require "dust"
+require "mocha"

data/examples/ruby/flickr_example.rb

@@ -0,0 +1,16 @@
+module FlickrExample
+  
+  def run_flickr_scenario(options)
+    browser.open "/"
+    page.location.should match(%r{http://images.google.com/})
+    page.type "q", options[:search_string]
+    page.click "btnG", :wait_for => :page
+    page.click "link=Advanced Image Search", :wait_for => :page
+    page.click "rimgtype4"
+    page.click "sf"
+    page.select "imgc", "full color"
+    page.click "btnG", :wait_for => :page
+    page.text?(options[:search_string].split(/ /).first).should be_true
+  end
+  
+end

data/examples/ruby/lib/array_extension.rb

@@ -0,0 +1,13 @@
+module ArrayExtension
+
+  def /(number_of_buckets)
+    buckets = (1..number_of_buckets).collect { [] }
+    while self.any? do
+      buckets.each do |bucket|
+        bucket << self.shift if self.any?
+      end
+    end
+    buckets
+  end
+
+end

data/examples/ruby/lib/multi_process_behaviour_runner.rb

@@ -0,0 +1,28 @@
+require File.expand_path(File.dirname(__FILE__) + '/array_extension')
+Array.send :include, ArrayExtension
+
+class MultiProcessSpecRunner
+
+  def initialize(max_concurrent_processes = 10)
+    @max_concurrent_processes = max_concurrent_processes
+  end
+  
+  def run(spec_files)
+    concurrent_processes = [ @max_concurrent_processes, spec_files.size ].min
+    spec_files_by_process = spec_files / concurrent_processes
+    concurrent_processes.times do |i|
+      cmd  = "spec #{spec_files_by_process[i].join(' ')}"
+      puts "Launching #{cmd}"
+      exec(cmd) if fork == nil
+    end
+    success = true
+    concurrent_processes.times do |i|
+      pid, status = Process.wait2
+      puts "Test process ##{i} with pid #{pid} completed with #{status}"
+      success &&= status.exitstatus.zero?
+    end
+    
+    raise "Build failed" unless success
+  end
+
+end

data/examples/ruby/paris_spec.rb

@@ -0,0 +1,34 @@
+require File.expand_path(File.dirname(__FILE__) + "/spec_helper")
+
+describe "Paris" do
+  include FlickrExample
+  
+  it "Has museums" do
+    run_flickr_scenario :search_string => "Louvre"
+  end
+
+  it "Has bridges" do
+    run_flickr_scenario :search_string => "Pont Neuf"
+  end
+
+  it "Has magnific cathedrals" do
+    run_flickr_scenario :search_string => "Notre Dame de Paris"
+  end
+
+  it "Has magnificent Castles" do
+    run_flickr_scenario :search_string => "Versailles"
+  end
+
+  it "Has a gorgeous river" do
+    run_flickr_scenario :search_string => "Seine by Night"
+  end
+
+  it "Has weird towers" do
+    run_flickr_scenario :search_string => "Tour Eiffel"
+  end
+
+  it "Has avenues" do
+    run_flickr_scenario :search_string => "Avenue des Champs Elysees"
+  end
+
+end

data/examples/ruby/perigord_spec.rb

@@ -0,0 +1,22 @@
+require File.expand_path(File.dirname(__FILE__) + "/spec_helper")
+
+describe "Perigord" do
+  include FlickrExample
+  
+  it "Has magnific cro-magnon paintings" do
+    run_flickr_scenario :search_string => "lascaux hall of the bull"
+  end
+
+  it "Has magnific cities" do
+    run_flickr_scenario :search_string => "Sarlat"
+  end
+
+  it "Has magnific cathedrals" do
+    run_flickr_scenario :search_string => "Cathedral in Périgueux"
+  end
+
+  it "Has great wines" do
+    run_flickr_scenario :search_string => "Montbazillac"
+  end
+
+end

data/examples/ruby/spec_helper.rb

@@ -0,0 +1,59 @@
+$:.unshift
+
+require "rubygems"
+gem "rspec", "=1.1.8"
+require 'spec/rake/spectask'
+
+require "rake"
+
+gem "selenium-client", "=1.2.7"
+require "selenium/rake/tasks"
+require "selenium/client"
+require "selenium/rspec/spec_helper"
+
+require File.expand_path(File.dirname(__FILE__) + "/flickr_example")
+
+Spec::Runner.configure do |config|
+
+  config.before(:each) do
+    create_selenium_driver
+    start_new_browser_session
+  end
+
+  # The system capture need to happen BEFORE the closing the Selenium session 
+  config.append_after(:each) do    
+    @selenium_driver.close_current_browser_session
+  end
+
+  def start_new_browser_session
+    @selenium_driver.start_new_browser_session
+    @selenium_driver.set_context "Starting example '#{self.description}'"
+  end
+
+  def selenium_driver
+    @selenium_driver
+  end
+
+  def browser
+    @selenium_driver
+  end
+
+  def page
+    @selenium_driver
+  end
+
+  def create_selenium_driver
+    remote_control_server = ENV['SELENIUM_RC_HOST'] || "localhost"
+    port = ENV['SELENIUM_RC_PORT'] || 4444
+    browser = ENV['SELENIUM_RC_BROWSER'] || "*firefox"
+    timeout = ENV['SELENIUM_RC_TIMEOUT'] || 200
+    application_host = ENV['SELENIUM_APPLICATION_HOST'] || "images.google.com"
+    application_port = ENV['SELENIUM_APPLICATION_PORT'] || "80"
+
+    @selenium_driver = Selenium::Client::Driver.new(
+        remote_control_server, port, browser,
+        "http://#{application_host}:#{application_port}", timeout)
+  end
+
+end
+

data/test/helper.rb

@@ -0,0 +1,10 @@
+require 'rubygems'
+require 'test/unit'
+require 'shoulda'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'davidtrogers-selenium-grid'
+
+class Test::Unit::TestCase
+end

data/test/test_davidtrogers-selenium-grid.rb

@@ -0,0 +1,7 @@
+require 'helper'
+
+class TestDavidtrogersSeleniumGrid < Test::Unit::TestCase
+  should "probably rename this file and start testing for real" do
+    flunk "hey buddy, you should probably rename this file and start testing for real"
+  end
+end

metadata

@@ -0,0 +1,104 @@
+--- !ruby/object:Gem::Specification 
+name: davidtrogers-selenium-grid
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Dave Rogers
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-02-24 00:00:00 -05:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: thoughtbot-shoulda
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id001
+description: ""
+email: david.t.rogers@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README
+- README.rdoc
+files: 
+- .document
+- .gitignore
+- LICENSE
+- README.rdoc
+- Rakefile
+- VERSION
+- lib/davidtrogers-selenium-grid.rb
+- test/helper.rb
+- test/test_davidtrogers-selenium-grid.rb
+- README
+has_rdoc: true
+homepage: 
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: selenium grid packaged as a gem
+test_files: 
+- test/helper.rb
+- test/test_davidtrogers-selenium-grid.rb
+- examples/ec2/lib/selenium_grid/aws/cloud.rb
+- examples/ec2/lib/selenium_grid/aws/ec2.rb
+- examples/ec2/lib/selenium_grid/aws/ec2_client.rb
+- examples/ec2/lib/selenium_grid/aws/hub.rb
+- examples/ec2/lib/selenium_grid/aws/remote_command.rb
+- examples/ec2/lib/selenium_grid/aws/server.rb
+- examples/ec2/test/unit/all_test.rb
+- examples/ec2/test/unit/cloud_test.rb
+- examples/ec2/test/unit/ec2_client_test.rb
+- examples/ec2/test/unit/hub_test.rb
+- examples/ec2/test/unit/remote_command_test.rb
+- examples/ec2/test/unit/server_test.rb
+- examples/ec2/test/unit/test_helper.rb
+- examples/ruby/flickr_example.rb
+- examples/ruby/lib/array_extension.rb
+- examples/ruby/lib/multi_process_behaviour_runner.rb
+- examples/ruby/paris_spec.rb
+- examples/ruby/perigord_spec.rb
+- examples/ruby/spec_helper.rb