watir 1.4.1 → 1.5.2

data/watir/WindowHelper.rb

@@ -1,26 +1,28 @@
-require 'win32ole'
+require 'watir/win32ole'
+require 'watir' # for Watir.autoit
 
 class WindowHelper
-    def initialize( )
-        @autoit = WIN32OLE.new('AutoItX3.Control')
+    @@ie_window_name = "Windows Internet Explorer"
+    def initialize()
+        @autoit = Watir.autoit
     end
     
-    def push_alert_button()
-        @autoit.WinWait "Microsoft Internet Explorer", ""
+    def push_alert_button
+        @autoit.WinWait @@ie_window_name, ""
         @autoit.Send "{ENTER}"
     end
     
-    def push_confirm_button_ok()
-        @autoit.WinWait "Microsoft Internet Explorer", ""
+    def push_confirm_button_ok
+        @autoit.WinWait @@ie_window_name, ""
         @autoit.Send "{ENTER}"
     end
     
-    def push_confirm_button_cancel()
-        @autoit.WinWait "Microsoft Internet Explorer", ""
+    def push_confirm_button_cancel
+        @autoit.WinWait @@ie_window_name, ""
         @autoit.Send "{ESCAPE}"
     end
     
-    def push_security_alert_yes()
+    def push_security_alert_yes
         @autoit.WinWait "Security Alert", ""
         @autoit.Send "{TAB}"
         @autoit.Send "{TAB}"
@@ -37,7 +39,7 @@ class WindowHelper
     
     def WindowHelper.check_autoit_installed
         begin
-            WIN32OLE.new('AutoItX3.Control')
+          Watir.autoit
         rescue
             raise Watir::Exception::WatirException, "The AutoIt dll must be correctly registered for this feature to work properly"
         end

data/watir/assertions.rb

@@ -0,0 +1,36 @@
+require 'test/unit/assertions'
+
+module Watir
+  # Verification methods
+  module Assertions
+  include Test::Unit::Assertions
+    
+    # Log a failure if the boolean is true. The message is the failure
+    # message logged.
+    # Whether true or false, the assertion count is incremented.
+    def verify boolean, message = 'verify failed.'
+      add_assertion
+      add_failure message, caller unless boolean
+    end
+    
+    def verify_equal expected, actual, message=nil
+      full_message = build_message(message, <<EOT, expected, actual)
+<?> expected but was
+<?>.
+EOT
+      verify(expected == actual, full_message)
+    end
+    def verify_match pattern, string, message=nil
+      pattern = case(pattern)
+      when String
+        Regexp.new(Regexp.escape(pattern))
+      else
+        pattern
+      end
+      full_message = build_message(message, "<?> expected to be =~\n<?>.", string, pattern)
+      verify(string =~ pattern, full_message)
+    end
+    
+  end
+
+end

data/watir/camel_case.rb

@@ -1,7 +1,7 @@
 # These are provided for backwards compatability with Watir 1.1
 
 module Watir
-    module SupportsSubElements
+    module Container
         alias waitForIE wait
         alias fileField file_field
         alias textField text_field
@@ -21,7 +21,16 @@ module Watir
         alias showLinks show_links
         alias showActive show_active
         alias showAllObjects show_all_objects
+        def   getIE; @ie; end
+        
     end
+    class Element
+        alias getOLEObject ole_object 
+        alias fireEvent fire_event        
+        alias innerText text
+        alias afterText after_text
+        alias beforeText before_text
+    end    
     class Frame        
         alias getDocument document
         alias waitForIE wait
@@ -29,9 +38,9 @@ module Watir
     class Form        
         alias waitForIE wait 
     end
-    
-    class SpanDivCommon
-        alias innerText text
+    class TextField
+        alias readOnly? :readonly?
+        alias getContents value
+        alias maxLength maxlength
     end    
-    
 end

data/watir/close_all.rb

@@ -0,0 +1,38 @@
+require 'watir'
+
+module Watir
+  class IE
+    # close all ie browser windows
+    def self.close_all
+      close_all_but nil
+    end
+    # find other ie browser windows and close them
+    def close_others
+      IE.close_all_but self
+    end
+    private
+    def self.close_all_but(except=nil)
+      Watir::IE.each do |ie|
+        ie.close_modal
+        ie.close unless except and except.hwnd == window.hwnd
+      end
+      sleep 1.0 # replace with polling for window count to be zero?
+    end
+    public
+    # close modal dialog. unlike IE#modal_dialog.close, does not wait for dialog
+    # to appear and does not raise exception if no window is found.
+    # returns true if modal was found and close, otherwise false
+    def close_modal
+      begin
+        original_attach_timeout = IE.attach_timeout
+        IE.attach_timeout = 0
+        self.modal_dialog.close
+        true
+      rescue NoMatchingWindowFoundException, TimeOutException
+        false
+      ensure
+        IE.attach_timeout = original_attach_timeout
+      end
+    end
+  end
+end

