12.数据驱动ddt

安装ddt：pip install ddt

安装xlrd：pip install xlrd

注：ddt不能脱离unittest使用，unittest框架会循环执行excel表中数据，无需写for循环

 

一、数据在脚本中维护(unittest框架，ddt)

从脚本中读取数据，基本思路：

1.数据data1写成list的格式

2.脚本中加2行修饰符：类前加@ddt.ddt; 测试用例前加@ddt.data(*data1) 

3.在test_开头的测试用例中，给一个参数testdata接收数据

4.脚本中读取data1

#  coding:utf-8
import ddt
import unittest

data1 = [
    {"user": "test11", "psw": "123"},
    {"user": "test22", "psw": "456"},
    {"user": "test33", "psw": "789"}
]

@ddt.ddt  #用到ddt需要在类上边加修饰符
class GetData(unittest.TestCase):

    @ddt.data(*data1)  #方法上边加修饰符，分开传入data1的数据
    def test_get_data(self, testdata):  #传变量testdata去接受数据
        print("测试数据为：%s" %testdata)

if __name__ == "__main__":
    unittest.main()

 

二、数据在Excel中维护(unittest框架，ddt)

Excel操作基本方法:

data = xlrd.open_workbook(‘test.xlsx’)  #打开excel表格，参数是文件路径

# table = data.sheets()[0]  #通过索引顺序获取

# table = data.sheet_by_index(0)  #通过索引顺序获取

table = data.sheet_by_name(u‘Sheet’)  #通过名称获取

nrows = table.nrows   #获取总行数

ncols = table.ncols   #获取总列数

# 获取一行或一列的值，参数是第几行

table.row_values(0)  #获取第一行值

table.col_values(0)  #获取第一列值

注意：excel中的数字内容，设置文本格式保存

 

从excel中读取数据，基本思路：

1.封装方法，读取excel。

# coding:utf-8
import xlrd
class ExcelUtil():
    def __init__(self, excelPath, sheetName):
        self.data = xlrd.open_workbook(excelPath)
        self.table = self.data.sheet_by_name(sheetName)
        # 获取第一行作为key值
        self.keys = self.table.row_values(0)
        # 获取总行数
        self.rowNum = self.table.nrows
        # 获取总列数
        self.colNum = self.table.ncols

    def dict_data(self):
        if self.rowNum <= 1:
            print("总行数小于1")
        else:
            r = []
            j=1
            for i in range(self.rowNum-1):
                s = {}
                # 从第二行取对应values值
                values = self.table.row_values(j)
                for x in range(self.colNum):
                    s[self.keys[x]] = values[x]
                r.append(s)
                j+=1
            return r

if __name__ == "__main__":
    filepath = r"D:\apidatadriver\data.xlsx"
    sheetName = "Sheet1"
    data = ExcelUtil(filepath, sheetName)
    print(data.dict_data())

2.以list格式打印出来，list中是字典格式

3.脚本中加2行修饰符：类前加@ddt.ddt; 测试用例前加@ddt.data(*data1) 

4.在test_开头的测试用例中，给一个参数testdata接收数据

5.脚本中读取data1

#  coding:utf-8
import requests

class Login():
    #登录，返回跳转的url地址
    def login(self, s, username="admin", pwd="123456"):
        url = "http://xxxx/login"
        h = {
            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; WOW64; rv:46.0) Gecko/20100101 Firefox/46.0"
        }
        body = {
            "username": username,
            "pwd": pwd,
            "kaptcha": None
        }
        r = s.post(url, headers=h, json=body)
        newurl = r.json()["redirectUrl"]
        # print(r.content.decode("utf-8"))
        # print(r.status_code)
        return newurl

    #打开上一步返回的url地址，获取cookies，用s去自动管理
    def get_cookies(self, s, newurl):
        r = s.get(newurl, verify=False)
        # print(r.content.decode("utf-8"))

    def get_version(self, s):
        url = "http://xxxxx/getversion"
        r = s.post(url)
        return r.json()["data"]["version"]


if __name__ == "__main__":
    login1 = Login()
    s = requests.session()
    newurl = login1.login(s)
    login1.get_cookies(s, newurl)
    login1.get_version(s)

 

# coding:utf-8
from version.erp_01_login import Login
from version.readexcel import ExcelUtil
import requests
import ddt
import unittest

# 从文件中读取数据，赋值给data1，以list中字典格式显示，需要维护excel表
d = ExcelUtil("userinfo.xlsx", sheetName="Sheet1")
data1 = d.dict_data()
print(data1)

@ddt.ddt
class TestGetVersion(unittest.TestCase):
    '''
    获取租户的版本号
    '''
    @ddt.data(*data1)
    def test_version(self, info):
        erplogin = Login()
        s = requests.session()
        username = info['code']
        code = info['username']
        password = info['password']

        try:
            newurl = erplogin.login(s, username, code, password)
            erplogin.get_cookies(s, newurl)
            ver = erplogin.get_version(s)
            print("租户%s的版本号为：%s" % (username.center(10), ver))    #% (username.ljust(10), ver)左对齐，填充到10位
        except:
            print("租户%s登录失败" % username.center(10))

if __name__ == "__main__":
    unittest.main()

维护的excel表如下：

 

三、从excel中读取数据(无ddt)

#  coding: utf-8
from readfromexcel.readexcel import ExcelUtil

class ReadFromExcel():
    def readFromExcel(self):
        d = ExcelUtil("userinfo.xlsx", sheetName="Sheet1")
        data1 = d.dict_data()
        print(data1)
        print(type(data1))

if __name__ == "__main__":
    dt = ReadFromExcel()
    dt.readFromExcel()

#  coding: utf-8
from readfromexcel.readexcel import ExcelUtil

class ReadFromExcel():
    def readFromExcelBySheetName(self):
        '''第一种方式读取多个Sheet中的数据'''
        sheets = ["Sheet1", "Sheet2", "Sheet3"]
        for i in sheets:
            try:
                d = ExcelUtil("userinfo.xlsx", sheetName=i)
                data1 = d.dict_data()
                print("%s的数据：" %i)
                print(data1)
            except:
                print("%s无数据：" %i)

    def readFromExcelBySheetName1(self):
        '''第二种方式读取多个Sheet中的数据'''
        for i in range(1, 4):
            sheetName = "Sheet%s" %i
            try:
                d = ExcelUtil("userinfo.xlsx", sheetName=sheetName)
                data1 = d.dict_data()
                print("Sheet%s"%i,"的数据")
                print(data1)
            except:
                print("Sheet%s"%i,"无数据")

if __name__ == "__main__":
    dt = ReadFromExcel()
    dt.readFromExcelBySheetName()
    dt.readFromExcelBySheetName1()