Python自動化測試之封裝讀取Excel類實現(xiàn)方式

更新時間：2025年11月24日 10:33:04 作者：葬愛家族小阿杰

文章介紹了數(shù)據(jù)驅動測試（DDT）的思想和使用步驟,包括使用@ddt和@list_data裝飾器,以及如何在測試方法中接收用例數(shù)據(jù),文章還簡要介紹了DDT的底層原理,涉及到保存和遍歷測試數(shù)據(jù),最后,文章提供了一個綜合運用的示例,并呼吁大家支持腳本之家

ddt的使用步驟

1、測試類前面使用@ddt

2、在測試方法前使用@list_data（測試數(shù)據(jù)）

3、在測試方法中定義一個參數(shù)，用于接收用例數(shù)據(jù)

ddt：DataDriverTest（數(shù)據(jù)驅動測試）

思想：把測試數(shù)據(jù)和用例代碼進行分離，通過外部數(shù)據(jù)去生成測試用例，方便后期維護，以及測試數(shù)據(jù)管理

ddt底層原理

1、保存測試數(shù)據(jù)

2、遍歷測試數(shù)據(jù)，生成測試方法

import unittest

from unittestreport import ddt,list_data
import requests

cases = [
    {"excepted":{"code":0, "msg":"登陸成功"},"params":{"username":"aaa", "password":"123456"}},
    {"excepted":{"code":1, "msg":"登陸失敗"},"params":{"username":"aaa", "password":"123"}},
    {"excepted":{"code":1, "msg":"登陸失敗"},"params":{"username":" ", "password":"123456"}},
]

@ddt
class TestLogin(unittest.TestCase):
    @list_data(cases)
    def test_login(self, item):
        data = item['params']
        expected = item['expected']
        res = login_check(**data)
        self.assertEqual(expected, res)

封裝讀取Excel類

import openpyxl

class HandleExcel:
    def __init__(self, filename, sheetname):
        '''
        :param filename: excel文件名字
        :param sheetname: 表單名
        '''
        self.filename = filename
        self.sheetname = sheetname

    def read_data(self):
        # 加載excel工作簿
        workbook = openpyxl.load_workbook(self.filename)
        # 選中表單
        sh = workbook[self.sheetname]
        # 獲取所有數(shù)據(jù)
        cel = list(sh.rows)
        # 變量第一行數(shù)據(jù)
        tital = [i.value for i in cel[0]]
        cases = []
        # 遍歷除第一行以外的行
        for item in cel[1:]:
            # 獲取該行的數(shù)據(jù)
            data = [i.value for i in item]
            # 把第一行數(shù)據(jù)和當前行數(shù)據(jù)打包成字典
            dic = dict(zip(tital, data))
            # 把字典添加到列表中
            cases.append(dic)
        return cases

    def write_data(self, row, column, value):
        '''
        :param row: 寫入行
        :param column: 寫入列
        :param value: 寫入值
        :return:
        '''
        workbook = openpyxl.load_workbook(self.filename)
        sh = workbook[self.sheetname]
        sh.cell(row=row, column=column, value=value)
        workbook.save(self.filename)

    def close_file(self):
        self.close_file(self.filename)

if __name__ == '__main__':
    t = HandleExcel("cases.xlsx", "login").read_data()
    print(t)

綜合運用：

import unittest
import os
import requests
from unittestreport import ddt,list_data
from commons.handle_excel import HandleExcel
from commons.handle_path import DATAS_DIR
from commons.handle_config import conf
from commons.tools import replace_data
from commons.handle_log import log
@ddt
class TestLogin(unittest.TestCase):
    excel=HandleExcel(os.path.join(DATAS_DIR,'apicases.xlsx'),'login')
    cases=excel.read_data()
    @list_data(cases)
    def test_login(self,item):
        # 1、準備數(shù)據(jù)
        url=conf.get('env','base_url')+item['url']
        item['data']=replace_data(item['data'],TestLogin)
        params=eval(item['data'])
        method=item['method'].lower()
        expected=eval(item['expected'])
        # 2、請求接口
        response=requests.request(method=method,url=url,json=params)
        res=response.json()
        code = response.status_code
        print('請求參數(shù)：', params)
        print('預期結果：', expected)
        print('實際結果：', res)
        print('返回碼：', code)
        # 3、斷言
        try:
            self.assertEqual(expected['code'],code)
            self.assertIn(expected['msg'],res.values())
        except AssertionError as e:
            log.error('用例-----【{}】-----執(zhí)行不通過'.format(item['title']))
            log.exception(e)
            raise e
        else:
            log.info('用例-----【{}】------執(zhí)行通過'.format(item['title']))