用Watir测试QTP的Demo程序Mercury Tours
参考：
http://wiki.openqa.org/display/WTR/Watir+demo+on+Mercury+Tours
require 'test/unit'
require 'watir'
require 'watir/screen_capture'
class Demo < Test::Unit::TestCase
include Watir
include Watir::ScreenCapture
def setup
@ie = Watir::IE.new_process
@ie.goto("http://newtours.demoaut.com/")
end
def test_logon
@ie = Watir::IE.attach(:url,"http://newtours.demoaut.com/" )
@ie.maximize()
assert_equal(@ie.title, "Welcome: Mercury Tours")
@ie.text_field(:name, "userName").set "test"
@ie.text_field(:name, "password").set "test"
@ie.button(:name,"login").click
screen_capture("logon.jpg",true)
assert_equal(@ie.title, "Find a Flight: Mercury Tours:")
assert(@ie.radio(:name,"tripType", "oneway").exists?)
assert(@ie.select_list(:name,"passCount").exists?)
assert(@ie.select_list(:name,"fromPort").exists?)
assert(@ie.select_list(:name,"fromMonth").exists?)
assert(@ie.select_list(:name,"fromDay").exists?)
assert(@ie.select_list(:name,"toPort").exists?)
assert(@ie.radio(:name, "servClass","Business").exists?)
assert(@ie.select_list(:name,"airline").exists?)
assert(@ie.form(:name, "findflight").exists?)
# booking page 1
@ie.radio(:name,"tripType", "oneway").set
@ie.select_list(:name,"passCount").select "1"
@ie.select_list(:name,"fromPort").select "Acapulco"
@ie.select_list(:name,"fromMonth").select "March"
@ie.select_list(:name,"fromDay").select "10"
@ie.select_list(:name,"toPort").select "Frankfurt"
@ie.radio(:name, "servClass","Business").set
@ie.select_list(:name,"airline").select "Unified Airlines"
@ie.form(:name, "findflight").submit
screen_capture("page1.jpg",true)
# booking page 2
assert(@ie.radio(:name, "outFlight","Blue Skies Airlines$360$270$5:03").exists?)
assert(@ie.radio(:name, "outFlight","Blue Skies Airlines$361$271$7:10").exists?)
assert(@ie.radio(:name, "outFlight","Pangea Airlines$362$274$9:17").exists?)
assert(@ie.radio(:name, "outFlight","Unified Airlines$363$281$11:24").exists?)
@ie.radio(:name, "outFlight","Pangea Airlines$362$274$9:17").set
assert(@ie.radio(:name, "inFlight","Blue Skies Airlines$631$273$14:30").exists?)
@ie.radio(:name, "inFlight","Blue Skies Airlines$631$273$14:30").set
assert(@ie.form(:name,"results").exists?)
@ie.form(:name,"results").submit
screen_capture("page2.jpg",true)
#booking page 3 -- purchase
assert(@ie.text_field(:name,"passFirst0").exists?)
assert(@ie.text_field(:name,"passLast0").exists?)
assert(@ie.text_field(:name,"creditnumber").exists?)
assert(@ie.form(:name,"bookflight").exists?)
@ie.text_field(:name,"passFirst0").set "samuel"
@ie.text_field(:name,"passLast0").set "luo"
@ie.text_field(:name,"creditnumber").set "1234"
@ie.form(:name,"bookflight").submit
screen_capture("page3.jpg",true)
end
def teardown
@ie.close()
end
end
这个例子测试QTP的WEB样例程序，使用了screen_capture包来实现屏幕截取，打开这个包（C:/ruby/lib/ruby/gems/1.8/gems/watir-1.5.2/watir/screen_capture.rb）可以看到，所谓的屏幕截取就是使用Win32API进行硬拷贝，然后打开画图程序保存成文件：
require 'Win32API'
module Watir
module ScreenCapture
KEYEVENTF_KEYUP = 0x2
SW_HIDE = 0
SW_SHOW = 5
SW_SHOWNORMAL = 1
VK_CONTROL = 0x11
VK_F4 = 0x73
VK_MENU = 0x12
VK_RETURN = 0x0D
VK_SHIFT = 0x10
VK_SNAPSHOT = 0x2C
VK_TAB = 0x09
GMEM_MOVEABLE = 0x0002
CF_TEXT = 1
# this method saves the current window or whole screen as either a bitmap or a jpeg
# It uses paint to save the file, so will barf if a duplicate filename is selected, or the path doesnt exist etc
# * filename - string - the name of the file to save. If its not fully qualified the current directory is used
# * active_window - boolean - if true, the whole screen is captured, if false, just the active window is captured
# * save_as_bmp - boolean - if true saves the file as a bitmap, saves it as a jpeg otherwise
def screen_capture(filename , active_window_only=false, save_as_bmp=false)
keybd_event = Win32API.new("user32", "keybd_event", ['I','I','L','L'], 'V')
vkKeyScan = Win32API.new("user32", "VkKeyScan", ['I'], 'I')
winExec = Win32API.new("kernel32", "WinExec", ['P','L'], 'L')
openClipboard = Win32API.new('user32', 'OpenClipboard', ['L'], 'I')
setClipboardData = Win32API.new('user32', 'SetClipboardData', ['I', 'I'], 'I')
closeClipboard = Win32API.new('user32', 'CloseClipboard', [], 'I')
globalAlloc = Win32API.new('kernel32', 'GlobalAlloc', ['I', 'I'], 'I')
globalLock = Win32API.new('kernel32', 'GlobalLock', ['I'], 'I')
globalUnlock = Win32API.new('kernel32', 'GlobalUnlock', ['I'], 'I')
memcpy = Win32API.new('msvcrt', 'memcpy', ['I', 'P', 'I'], 'I')
filename = Dir.getwd.tr('/','//') + '//' + filename unless filename.index('//')
if active_window_only ==false
keybd_event.Call(VK_SNAPSHOT,0,0,0) # Print Screen
else
keybd_event.Call(VK_SNAPSHOT,1,0,0) # Alt+Print Screen
end
winExec.Call('mspaint.exe', SW_SHOW)
sleep(1)
# Ctrl + V : Paste
keybd_event.Call(VK_CONTROL, 1, 0, 0)
keybd_event.Call(vkKeyScan.Call(?V), 1, 0, 0)
keybd_event.Call(vkKeyScan.Call(?V), 1, KEYEVENTF_KEYUP, 0)
keybd_event.Call(VK_CONTROL, 1, KEYEVENTF_KEYUP, 0)
# Alt F + A : Save As
keybd_event.Call(VK_MENU, 1, 0, 0)
keybd_event.Call(vkKeyScan.Call(?F), 1, 0, 0)
keybd_event.Call(vkKeyScan.Call(?F), 1, KEYEVENTF_KEYUP, 0)
keybd_event.Call(VK_MENU, 1, KEYEVENTF_KEYUP, 0)
keybd_event.Call(vkKeyScan.Call(?A), 1, 0, 0)
keybd_event.Call(vkKeyScan.Call(?A), 1, KEYEVENTF_KEYUP, 0)
sleep(1)
# copy filename to clipboard
hmem = globalAlloc.Call(GMEM_MOVEABLE, filename.length+1)
mem = globalLock.Call(hmem)
memcpy.Call(mem, filename, filename.length+1)
globalUnlock.Call(hmem)
openClipboard.Call(0)
setClipboardData.Call(CF_TEXT, hmem)
closeClipboard.Call
sleep(1)
# Ctrl + V : Paste
keybd_event.Call(VK_CONTROL, 1, 0, 0)
keybd_event.Call(vkKeyScan.Call(?V), 1, 0, 0)
keybd_event.Call(vkKeyScan.Call(?V), 1, KEYEVENTF_KEYUP, 0)
keybd_event.Call(VK_CONTROL, 1, KEYEVENTF_KEYUP, 0)
if save_as_bmp == false
# goto the combo box
keybd_event.Call(VK_TAB, 1, 0, 0)
keybd_event.Call(VK_TAB, 1, KEYEVENTF_KEYUP, 0)
sleep(0.5)
# select the first entry with J
keybd_event.Call(vkKeyScan.Call(?J), 1, 0, 0)
keybd_event.Call(vkKeyScan.Call(?J), 1, KEYEVENTF_KEYUP, 0)
sleep(0.5)
end
# Enter key
keybd_event.Call(VK_RETURN, 1, 0, 0)
keybd_event.Call(VK_RETURN, 1, KEYEVENTF_KEYUP, 0)
sleep(1)
# Alt + F4 : Exit
keybd_event.Call(VK_MENU, 1, 0, 0)
keybd_event.Call(VK_F4, 1, 0, 0)
keybd_event.Call(VK_F4, 1, KEYEVENTF_KEYUP, 0)
keybd_event.Call(VK_MENU, 1, KEYEVENTF_KEYUP, 0)
sleep(1)
end
end
end
#screenCapture( "f.bmp", false , true)