Selenium系列教程 - 文件上传-Toy模板网

这篇具有很好参考价值的文章主要介绍了Selenium系列教程 - 文件上传。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

在Web UI自动化测试中可能会遇到文件上传的场景，针对该场景我们要区分上传按钮的种类，大体上可以分为两种，一种是input框 <input type="file">，另外一种就比较复杂，通过js等实现。

一、通过send_keys方法（该方法只适用于input标签）

查看上传按钮的元素标签为input标签，调用send_keys方法传入文件路径即可，无需模拟弹出windows选择框且选择文件的过程。通过该方法可以简单、高效的实现选择文件的目的，避免将处理思路卡在如何选择文件这个问题上。
Selenium系列教程 - 文件上传
代码：

from selenium import webdriver
import time

driver = webdriver.Chrome()
driver.get('http://sahitest.com/demo/php/fileUpload.htm')
driver.implicitly_wait(10)

driver.find_element_by_css_selector('form>#file').send_keys(r'C:\Users\Administrator\Desktop\1.txt')
driver.find_element_by_css_selector('form>input[type="submit"]').click()

time.sleep(3)
driver.quit()

二、通过AutoIt来处理上传文件（适用所有）

对于那些不是input框实现的上传怎么办，这种上传千奇百怪，有用a标签的，有用div的，有用button的，有用object的，我们没有办法通过直接在网页上处理掉这些上传，唯一的办法就是打开windows弹框，去处理弹框。

问题又来了，windows弹框涉及的层面已经不是selenium能解决的了，怎么办？很简单，使用AutoIt去处理呗。

关于AutoIt的使用方法我在另一篇博文中讲，请参见Selenium系列教程 - AutoIt命令行参数。这里不再赘述。

由于不同的浏览器，获取到的弹窗标题不一样，所以要区分开，通过传参的方式判定，废话少说，直接上AutoIt的代码：

;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获取到的
	  Sleep(1000)
	  ControlClick($title,"","Button1")               ;点击保存或者打开或者上传按钮，此“Button1”使用FinderTool获取到的
   Else
	  Return False
   EndIf
EndFunc

将上面的代码编译为exe文件，以供代码调用。
通过python代码调用编译的exe文件处理上传弹窗，使用什么浏览器就传什么参数。

import os
import time
from selenium import webdriver

driver = webdriver.Chrome()
driver.get('http://sahitest.com/demo/php/fileUpload.htm')
# 点击“选择文件”按钮
driver.find_element_by_css_selector('form>#file').click()
# 调用exe上传文件
command = "C:\\upload.exe" + " " + "chrome" + " " + " C:\\Users\\Administrator\\Desktop\\1.txt"
os.system(command)
time.sleep(3)
# 点击“Submit Single”按钮
driver.find_element_by_css_selector('form>input[type="submit"]').click()

time.sleep(3)
driver.quit()