data/watir/contrib/enabled_popup.rb

@@ -0,0 +1,21 @@
+# http://www.vbcity.com/forums/topic.asp?tid=108859
+
+module Watir
+  module PageContainer
+    include Win32
+    def enabled_popup(timeout=4)
+      # Use handle of our parent window to see if we have any currently
+      # enabled popup.
+      hwnd_modal = 0
+      Waiter.wait_until(timeout) do
+        hwnd_modal, arr = GetWindow.call(hwnd, GW_ENABLEDPOPUP)
+        hwnd_modal > 0
+      end
+      # use hwnd() method to find the IE or Container hwnd (overriden by IE)
+      if hwnd_modal == hwnd() || 0 == hwnd_modal
+        hwnd_modal = nil
+      end
+      hwnd_modal
+    end
+  end
+end

data/watir/contrib/ie-new-process.rb

@@ -0,0 +1,27 @@
+# based on http://svn.instiki.org/instiki/trunk/test/watir/e2e.rb
+# and http://rubyforge.org/pipermail/wtr-general/2005-November/004108.html
+
+require 'watir/ie-process'
+
+class IEProcess < Watir::IE::Process 
+  def stop
+    right_to_terminate_process = 1
+    handle = Win32API.new('kernel32.dll', 'OpenProcess', 'lil', 'l').
+    call(right_to_terminate_process, 0, @process_id)
+    Win32API.new('kernel32.dll', 'TerminateProcess', 'll', 'l').call(handle, 0)
+  end
+  
+end
+
+module Watir
+  class IE
+    def process_id
+      @process_id ||= IEProcess.process_id_from_hwnd @ie.hwnd
+    end
+    attr_writer :process_id
+    def kill
+      iep = IEProcess.new process_id
+      iep.stop
+    end
+  end
+end

data/watir/contrib/page_checker.rb

@@ -0,0 +1,29 @@
+#  This module includes checkers which are run on every page load
+# 
+#  In order to use this module, add a require to one of your test scripts: 
+#  require 'watir/contrib/page_checker' 
+#  To add checkers, call the ie.add_checker method
+#
+#  ie.add_checker(PageCheckers::NAVIGATION_CHECKER)
+#
+#  Checkers are Ruby proc objects which are called within Watir::IE and passed 
+#  the current instance of ie. 
+
+module PageCheckers        
+
+	# This checker iterates through  the current document including any frames
+	# and checks for http errors, 404, 500 etc
+	NAVIGATION_CHECKER = lambda do |ie|
+		if ie.document.frames.length > 1
+			1.upto ie.document.frames.length do |i|
+				begin
+					ie.frame(:index, i).check_for_http_error
+				rescue Watir::Exception::UnknownFrameException
+					# frame can be already destroyed
+				end          
+			end
+		else
+			ie.check_for_http_error
+		end
+	end
+end

data/watir/contrib/visible.rb

@@ -0,0 +1,47 @@
+# visible.rb
+# Original Author: David Schmidt
+
+# See http://wiki.openqa.org/pages/viewpage.action?pageId=1119
+# for description and sample code.
+
+# Here is an example of adding a new #visible? method to the Watir::Element 
+# class which returns true or false to indicate whether the cell is visible. 
+# The question of visibility isn't as easy as it seems, because there are 
+# multiple styles to indicate visibility ("display: none" or 
+# "visibility: hidden"). In addition, an element is hidden if any PARENT 
+# element is hidden, even if that element would otherwise be visible.
+
+# This method first checks the current element and then each parent element 
+# for visibility until it reaches the top of the DOM element tree. Note that 
+# this method is quite slow compared to other methods because it must recurse 
+# all the way to the top of the DOM to guarantee that an element is truly 
+# visible.
+
+class Watir::Element
+
+  # If any parent element isn't visible then we cannot write to the
+  # element. The only realiable way to determine this is to iterate
+  # up the DOM element tree checking every element to make sure it's
+  # visible.
+  def visible?
+    # Now iterate up the DOM element tree and return false if any
+    # parent element isn't visible or is disabled.
+    object = document
+    while object
+      begin
+        if object.style.invoke('visibility') =~ /^hidden$/i
+          return false
+        end
+        if object.style.invoke('display') =~ /^none$/i
+          return false
+        end
+        if object.invoke('isDisabled')
+          return false
+        end
+      rescue WIN32OLERuntimeError
+      end
+      object = object.parentElement
+    end
+    true
+  end
+end

data/watir/cookiemanager.rb

@@ -1,3 +1,6 @@
+    require 'find'
+module Watir
+  module CookieManager 
 
     class Dir
         def Dir.visit(dir = '.', files_first = false, &block)
@@ -27,9 +30,6 @@
         end
     end
 
-module Watir
-  module CookieManager 
-    
     class WatirHelper
         #taken from shlObj.h  used in win32 SHGetSpecialFolderLocation
         #define CSIDL_INTERNET_CACHE            0x0020
