用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)