rwebspec 4.0 → 4.1.0

data/CHANGELOG

@@ -1,6 +1,12 @@
 CHANGELOG
 =========
 
+4.1
+[Fixed] - click_button, click_button_with_id support :index
+[Refactored] - unit tests
+[Feature] - Save screenshot for Selenium
+[CHANGE] - open_browser(base_url, options) => open_browser(:base_url => "http://")
+
 4.0
 - Merge rwebspec-webdriver with rwebspec!
 

data/README

@@ -9,7 +9,7 @@ specification "User Profile" do
   include TestHelper
 
   before(:all) do
-    open_browser("http://demo.adminwise.com")
+    open_browser(:base_url => "http://demo.adminwise.com")
     reset_database
   end
 

data/Rakefile

@@ -1,7 +1,8 @@
 require 'rubygems'
 require 'rspec/core/rake_task'
 # require 'rake/rdoctask'
-require 'rake/gempackagetask'
+require 'rubygems/package_task'
+
 require 'rdoc' # require rdoc 2
 # gem 'darkfish-rdoc'
 # require 'darkfish-rdoc'
@@ -24,7 +25,7 @@ end
 
 desc 'Run all specs'
 RSpec::Core::RakeTask.new('spec') do |t|
-  t.rspec_opts = ['--format', 'specdoc', '--colour']
+  t.rspec_opts = ['']
   # t.libs = ["lib", "server/lib" ]
   t.pattern = Dir['spec/**/*_spec.rb'].sort
 end
@@ -77,7 +78,7 @@ end
 spec = Gem::Specification.new do |s|
   s.platform= Gem::Platform::RUBY
   s.name = "rwebspec"
-  s.version = "4.0"
+  s.version = "4.1.0"
   s.summary = "Web application functional specification in Ruby"
   s.description = "Executable functional specification for web applications in RSpec syntax with Watir or Selenium"
 
@@ -97,13 +98,13 @@ spec = Gem::Specification.new do |s|
   s.files = s.files + Dir.glob( "sample/**/*")
   s.files = s.files + Dir.glob( "docs/**/*" )
   s.add_dependency(%q<rspec>, [">= 2.10"])
-  s.add_dependency(%q<rspec-core>, [">= 2.10.1"])
+  s.add_dependency(%q<rspec-core>, ["= 2.10.1"])
   s.add_dependency("commonwatir", ">= 3.0")
-  if RUBY_PLATFORM =~ /mingw/
+  unless RUBY_PLATFORM =~ /mingw/
     s.add_dependency("selenium-webdriver")
   end
 end
 
-Rake::GemPackageTask.new(spec) do |pkg|
+Gem::PackageTask.new(spec) do |pkg|
   pkg.need_zip = true
 end

data/lib/plugins/testwise_plugin.rb

@@ -57,7 +57,11 @@ module RWebSpec
 			end
 
     def notify_screenshot_location(image_file_path)
-      connect_to_testwise("  SHOT", image_file_path)
+      begin
+        connect_to_testwise("  SHOT", image_file_path)
+      rescue => e
+        puts "[DEBUG] failed to notify TestWise a screenshot"
+      end
     end
 
     # find out the line (and file) the execution is on, and notify iTest via Socket

data/lib/rwebspec-common/assert.rb

@@ -80,11 +80,13 @@ module RWebSpec
     #  assert_link_present_with_text("Click ") # => 
     #
     def assert_link_present_with_text(link_text)
+			
       @web_browser.links.each { |link|
         return if link.text.include?(link_text)
       }
       fail( "can't find the link containing text: #{link_text}")
-    end
+  
+  	end
 
     def assert_link_not_present_with_text(link_text)
       @web_browser.links.each { |link|
@@ -92,23 +94,43 @@ module RWebSpec
       }
     end
 
