文件上传自动化测试方案

news2024/12/26 0:05:33

一、概述

【测试地址】：https://pan.baidu.com

【测试工具】selenium、requests

【脚本语言】Python

【运行环境】Windows

百度网盘作为文件存储及分享的平台，核心功能大部分是对文件的操作，如果要对它进行自动化测试，优先覆盖的就是上传、分享、复制功能

测试方案的大致结构如下：

二、上传测试用例

以文件上传功能为例，测试用例描述如下：

预置条件：生成指定数量和大小的测试文件（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看不懂的同学可以关注我的公众号进群交流

# -*- 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")

上传文件脚本：

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”）：

在这里插入图片描述
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()