文件上傳自動化測試方案

1、概述

【測試地址】：https://pan.baidu.com

【測試工具】selenium、requests

【腳本語言】Python

【運行環境】Windows

百度網盤做爲文件存儲及分享的平臺，核心功能大部分是對文件的操做，若是要對它進行自動化測試，優先覆蓋的就是上傳、分享、複製功能

測試方案的大體結構以下：

　　　　　　　　　　　　　　圖 1-1 系統流程圖

2、上傳測試用例

以文件上傳功能爲例，測試用例描述以下：

　　預置條件： 生成指定數量和大小的測試文件（txt、csv等）

　　操做步驟：一、用戶登陸

　　　　　　　二、遍歷生成的文件所在的文件夾，獲得全部的文件路徑

　　　　　　　三、經過給FileUpload對象循環發送文件路徑完成上傳操做

　　預期結果：全部文件上傳成功，查詢文件信息顯示正確

3、每一個步驟的具體實現

一、生成指定數量和大小的測試文件：

　　以txt爲例，其餘文件（csv、word、pdf、jpg等）生成方式相似，能夠網上找相關的方法和第三方庫

 

 1 # -*- recoding:utf-8 -*-
 2 import os
 3 import time
 4 
 5 PATH = "D:/test_data/"
 6 
 7 
 8 def create_txt(path=PATH, file_num=1, file_size=8):
 9     """
10     :param path: 測試文件的存放路徑
11     :param file_num: 須要生成的數量
12     :param file_size: 每一個文件的大小，以KB爲單位
13     """
14     if not os.path.exists(path):
15         os.mkdir(path)
16     for i in range(file_num):
17         file_name = "test_txt_"+str(i)+".txt"
18         data = "test_t"+str(time.localtime())
19         with open(os.path.join(path, file_name), 'w') as f_obj:
20             for j in range(8*file_size):
21                 f_obj.write(data)
22         print "create file: test_txt_(0-%d).txt in path %s success, file size: %sKB" % (file_num, path, file_size)
23 
24 
25 if __name__ == "__main__":
26     # 生成包含不一樣文件大小的文件夾
27     for i in range(10):
28         path = "D:/test_data/" + str(i)
29         create_txt(path=path, file_num=i, file_size=5*1024*i)

 

二、使用selenium經過web頁面上傳到服務器

　　首先是登陸腳本，對webdriver看不懂的同窗先看個人另一篇帖子：http://www.cnblogs.com/fightever/p/6867619.html

 1 # -*- coding:utf-8 -*-
 2 from WebDriver.webdriver import Driver
 3 from selenium.webdriver.common.by import By
 4 
 5 
 6 class Login():
 7 
 8     def __init__(self, user_name="admin", pwd="admin@123"):
 9         self.page = Driver()
10         page = self.page
11         page.start()
12         self.account_login_btn = page.find_element(By.LINK_TEXT, u"賬號密碼登陸")
13         self.account_text = page.find_element(By.ID, "TANGRAM__PSP_4__userName")
14         self.pwd_text = page.find_element(By.ID, "TANGRAM__PSP_4__password")
15         self.login_btn = page.find_element(By.ID, "TANGRAM__PSP_4__submit")
16         self.login(user_name, pwd)
17 
18     def login(self, user_name, pwd):
19         self.account_login_btn.click()
20         self.account_text.clear()
21         self.account_text.send_keys(user_name)
22         self.pwd_text.clear()
23         self.pwd_text.send_keys(pwd)
24         self.login_btn.click()
25         print "login success"
26 
27 
28 if __name__ == "__main__":
29     # 將XXX換成本身註冊的百度網盤帳號
30     Login("XXX", "XXX")

上傳文件腳本：

 1 # -*- coding:utf-8 -*-
 2 from WebDriver.webdriver import Driver
 3 from selenium.webdriver.common.by import By
 4 from File.create import create_txt
 5 from webpage.LoginPage.login import Login
 6 import os
 7 
 8 OVER_TIME = 10
 9 
10 
11 class Upload(object):
12 
13     def __init__(self, path):
14         self.page = Driver()
15         self.upload_file(path)
16 
17     def send_path(self, path):
18         for i in range(OVER_TIME):
19             try:
20                 self.page.find_element(By.NAME, "html5uploader").send_keys(path)
21                 return True
22             except Exception, e:
23                 print e
24 
25     def upload_file(self, path, file_num=3):
27         for root, dirs, files in os.walk(path):
28             for f_obj in files:
29                 self.send_path(os.path.join(path, f_obj))
30 
31 
32 if __name__ == "__main__":
33     Login("XXX", "XXX")
34     Upload("D:\\test_data")