+		def is_selenium_element?(elem)
+			elem.class.name =~ /Selenium/
+		end
+
+		def element_name(elem)
+			elem.class.name =~ /Selenium/ ? elem['name'] : elem.name 			
+		end
+
+		def element_value(elem)
+			elem.class.name =~ /Selenium/ ? elem['value'] : elem.value 			
+		end
 
     ##
     # Checkbox
     def assert_checkbox_not_selected(checkbox_name)
       @web_browser.checkboxes.each { |checkbox|
-        if (checkbox.name == checkbox_name) then
-          perform_assertion {  assert(!checkbox.set?, "Checkbox #{checkbox_name} is checked unexpectly") }
+				the_element_name = element_name(checkbox)
+        if (the_element_name == checkbox_name) then		
+	 				if is_selenium_element?(checkbox)
+          	perform_assertion {  assert(!checkbox.selected?, "Checkbox #{checkbox_name} is checked unexpectly") }			
+					else
+          	perform_assertion {  assert(!checkbox.set?, "Checkbox #{checkbox_name} is checked unexpectly") }
+					end
         end
       }
     end
-
     alias assert_checkbox_not_checked assert_checkbox_not_selected
 
     def assert_checkbox_selected(checkbox_name)
       @web_browser.checkboxes.each { |checkbox|
-        if (checkbox.name == checkbox_name) then
-          perform_assertion { assert(checkbox.set?, "Checkbox #{checkbox_name} not checked") }
+				the_element_name = element_name(checkbox)
+        if (the_element_name == checkbox_name) then
+					if is_selenium_element?(checkbox)	 	
+          	perform_assertion { assert(checkbox.selected?, "Checkbox #{checkbox_name} not checked") }
+					else
+          	perform_assertion { assert(checkbox.set?, "Checkbox #{checkbox_name} not checked") }					
+					end
         end
       }
     end
