在使用selenium进行UI自动化时,经常会遇到文件上传的坑。所以本文总结一下selenium常用的文件上传方法。
1、使用input标签上传文件
在元素定位时,一旦定位到页面中使用的是input标签,即可使用sendKeys上传文件。如下:
public void uploadAttachment(String sPicPath) {
WebElement uploadFile = driver.findElement(By.xpath("//input[@type = 'file']"));
log.info("上传图片" + sPicPath);
uploadFile.sendKeys(sPicPath);
log.info("上传图片中");
this.sleep();
log.info("上传完成");
}
注意,使用时,无需在页面点击“上传”按钮。可直接调用上述方法。
2、非input标签上传文件
元素定位时,发现上传时有的使用的是<a>标签,有的用<div>的,有的用<button>,有的用<object>.......如下图
针对上述场景,不能直接通过在网页上处理上传,唯一的办法就是打开OS弹框,去处理弹框。常使用的方法是,借助autoIT,调用其生成的au3或exe文件。
下面就具体介绍一下使用autoit自动上传文件的步骤。
autoit即使用类似BASIC脚本语言的免费软件,它设计用于Windows GUI(图形用户界面)中进行自动化操作。它利用模拟键盘按键,鼠标移动和窗口/控件的组合来实现自动化任务。
1)、在官网下载Autoit(https://www.autoitscript.com/site/)默认安装即可。
2)、安装完成后,到安装目录下,启动autoit程序,即:Au3Info_x64.exe或Au3Info.exe(选择适合自己的系统版本)
启动后界面:
3)、打开要上传文件的应用,这里有一个测试用的网址,可以在这里练习。[http://fex-team.github.io/webuploader/demo.html]
因为autoit只负责上传,因此需要已经弹出windows弹窗。
弹窗需要注意的地方就是弹窗名称“文件上传”、“文件名”和“打开”按钮
4)、切换到已打开的AutoIt Window Info 工具,找到Finder Tool:然后用鼠标拖动此图标到文件上传弹窗的 “文件名输入框”:
记下ClassnameNN的Value值,这里是“Edit1”。
然后用鼠标拖动此图标到文件上传弹窗的 “打开” 按钮上:
记下ClassnameNN的Value值,这里是“Button1”。
5)、打开安装目录下的SciTE Script Editor应用程序。
将下列代码复制到该编辑器中:
;first make sure the number of arguments passed into the scripts is more than 1
If $CmdLine[0]<2 Then Exit EndIf ;if parmas num <2 ,then break
;$CmdLine[0] ;参数的数量
;$CmdLine[1] ;第一个参数 (脚本名称后面)
;$CmdLine[2] ;第二个参数
;都是从cmd传入参数
handleUpload($CmdLine[1],$CmdLine[2])
;定义上传函数,有两个参数,第一个是浏览器名字,第二参数是文件路径
Func handleUpload($browser, $uploadfile)
Dim $title ;定义一个title变量
;根据弹窗的title来判断是什么浏览器
If $browser='ie' Then ; 代表IE浏览器
$title='选择要加载的文件'
ElseIf $browser='chrome' Then ; 代表谷歌浏览器
$title='打开'
ElseIf $browser='firefox' Then ; 代表火狐浏览器
$title='文件上传'
EndIf
if WinWait($title,'',4) Then ;等待弹出出现,最大等待时间是4秒
WinActivate($title) ;找到弹出窗口之后,激活当前窗口
ControlSetText($title,'','Edit1',$uploadfile) ;把文件路径放入输入框,此“Edit1”是用FinderTool获取到的“文件名”输入框的ClassnameNN
ControlClick($title,'','Button1') ;点击保存或者打开或者上传按钮,此“Button1”使用FinderTool获取到的“打开”按钮的ClassnameNN
Else
Return False
EndIf
EndFunc
6)、保存该文件到自定义目录下,默认后缀名为.au3。
autoit是根据弹窗的名字来给文件名输入框赋值和点击按钮等操作的,由于不同的浏览器弹窗的名字可能不一样,这里要谨慎选择。IE为'选择要加载的文件',Chrome为'打开',火狐为“文件上传”。
7)、autoit的脚本已编写完毕,但由于是在cmd中调用,需要将其转换成exe可执行文件。
打开autoit安装目录下的应用程序:Aut2exe_x64.exe或者Aut2exe.exe,根据自身系统选择。
点击Conver按钮,则成功讲upload.au3脚本转换为运行文件upload.exe,如下图。
8)、测试一下。打开cmd,输入:"F:\py_learn\selenium\upload.exe" "firefox" "F:\aa.png"双引号不要省略,每项之间有空格,如图:
之后回到网页,查看刚才的文件已被上传:
注意,运行cmd命令之前,一定要把页面的弹窗弹出,autoit只操作弹窗,不对页面元素定位。
第一个参数的是autoit exe脚本文件,第二是 autoit脚本函数需要的参数,回头看看脚本,一个参数是浏览器名字:chrome或者ie或者firefox,第二个参数是要上传文件的路径,并且都要用英文环境下的双引号括起来,就形成了下面的代码:
"F:\py_learn\selenium\upload.exe" "firefox" "F:\aa.png"
9)、OK,测试成功,下面需要在具体的selenium脚本中调用。
/**
* 非input输入框上传,需要传入上传脚本的执行文件路径、浏览器、图片地址
* 框架中已经存有2张待用图片,传入响应参数即可
* TEST_PICTURE_PATH_1
* TEST_PICTURE_PATH_2
*
*/
public void uploadFile(String picPath){
Runtime r = Runtime.getRuntime();
String browser=null;
if(TEST_BROWSER.equalsIgnoreCase("chrome")){
browser="chrome";
}else if(TEST_BROWSER.equalsIgnoreCase("ie")){
browser="ie";
}
try {
r.exec(UPLOAD_SCRIPT_PATH+" "+browser+" "+picPath);
log.info("upload file:" + picPath);
} catch (IOException e) {
e.printStackTrace();
log.error("Run upload.exe fail");
}
//等待上传完成
this.sleep();
}
在具体需要使用文件上传的类中直接调用上述方法,传入图片地址即可。