　　大部分網站的上傳控件使用的FileUpload對象，只要在頁面找到上傳控件的id或name等屬性，經過給使用send_keys給控件發送文件在本地的路徑，就能自動上傳

百度的上傳控件（這裏使用的是name屬性「html5uploader」）：

 

 三、上傳結果檢查，有兩種檢查方法：

　　一、檢查上傳時右下角的傳輸狀態，經過上傳成功標誌的個數來判斷是否上傳完成及成功

　　二、在文件列表的中判斷是否存在某個文件，代碼實現以下：

 1 # -*- coding:utf-8 -*-
 2 from WebDriver.webdriver import Driver
 3 from webpage.LoginPage.login import Login
 4 from packages.selenium.webdriver.common.by import By
 5 
 6 
 7 class List(object):
 8 
 9     def __init__(self):
10         self.page = Driver()
11 
12     @property
13     def checkbox(self):
14         return self.page.find_elements(By.CLASS_NAME, "EOGexf")
15 
16     @property
17     def file_num(self):
18         return len(self.checkbox)
19 
20     @property
21     def file_name(self):
22         file_class = self.page.find_elements(By.CLASS_NAME, "nnmz6W8")
23         num = len(file_class)
24         name = []
25         for i in range(num):
26             file_name = file_class.__getitem__(i).get_attribute("title")
27             name.append(file_name)
28         return name
29 
30     def select(self, name="test_txt_1.txt"):
31         """
32         選中列表中的某個文件
33         :param name: 須要選中的文件名
34         """
35         for i in range(self.file_num):
36             check = self.checkbox.__getitem__(i)
37             file_name = self.file_name.__getitem__(i)
38             check.click()
39             if file_name == name:
40                 print file_name
41                 return True
42             else:
43                 check.click()
44         return False
45 
46 
47 if __name__ == "__main__":
48     Login("XXX", "XXX")
49     li = List()
50     li.select("test_file.txt")

3、用例的管理

　　把main函數裏面的操做步驟連起來就是測試用例了，當用例過多時，須要用框架來管理，目前比較流行的用例管理框架有RobotFramework、unittest，這裏推薦unittest

 1 # -*- coding:utf-8 -*-
 2 from WebDriver.webdriver import Driver
 3 from webpage.LoginPage.login import Login
 4 from webpage.ListPage.list import List
 5 from File.create import create_txt
 6 from webpage.UploadFile.upload import Upload
 7 import unittest
 8 
 9 driver = Driver()
10 
11 
12 class TestCase(unittest.TestCase):
13 
14     @staticmethod
15     def setUpClass():
16         # 準備測試數據
17         create_txt(path="D:\\test_data2", file_num=10, file_size=50)
18 
19     def test_upload(self):
20         # 登陸系統
21         Login("XXX", "XXX")
22         # 上傳文件，upload方法後面少了一個wait_upload_finish()函數，可自行添加，若是想先看看效果，先time.sleep(20)
23         Upload("D:\\test_data2")
24         # 上傳檢查
25         li = List()
26         li.select("test_file.txt")
27 
28     def tearDown(self):
29         driver.quit()
30 
31     @staticmethod
32     def tearDownClass():
33         # 這個方法用來作一些清理工做，好比將已上傳的文件刪除
34         pass
35 
36     @staticmethod
37     def run_test():
38         suite = unittest.TestSuite()
39         # 此處只添加了一個用例，若是須要增長用例繼續用addTest添加便可
40         suite.addTest(TestCase("test_upload"))
41         unittest.TextTestRunner().run(suite)
42 
43 if __name__ == "__main__":
44     TestCase.run_test()

 

3、使用requests完成接口自動化

　　作過UI自動化的同窗必定會有這樣的感覺：

　　　　一、腳本運行慢，常常須要打開、關閉瀏覽器

　　　　二、常常出問題，維護須要消耗大量的精力，寫了這麼多用例沒有發現幾個BUG

　　因此如今大部分高自動化的童鞋都贊成優先作接口的自動化測試，這裏能夠按照相似的思路實現接口自動化測試用例，還能夠在UI自動化用例中引入接口腳本，

　　提高用例執行速度和減小維護工做，好比：須要測試分享功能，那能夠經過調用接口來獲得分享的URL，而後用selenium打開這個URL作接下來的測試

　　百度網盤有提供第三方開發者平臺：https://pan.baidu.com/platform/read；有興趣的同窗能夠了解下

 

4、擴展及豐富系統

　　爲了方便多人使用，能夠用Django作一個測試平臺出來，用來準備測試數據或管理測試用例，好比：

　　　　一、測試任務中須要20個文件的分享地址

　　　　二、須要測試某個操做的性能，須要50萬個文件

　　這樣能夠在平臺上生成便可，能在必定程度上提高測試效率，有興趣的朋友本身實現或關注我，下期奉上！