@@ -39,7 +39,7 @@ module Watir
         INTERNET_CACHE = 0x0020
         
         def  WatirHelper.getSpecialFolderLocation(specFolderName)
-            shell = WIN32OLE.new("Shell.Application")
+            shell = WIN32OLE.new('Shell.Application')
             folder = shell.Namespace(specFolderName)
             folderItem = folder.Self
             folderPath = folderItem.Path

data/watir/datahandler.rb

@@ -0,0 +1,107 @@
+require 'watir/win32ole'
+
+class String
+
+	def pred
+		self[0..-2] << (self[-1]-1).chr
+	end
+
+end
+
+class DataHandler
+
+	# This allows you to put data in a csv file or an xls file and access it in your scripts for
+	# a data driven approach.
+	# 
+	# You can supply a file with extension csv or xls and it will just work
+	#
+	# e.g.
+	# require 'datahandler'
+	# d = DataHandler.new("data.csv")
+	# loginId = d.data[0].LoginId		--> line 1 of the csv file and column LoginId
+	# pin = d.data[0].Pin				--> line 1 of the cvs file and column Pin
+	#
+	# b = Datahandler.new("data.xls")
+	# loginId = b.data[0].LoginId                   --> line 1 of workbook 1 of the xls file with column LoginId
+	#
+	# Remember that line 1 of the data is referenced by using 0 - and this excludes the header line.
+	#
+	def initialize(datafile)
+		
+		# if the data files are in the same dir as the one you execute from this works
+		directory = `dir`
+		directory.match(/Directory of(.+)$/)
+		working_directory = $1.to_s.gsub(/\\/,"/").gsub(/^\s/,"")
+		file = "#{working_directory.chomp}\\#{datafile}"
+		
+		# otherwise it will take the location you specify
+		if File.exists?(file)
+			@datafile = file
+		else
+			@datafile = datafile
+		end
+		
+		@excel = WIN32OLE::new('excel.Application')
+		@workbook = @excel.Workbooks.Open(@datafile)
+		@worksheet = @workbook.Worksheets(1)
+		@excel['Visible'] = false
+		
+	end
+	
+	def csv_data
+		csv_data = File.readlines(@datafile)
+		header = csv_data[0].split(",")
+		header.each{|z| z.chomp!}
+		data = Struct.new("Data", *header)
+		csv_data.shift
+		data_array = []
+		csv_data.each do |line|
+			line.each{|z| line.chomp!}
+			data_array << data.new(*line.split(","))
+		end
+	return data_array
+	end
+	
+	def number_columns
+		column = last_column = "a"
+		while @worksheet.Range("a1:#{column}1")['Value']
+			if !@worksheet.Range("#{column}1")['Value']
+				last_column = column.pred
+				break
+			end
+			column.succ!
+		end
+	last_column
+	end
+		
+	def xls_data
+		line = "1"
+		data = []
+			while @worksheet.Range("a#{line}")['Value']
+			   data << @worksheet.Range("a#{line}:#{number_columns}#{line}")['Value']
+			   line.succ!
+			end
+		
+		header = data[0].flatten
+		dataObject = Struct.new("DataObject", *header)
+		data.shift
+		data_array = []
+		data.each do |line|
+			data_array << dataObject.new(*line.flatten)
+		end
+	@excel.Quit
+	@excel = nil
+	GC.start
+	data_array
+	end
+	
+	def data
+		extension = @datafile.split(".")[-1]
+		if extension == "xls"
+			xls_data
+		elsif extension == "csv"
+			csv_data
+		end
+	end
+	
+end

data/watir/dialog.rb

@@ -0,0 +1,46 @@
+require 'watir'
+# TODO: move this file to watir/contrib
+
+module Watir
+  
+  class Dialog
+    WindowName = 'Windows Internet Explorer'    
+    def button(name)
+      DialogButton.new(name)
+    end
+    def close
+      # TODO: register autoit before use
+      autoit = WIN32OLE.new('AutoItX3.Control')
+      autoit.WinClose WindowName, ""
+    end
+    def exists?
+      # TODO: register autoit before use
+      autoit = WIN32OLE.new('AutoItX3.Control')
+      found = autoit.WinWait(WindowName, "", 1)
+      return found == 1
+    end 
+  end
+  
+  def dialog
+    Dialog.new
+  end
+  
+  class DialogButton
+    def initialize(name)
+      @name = name
+    end
+    def click
+      # TODO: register autoit before use
+      autoit = WIN32OLE.new('AutoItX3.Control')
+      autoit.WinWait Dialog::WindowName, "", 1
+      name_pattern = Regexp.new "^#{@name}$"
+      unless name_pattern =~ autoit.WinGetText(Dialog::WindowName, "")
+        raise Watir::Exception::UnknownObjectException
+      end
+      autoit.Send "{ENTER}"
+    end
+  end
+  
+end
+
+