文件上傳自動化測試方案

一、概述

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

【測試工具】selenium、requests

【腳本語言】Python

【運行環境】Windows

百度網盤作為文件存儲及分享的平台，核心功能大部分是對文件的操作，如果要對它進行自動化測試，優先覆蓋的就是上傳、分享、復制功能

測試方案的大致結構如下：

　　　　　　　　　　　　　　圖 1-1 系統流程圖

二、上傳測試用例

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

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

　　操作步驟：1、用戶登錄

　　　　　　　2、遍歷生成的文件所在的文件夾，得到所有的文件路徑

　　　　　　　3、通過給FileUpload對象循環發送文件路徑完成上傳操作

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

三、每個步驟的具體實現

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

　　以txt為例，其他文件（csv、word、pdf、jpg等）生成方式類似，可以網上找相關的方法和第三方庫

 

# -*- recoding:utf-8 -*-
import os
import time

PATH = "D:/test_data/"


def create_txt(path=PATH, file_num=1, file_size=8):
    """
    :param path: 測試文件的存放路徑
    :param file_num: 需要生成的數量
    :param file_size: 每個文件的大小，以KB為單位
    """
    if not os.path.exists(path):
        os.mkdir(path)
    for i in range(file_num):
        file_name = "test_txt_"+str(i)+".txt"
        data = "test_t"+str(time.localtime())
        with open(os.path.join(path, file_name), 'w') as f_obj:
            for j in range(8*file_size):
                f_obj.write(data)
        print "create file: test_txt_(0-%d).txt in path %s success, file size: %sKB" % (file_num, path, file_size)


if __name__ == "__main__":
    # 生成包含不同文件大小的文件夾
    for i in range(10):
        path = "D:/test_data/" + str(i)
        create_txt(path=path, file_num=i, file_size=5*1024*i)

 

2、使用selenium通過web頁面上傳到服務器

　　首先是登錄腳本，對webdriver看不懂的同學先看我的另外一篇帖子：http://www.cnblogs.com/fightever/p/6867619.html

# -*- coding:utf-8 -*-
from WebDriver.webdriver import Driver
from selenium.webdriver.common.by import By


class Login():

    def __init__(self, user_name="admin", pwd="admin@123"):
        self.page = Driver()
        page = self.page
        page.start()
        self.account_login_btn = page.find_element(By.LINK_TEXT, u"帳號密碼登錄")
        self.account_text = page.find_element(By.ID, "TANGRAM__PSP_4__userName")
        self.pwd_text = page.find_element(By.ID, "TANGRAM__PSP_4__password")
        self.login_btn = page.find_element(By.ID, "TANGRAM__PSP_4__submit")
        self.login(user_name, pwd)

    def login(self, user_name, pwd):
        self.account_login_btn.click()
        self.account_text.clear()
        self.account_text.send_keys(user_name)
        self.pwd_text.clear()
        self.pwd_text.send_keys(pwd)
        self.login_btn.click()
        print "login success"


if __name__ == "__main__":
    # 將XXX換成自己注冊的百度網盤賬號
    Login("XXX", "XXX")

上傳文件腳本：

# -*- coding:utf-8 -*-
from WebDriver.webdriver import Driver
from selenium.webdriver.common.by import By
from File.create import create_txt
from webpage.LoginPage.login import Login
import os

OVER_TIME = 10


class Upload(object):

    def __init__(self, path):
        self.page = Driver()
        self.upload_file(path)

    def send_path(self, path):
        for i in range(OVER_TIME):
            try:
                self.page.find_element(By.NAME, "html5uploader").send_keys(path)
                return True
            except Exception, e:
                print e

    def upload_file(self, path, file_num=3):
        for root, dirs, files in os.walk(path):
            for f_obj in files:
                self.send_path(os.path.join(path, f_obj))


if __name__ == "__main__":
    Login("XXX", "XXX")
    Upload("D:\\test_data")

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

百度的上傳控件（這里使用的是name屬性“html5uploader”）：

 

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

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

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

# -*- coding:utf-8 -*-
from WebDriver.webdriver import Driver
from webpage.LoginPage.login import Login
from packages.selenium.webdriver.common.by import By


class List(object):

    def __init__(self):
        self.page = Driver()

    @property
    def checkbox(self):
        return self.page.find_elements(By.CLASS_NAME, "EOGexf")

    @property
    def file_num(self):
        return len(self.checkbox)

    @property
    def file_name(self):
        file_class = self.page.find_elements(By.CLASS_NAME, "nnmz6W8")
        num = len(file_class)
        name = []
        for i in range(num):
            file_name = file_class.__getitem__(i).get_attribute("title")
            name.append(file_name)
        return name

    def select(self, name="test_txt_1.txt"):
        """
        選中列表中的某個文件
        :param name: 需要選中的文件名
        """
        for i in range(self.file_num):
            check = self.checkbox.__getitem__(i)
            file_name = self.file_name.__getitem__(i)
            check.click()
            if file_name == name:
                print file_name
                return True
            else:
                check.click()
        return False


if __name__ == "__main__":
    Login("XXX", "XXX")
    li = List()
    li.select("test_file.txt")

三、用例的管理

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

# -*- coding:utf-8 -*-
from WebDriver.webdriver import Driver
from webpage.LoginPage.login import Login
from webpage.ListPage.list import List
from File.create import create_txt
from webpage.UploadFile.upload import Upload
import unittest

driver = Driver()


class TestCase(unittest.TestCase):

    @staticmethod
    def setUpClass():
        # 准備測試數據
        create_txt(path="D:\\test_data2", file_num=10, file_size=50)

    def test_upload(self):
        # 登錄系統
        Login("XXX", "XXX")
        # 上傳文件，upload方法后面少了一個wait_upload_finish()函數，可自行添加，如果想先看看效果，先time.sleep(20)
        Upload("D:\\test_data2")
        # 上傳檢查
        li = List()
        li.select("test_file.txt")

    def tearDown(self):
        driver.quit()

    @staticmethod
    def tearDownClass():
        # 這個方法用來做一些清理工作，比如將已上傳的文件刪除
        pass

    @staticmethod
    def run_test():
        suite = unittest.TestSuite()
        # 此處只添加了一個用例，如果需要增加用例繼續用addTest添加即可
        suite.addTest(TestCase("test_upload"))
        unittest.TextTestRunner().run(suite)

if __name__ == "__main__":
    TestCase.run_test()

 

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

　　做過UI自動化的同學一定會有這樣的感受：

　　　　1、腳本運行慢，經常需要打開、關閉瀏覽器

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

　　所以現在大部分高自動化的童鞋都同意優先做接口的自動化測試，這里可以按照類似的思路實現接口自動化測試用例，還可以在UI自動化用例中引入接口腳本，

　　提升用例執行速度和減少維護工作，比如：需要測試分享功能，那可以通過調用接口來得到分享的URL，然后用selenium打開這個URL做接下來的測試

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

 

四、擴展及豐富系統

　　為了方便多人使用，可以用Django做一個測試平台出來，用來准備測試數據或管理測試用例，比如：

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

　　　　2、需要測試某個操作的性能，需要50萬個文件

　　這樣可以在平台上生成即可，能在一定程度上提升測試效率，有興趣的朋友自己實現或關注我，下期奉上！