@@ -130,7 +152,8 @@ module RWebSpec
 
     def assert_option_not_present(select_name, option_label)
       @web_browser.select_lists.each { |select|
-        next unless select.name == select_name
+				the_element_name = element_name(select)
+        next unless the_element_name == select_name
         select.options.each do |option| # items in the list
           perform_assertion {  assert(!(option.text == option_label), "unexpected select option: #{option_label} for #{select_name} found") }
         end
@@ -141,10 +164,19 @@ module RWebSpec
 
     def assert_option_value_present(select_name, option_value)
       @web_browser.select_lists.each { |select|
-        next unless select.name == select_name
-        select.options.each do |option| # items in the list
-          return if option.value == option_value
-        end
+				the_element_name = element_name(select)
+        next unless the_element_name  == select_name
+
+				if RWebSpec.framework == "Watir"
+	        select.options.each do |option| # items in the list
+	          return if option.value == option_value
+	        end					
+				else					
+					select.find_elements(:tag_name => "option" ).each do |option|
+						return if element_value(option) == option_value
+					end
+				end
+				
       }
       fail("can't find the combo box with value: #{option_value}")
     end
@@ -154,12 +186,21 @@ module RWebSpec
 
     def assert_option_present(select_name, option_label)
       @web_browser.select_lists.each { |select|
-        next unless select.name == select_name
-        select.options.each do |option| # items in the list
-          return if option.text == option_label
-        end
+				the_element_name = element_name(select)
+        next unless the_element_name == select_name
+
+				if RWebSpec.framework == "Watir"
+	        select.options.each do |option| # items in the list
+	          return if option.text == option_label
+	        end
+				else
+					select.find_elements(:tag_name => "option" ).each do |option|
+	          return if option.text == option_label
+					end			
+				end
+
       }
-      fail("can't find the combob box: #{select_name} with value: #{option_label}")
+      fail("can't find the combo box: #{select_name} with value: #{option_label}")
     end
 
     alias assert_select_label_present assert_option_present
@@ -167,12 +208,25 @@ module RWebSpec
 
     def assert_option_equals(select_name, option_label)
       @web_browser.select_lists.each { |select|
-        next unless select.name == select_name
-        select.options.each do |option| # items in the list
-          if (option.text == option_label) then
-            perform_assertion { assert_equal(select.value, option.value, "Select #{select_name}'s value is not equal to expected option label: '#{option_label}'") }
-          end
-        end
+				the_element_name = element_name(select)
+        next unless the_element_name == select_name
+
+				if RWebSpec.framework == "Watir"
+
+	        select.options.each do |option| # items in the list
+	          if (option.text == option_label) then
+	            perform_assertion { assert_equal(select.value, option.value, "Select #{select_name}'s value is not equal to expected option label: '#{option_label}'") }
+	          end
+	        end
+	
+				else
+					select.find_elements(:tag_name => "option" ).each do |option|
+		        if (option.text == option_label) then
+							assert option.selected?
+						end
+					end
+					
+				end
       }
     end
 
@@ -181,8 +235,14 @@ module RWebSpec
 
     def assert_option_value_equals(select_name, option_value)
       @web_browser.select_lists.each { |select|
-        next unless select.name == select_name
-        perform_assertion {  assert_equal(select.value, option_value, "Select #{select_name}'s value is not equal to expected: '#{option_value}'") }
+				the_element_name = element_name(select)
+        next unless the_element_name == select_name
+
+				if RWebSpec.framework == "Watir"
+	        perform_assertion {  assert_equal(select.value, option_value, "Select #{select_name}'s value is not equal to expected: '#{option_value}'") }
+				else
+	        perform_assertion {  assert_equal(element_value(select), option_value, "Select #{select_name}'s value is not equal to expected: '#{option_value}'") }					
+				end
       }
     end
 
@@ -195,24 +255,31 @@ module RWebSpec
     # radio_group is the name field, radio options 'value' field
     def assert_radio_option_not_present(radio_group, radio_option)
       @web_browser.radios.each { |radio|
-        if (radio.name == radio_group) then
-          perform_assertion {  assert(!(radio_option == radio.value), "unexpected radio option: " + radio_option  + " found") }
+				the_element_name = element_name(radio)					
+        if (the_element_name == radio_group) then
+          perform_assertion {  assert(!(radio_option == element_value(radio) ), "unexpected radio option: " + radio_option  + " found") }
         end
       }
     end
 
     def assert_radio_option_present(radio_group, radio_option)
       @web_browser.radios.each { |radio|
-        return if (radio.name == radio_group) and (radio_option == radio.value)
+				the_element_name = element_name(radio)				
+        return if (the_element_name == radio_group) and (radio_option == element_value(radio) )
       }
       fail("can't find the radio option : '#{radio_option}'")
     end
 
     def assert_radio_option_selected(radio_group, radio_option)
       @web_browser.radios.each { |radio|
-        if (radio.name == radio_group and radio_option == radio.value) then
-          perform_assertion { assert(radio.set?, "Radio button #{radio_group}-[#{radio_option}] not checked") }
-        end
+				the_element_name = element_name(radio)					
+        if (the_element_name == radio_group and radio_option == element_value(radio) ) then
+					if is_selenium_element?(radio)
+          	perform_assertion { assert(radio.selected?, "Radio button #{radio_group}-[#{radio_option}] not checked") }
+					else
+          	perform_assertion { assert(radio.set?, "Radio button #{radio_group}-[#{radio_option}] not checked") }					
+					end  
+      	end
       }
     end
 
@@ -221,8 +288,13 @@ module RWebSpec
 
     def assert_radio_option_not_selected(radio_group, radio_option)
       @web_browser.radios.each { |radio|
-        if (radio.name == radio_group and radio_option == radio.value) then
-          perform_assertion {  assert(!radio.set?, "Radio button #{radio_group}-[#{radio_option}] checked unexpected") }
+				the_element_name = element_name(radio)						
+        if (the_element_name == radio_group and radio_option == element_value(radio) ) then	
+					if is_selenium_element?(radio)
+          	perform_assertion {  assert(!radio.selected?, "Radio button #{radio_group}-[#{radio_option}] checked unexpected") }
+					else
+          	perform_assertion {  assert(!radio.set?, "Radio button #{radio_group}-[#{radio_option}] checked unexpected") }
+					end
         end
       }
     end
@@ -234,26 +306,28 @@ module RWebSpec
     # Button
     def assert_button_not_present(button_id)
       @web_browser.buttons.each { |button|
-        perform_assertion { assert(button.id != button_id, "unexpected button id: #{button_id} found") }
+				the_button_id = RWebSpec.framework == "Watir" ? button.id : button["id"]	
+        perform_assertion { assert(the_button_id != button_id, "unexpected button id: #{button_id} found") }
       }
     end
 
     def assert_button_not_present_with_text(text)
       @web_browser.buttons.each { |button|
-        perform_assertion { assert(button.value != text, "unexpected button id: #{text} found") }
+        perform_assertion { assert(element_value(button) != text, "unexpected button id: #{text} found") }
       }
     end
 
     def assert_button_present(button_id)
       @web_browser.buttons.each { |button|
-        return if button_id == button.id
+				the_button_id = RWebSpec.framework == "Watir" ? button.id : button["id"]
+        return if button_id == the_button_id
       }
       fail("can't find the button with id: #{button_id}")
     end
 
     def assert_button_present_with_text(button_text)
-      @web_browser.buttons.each { |button|
-        return if button_text == button.value
+      @web_browser.buttons.each { |button|			
+        return if button_text == element_value(button)
       }
       fail("can't find the button with text: #{button_text}")
     end
@@ -269,7 +343,12 @@ module RWebSpec
     #  assert_exists("label", "receipt_date")
     #  assert_exists(:span, :receipt_date)
     def assert_exists(tag, element_id)
-      perform_assertion { assert(eval("#{tag}(:id, '#{element_id.to_s}').exists?"), "Element '#{tag}' with id: '#{element_id}' not found") }
+			if RWebSpec.framework == "Watir"
+      	perform_assertion { assert(eval("#{tag}(:id, '#{element_id.to_s}').exists?"), "Element '#{tag}' with id: '#{element_id}' not found") }		
+			else
+				perform_assertion { assert( @web_browser.driver.find_element(:tag_name => tag, :id => element_id))}
+			end
+		
     end
 
     alias assert_exists? assert_exists
@@ -350,7 +429,12 @@ module RWebSpec
     # assert_text_field_value("text1", "text already there") => true
     #
     def assert_text_field_value(textfield_name, text)
-      perform_assertion { assert_equal(text, text_field(:name, textfield_name).value) }
+			if RWebSpec.framework == "Watir"				
+      	perform_assertion { assert_equal(text, text_field(:name, textfield_name).value) }
+			else
+				the_element = @web_browser.driver.find_element(:name, textfield_name)
+      	perform_assertion { assert_equal(text, element_value(the_element)) }
+			end			
     end
 
 
@@ -373,9 +457,9 @@ module RWebSpec
 
     private
     def table_source(table_id, options)
-      elem_table = table(:id, table_id.to_s)
+      elem_table = RWebSpec.framework == "Watir" ? table(:id, table_id.to_s) : @web_browser.driver.find_element(:id => table_id)
       elem_table_text = elem_table.text
-      elem_table_html = is_firefox? ? elem_table.innerHTML : elem_table.html
+      elem_table_html = RWebSpec.framework == "Watir" ?  elem_table.html : elem_table["innerHTML"];
       table_source = options[:just_plain_text] ? elem_table_text : elem_table_html
     end
 

data/lib/rwebspec-common/test_utils.rb

@@ -40,8 +40,8 @@ module RWebSpec
     alias try_until try_for
 
     def try(timeout = $testwise_polling_timeout, polling_interval = $testwise_polling_interval || 1, &block)
-      puts "Warning: method 'try' is deprecated (won't support in RWebSpec 3), use try_until instead."
-      try_until(timeout, polling_interval) {
+      puts "Warning: method 'try' is deprecated (won't support in RWebSpec 3), use try_for instead."
+      try_for(timeout, polling_interval) {
         yield
       }
     end
@@ -92,42 +92,6 @@ module RWebSpec
     end
 
 
-
-    # use win32screenshot library to save curernt active window, which shall be IE
-    #
-    # opts[:to_dir] => the direcotry to save image under
-    def take_screenshot(opts = {})
-      # puts "calling new take screenshot: #{$screenshot_supported}"
-      unless $screenshot_supported
-        puts " [WARN] Screenhost not supported, check whether win32screenshot gem is installed" 
-        return
-      end
-  
-        begin
-          screenshot_image_filename =  "screenshot_" + Time.now.strftime("%m%d%H%M%S") + ".jpg"
-          the_dump_dir = opts[:to_dir] || default_dump_dir
-          FileUtils.mkdir_p(the_dump_dir) unless File.exists?(the_dump_dir)
-          screenshot_image_filepath = File.join(the_dump_dir, screenshot_image_filename)
-          screenshot_image_filepath.gsub!("/", "\\") if is_windows?
-
-          FileUtils.rm_f(screenshot_image_filepath) if File.exist?(screenshot_image_filepath)
-
-          if is_firefox? then
-            Win32::Screenshot::Take.of(:window, :title => /mozilla\sfirefox/i).write(screenshot_image_filepath)					
-		  elsif ie
-            Win32::Screenshot::Take.of(:window, :title => /internet\sexplorer/i).write(screenshot_image_filepath)					
-          else
-            Win32::Screenshot::Take.of(:foreground).write(screenshot_image_filepath)
-          end
-          notify_screenshot_location(screenshot_image_filepath)
-				rescue ::DL::DLTypeError => de
-					puts "No screenshot libray found: #{de}"
-        rescue => e
-          puts "error on taking screenshot: #{e}"
-        end
-      
-    end
-
     #= Convenient functions
     #
 

data/lib/rwebspec-watir/driver.rb

@@ -29,7 +29,7 @@ module RWebSpec
     #   1. pass as first argument
     #   2. If running using TestWise, used as confiured
     #   3. Use default value set
-    def open_browser(base_url = nil, options = {})
+    def open_browser(options = {})
 
       begin
         support_unicode
@@ -38,7 +38,12 @@ module RWebSpec
       end
 
       base_url ||= $TESTWISE_PROJECT_BASE_URL
-      base_url ||= $TESTWISE_PROJECT_BASE_URL
+			if options && options.class == String
+      	base_url ||= options
+			elsif options && options.class == Hash && options[:base_url]
+      	base_url ||= options[:base_url]
+			end
+			
       base_url ||= $BASE_URL
       raise "base_url must be set" if base_url.nil?
 
@@ -50,7 +55,7 @@ module RWebSpec
                          :go => true}
 
       options = default_options.merge options
-      ($TESTWISE_HIDE_BROWSER || $TESTWISE_HIDE_BROWSER) ? $HIDE_IE = true : $HIDE_IE = false
+      ($TESTWISE_HIDE_BROWSER) ? $HIDE_IE = true : $HIDE_IE = false
 
       if base_url =~ /^file:/
         uri_base = base_url
@@ -152,7 +157,7 @@ module RWebSpec
 
     # Go to another page on the testing site.
     #
-    #  open_browser("http://www.itest2.com")
+    #  open_browser(:base_url => "http://www.itest2.com")
     #  goto_page("/demo")  # visit page http://www.itest2.com/demo
     #
     def goto_page(page)
@@ -165,7 +170,7 @@ module RWebSpec
     
     # Go to another web site, normally different site being tested on
     #
-    #  open_browser("http://www.itest2.com")
+    #  open_browser(:base_url => "http://www.itest2.com")
     #  goto_url("http://myorganized.info")
     def goto_url(url)
       @web_browser.goto_url url
@@ -751,6 +756,58 @@ module RWebSpec
     def basic_authentication(username, password, options = {})
       basic_authentication_ie(options[:title], username, password, options)
     end
+    
+    # TODO: Common driver module =>  this is shared by both Watir and Selenium
+    #
+
+
+  # TODO: Common driver module =>  this is shared by both Watir and Selenium
+  #
+
+  # use win32screenshot library or Selenium to save curernt active window
+  #
+  # opts[:to_dir] => the direcotry to save image under
+  def take_screenshot(to_file = nil, opts = {})
+    # puts "calling new take screenshot: #{$screenshot_supported}"
+    # unless $screenshot_supported
+    #   puts " [WARN] Screenhost not supported, check whether win32screenshot gem is installed" 
+    #   return
+    # end
+
+    if to_file
+      screenshot_image_filepath = to_file
+    else
+      screenshot_image_filename =  "screenshot_" + Time.now.strftime("%m%d%H%M%S") + ".jpg"
+      the_dump_dir = opts[:to_dir] || default_dump_dir
+      FileUtils.mkdir_p(the_dump_dir) unless File.exists?(the_dump_dir)
+      screenshot_image_filepath = File.join(the_dump_dir, screenshot_image_filename)
+      screenshot_image_filepath.gsub!("/", "\\") if is_windows?
+
+      FileUtils.rm_f(screenshot_image_filepath) if File.exist?(screenshot_image_filepath)
+    end
+
+    if RWebSpec.framework == "Watir"        
+        begin       
+          if is_firefox? then
+            Win32::Screenshot::Take.of(:window, :title => /mozilla\sfirefox/i).write(screenshot_image_filepath)					
+		      elsif ie
+            Win32::Screenshot::Take.of(:window, :title => /internet\sexplorer/i).write(screenshot_image_filepath)					
+          else
+            Win32::Screenshot::Take.of(:foreground).write(screenshot_image_filepath)
+          end
+          notify_screenshot_location(screenshot_image_filepath)
+				rescue ::DL::DLTypeError => de
+					puts "No screenshot libray found: #{de}"
+        rescue => e
+          puts "error on taking screenshot: #{e}"
+        end
+    else
+      # save screenshot with selenium
+      @web_browser.driver.save_screenshot(screenshot_image_filepath)
+      notify_screenshot_location(screenshot_image_filepath)
+    end  
+
+  end
 
 	 # end of methods
 

data/lib/rwebspec-watir/web_browser.rb

@@ -295,7 +295,7 @@ module RWebSpec
 
     # Go to a page
     #  Usage:
-    #    open_browser("http://www.itest2.com"
+    #    open_browser(:base_url => "http://www.itest2.com")
     #    ....
     #    goto_page("/purchase")  # full url => http://www.itest.com/purchase
     def goto_page(page)

data/lib/rwebspec-webdriver/driver.rb

@@ -29,8 +29,9 @@ module RWebSpec
       #
       # New Options:
       #    :browser => :ie | :firefox | :chrome
-      def open_browser(base_url = nil, options = {})
+      def open_browser(options = {})
         # puts "[DEBUG] [SeleniumDriver] Callling open_browser #{base_url}"
+
         begin
           support_unicode
         rescue => e
@@ -38,6 +39,11 @@ module RWebSpec
         end
 
         base_url ||= $TESTWISE_PROJECT_BASE_URL
+				if options && options.class == String
+	      	base_url ||= options
+				elsif options && options.class == Hash && options[:base_url]
+	      	base_url ||= options[:base_url]
+				end
         base_url ||= $BASE_URL
         raise "base_url must be set" if base_url.nil?
 
@@ -80,7 +86,8 @@ module RWebSpec
         end
 
 				# remembering browser handle for debugging need
-				$browser = @web_browser
+				$browser = @web_browser				
+		
         return @web_browser
       end
 
@@ -179,7 +186,7 @@ module RWebSpec
 
       # Go to another page on the testing site.
       #
-      #  open_browser("http://www.itest2.com")
+      #  open_browser(:base_url => "http://www.itest2.com")
       #  goto_page("/demo")  # visit page http://www.itest2.com/demo
       #
       def goto_page(page)
@@ -191,7 +198,7 @@ module RWebSpec
 
       # Go to another web site, normally different site being tested on
       #
-      #  open_browser("http://www.itest2.com")
+      #  open_browser(:base_url => "http://www.itest2.com")
       #  goto_url("http://myorganized.info")
       def goto_url(url)
         puts "Calling web_browser goto: #{@web_browser.inspect}"
@@ -394,6 +401,8 @@ module RWebSpec
         else
           to_dir = ENV['TEMP_DIR'] || (is_windows? ? "C:\\temp" : "/tmp")
         end
+        puts "Save screenshot to default dir: #{to_dir}"
+        return to_dir
       end
 
       # For current page souce to a file in specified folder for inspection
@@ -741,5 +750,56 @@ module RWebSpec
         basic_authentication_ie(options[:title], username, password, options)
       end
   
+  
+  
+      # TODO: Common driver module =>  this is shared by both Watir and Selenium
+      #
+
+      # use win32screenshot library or Selenium to save curernt active window
+      #
+      # opts[:to_dir] => the direcotry to save image under
+      def take_screenshot(to_file = nil, opts = {})
+        # puts "calling new take screenshot: #{$screenshot_supported}"
+        # unless $screenshot_supported
+        #   puts " [WARN] Screenhost not supported, check whether win32screenshot gem is installed" 
+        #   return
+        # end
+
+        if to_file
+          screenshot_image_filepath = to_file
+        else
+          screenshot_image_filename =  "screenshot_" + Time.now.strftime("%m%d%H%M%S") + ".jpg"
+          the_dump_dir = opts[:to_dir] || default_dump_dir
+          FileUtils.mkdir_p(the_dump_dir) unless File.exists?(the_dump_dir)
+          screenshot_image_filepath = File.join(the_dump_dir, screenshot_image_filename)
+          screenshot_image_filepath.gsub!("/", "\\") if is_windows?
+
+          FileUtils.rm_f(screenshot_image_filepath) if File.exist?(screenshot_image_filepath)
+        end
+
+        if RWebSpec.framework == "Watir"        
+            begin       
+              if is_firefox? then
+                Win32::Screenshot::Take.of(:window, :title => /mozilla\sfirefox/i).write(screenshot_image_filepath)					
+    		      elsif ie
+                Win32::Screenshot::Take.of(:window, :title => /internet\sexplorer/i).write(screenshot_image_filepath)					
+              else
+                Win32::Screenshot::Take.of(:foreground).write(screenshot_image_filepath)
+              end
+              notify_screenshot_location(screenshot_image_filepath)
+    				rescue ::DL::DLTypeError => de
+    					puts "No screenshot libray found: #{de}"
+            rescue => e
+              puts "error on taking screenshot: #{e}"
+            end
+        else
+          # save screenshot with selenium
+          @web_browser.driver.save_screenshot(screenshot_image_filepath)
+          notify_screenshot_location(screenshot_image_filepath)
+        end  
+
+      end
+      
+      
   end
 end

data/lib/rwebspec-webdriver/web_browser.rb

@@ -37,8 +37,16 @@ module RWebSpec
           when "ie"
             initialize_ie_browser(existing_browser, options)
           when  "htmlunit"
-             initialize_htmlunit_browser(base_url, options)
+            initialize_htmlunit_browser(base_url, options)
         end
+
+				begin
+					if options[:resize_to] && options[:resize_to].class == Array
+						@browser.manage.window.resize_to(options[:resize_to][0], options[:resize_to][1])
+		    	end
+				rescue => e
+					puts "[ERROR] failed to resize => #{options[:resize_to]}"
+				end
       end
 
       def initialize_firefox_browser(existing_browser, base_url, options)
@@ -83,7 +91,7 @@ module RWebSpec
           else
             @browser.speed = :zippy
           end
-          return
+          return @browser
         end
 
         @browser = Selenium::WebDriver.for :ie
@@ -101,6 +109,7 @@ module RWebSpec
         #      else
         #        puts "close other browser instances not working yet in Ruby 1.9.1 version of Watir"
         #      end
+
       end
 
       # TODO resuse not working yet
@@ -258,12 +267,43 @@ module RWebSpec
 				end
 			end
 			
-      [:images, :links, :buttons, :select_lists, :checkboxes, :radios, :text_fields, :divs, :dls, :dds, :dts, :ems, :lis, :maps, :spans, :strongs, :ps, :pres, :labels].each do |method|
+			# :links => removed
+			# :checkboxes => removed
+			# :radios => removed
+			# :select_lists => removed
+			# :buttons => removed
+			# :divs => removed
+      [:images, :text_fields, :dls, :dds, :dts, :ems, :lis, :maps, :spans, :strongs, :ps, :pres, :labels].each do |method|
         define_method method do
           @browser.send(method)
         end
       end
 
+			def links
+				@browser.find_elements(:tag_name, "a")
+			end
+
+			def checkboxes
+				@browser.find_elements(:xpath, "//input[@type='checkbox']")
+			end
+			
+			def radios
+				@browser.find_elements(:xpath, "//input[@type='radio']")
+			end		
+			
+			def select_lists
+				@browser.find_elements(:tag_name, "select")
+			end			
+			
+			def buttons
+				button_array = @browser.find_elements(:tag_name, "button") + @browser.find_elements(:xpath, "//input[@type='submit']") + @browser.find_elements(:xpath, "//input[@type='button']")
+				return button_array
+			end			
+			
+			def divs
+				@browser.find_elements(:tag_name, "divs")
+			end
+			
       # current url
       def current_url
         @browser.current_url
@@ -373,7 +413,7 @@ module RWebSpec
 
       # Go to a page
       #  Usage:
-      #    open_browser("http://www.itest2.com"
+      #    open_browser(:base_url => "http://www.itest2.com")
       #    ....
       #    goto_page("/purchase")  # full url => http://www.itest.com/purchase
       def goto_page(page)
@@ -435,8 +475,17 @@ module RWebSpec
       # Click a button with give HTML id
       # Usage:
       #   click_button_with_id("btn_sumbit")
+      #   click_button_with_id("btn_sumbit", :index => 2) # the secone link with same id, not good gractice in HTML
       def click_button_with_id(id, opts = {})
-        find_element(:id, id).click
+	     if opts && opts[:index] && opts[:index].to_i() > 0
+					elements = find_elements(:id, id)
+					the_index = opts[:index].to_i() - 1          
+          first_match = elements[the_index]
+          first_match.click
+				else
+	        find_element(:id, id).click
+        end
+		
       end
 
       # Click a button with give name
@@ -459,14 +508,15 @@ module RWebSpec
         if matching_buttons.size > 0
 
           if opts && opts[:index]
-            puts "Call matching buttons: #{matching_buttons.inspect}"
-            first_match = matching_buttons[opts[:index].to_i() - 1]
+						the_index = opts[:index].to_i() - 1
+            puts "Call matching buttons: #{matching_buttons.inspect} => #{the_index}"
+            first_match = matching_buttons[the_index]
             first_match.click
+					else
+	          the_button = matching_buttons[0]
+	          the_button.click
           end
 
-          the_button = matching_buttons[0]
-          the_button.click
-
         else
           raise "No button with value: #{caption} found"
         end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rwebspec
 version: !ruby/object:Gem::Version
-  version: '4.0'
+  version: 4.1.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: rwebspec
 bindir: bin
 cert_chain: []
-date: 2012-11-23 00:00:00.000000000 Z
+date: 2013-01-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
@@ -32,7 +32,7 @@ dependencies:
   requirement: !ruby/object:Gem::Requirement
     none: false
     requirements:
-    - - ! '>='
+    - - '='
       - !ruby/object:Gem::Version
         version: 2.10.1
   type: :runtime
@@ -40,7 +40,7 @@ dependencies:
   version_requirements: !ruby/object:Gem::Requirement
     none: false
     requirements:
-    - - ! '>='
+    - - '='
       - !ruby/object:Gem::Version
         version: 2.10.1
 - !ruby/object:Gem::Dependency
@@ -59,6 +59,22 @@ dependencies:
     - - ! '>='
       - !ruby/object:Gem::Version
         version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: selenium-webdriver
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Executable functional specification for web applications in RSpec syntax
   with Watir or Selenium
 email: zhimin@agileway.com.au