如何用 python +ddt+excel 實現(xiàn)接口自動化測試

接口自動化測試可以提高測試效率和質量，節(jié)省測試成本和時間，保證測試覆蓋率和可維護性。

讓勇哥帶你入門如何用 python +ddt+excel 實現(xiàn)接口自動化測試，請瞧如下內(nèi)容：

一、準備工作

1. 安裝 python 環(huán)境

python 是一種簡潔、優(yōu)雅、易學的編程語言，它有豐富的第三方庫和社區(qū)支持，非常適合用來進行接口自動化測試。安裝 python 的3.0以上的版本），并配置好環(huán)境變量，以便在命令行中使用 python 命令。

2. 安裝相關庫

為了進行接口自動化測試，我們需要安裝以下幾個庫：

• requests：用來發(fā)送 HTTP 請求，支持各種方法、參數(shù)、頭部、身份驗證等。

• xlrd：用來讀取 excel 文件中的數(shù)據(jù)，支持 xls 和 xlsx 格式。

• openpyxl：用來寫入 excel 文件中的數(shù)據(jù)，支持 xlsx 格式。

• ddt：用來實現(xiàn)數(shù)據(jù)驅動測試，可以從 excel 文件中讀取多組數(shù)據(jù)，并生成多個測試用例。

• unittest：用來編寫和執(zhí)行測試用例，支持斷言、前置后置條件、測試套件等。

• HTMLTestRunner_api：用來生成 HTML 格式的測試報告，支持圖表、日志、截圖等。

我們可以使用 pip 命令來安裝這些庫，例如：

pip install requests

二、設計測試

1. 編寫 excel 文件

我們需要在 excel 文件中填寫接口的相關信息，包括：

• 接口名稱

• 請求地址

• 請求方法

• 請求參數(shù)

• 預期結果

例如：

我們可以將這個 excel 文件保存為 test_data.xlsx，并放在項目的 test_datas 目錄下。

2. 封裝 excel 操作

為了方便地讀取和寫入 excel 文件中的數(shù)據(jù)，我們需要封裝一些 excel 操作的函數(shù)，例如：

• 打開表單

• 獲取表頭

• 獲取所有數(shù)據(jù)

• 寫入數(shù)據(jù)

我們可以將這些函數(shù)定義在一個類中，并保存為 excel_handler.py，并放在項目的 Lib 目錄下。具體代碼如下：

from openpyxl import load_workbook
from openpyxl.worksheet.worksheet import Worksheet

class ExcelHandler():
    '''操作Excel'''
    def __init__(self, file):
        '''初始化函數(shù)'''
        self.file = file
    
    def open_sheet(self, sheet_name) -> Worksheet:
        '''打開表單'''
        wb = load_workbook(self.file)
        sheet = wb[sheet_name]
        return sheet
    
    def read_rows(self,sheet_name):
        '''讀取除表頭外所有數(shù)據(jù)（除第一行外的所有數(shù)據(jù)）'''
        sheet = self.open_sheet(sheet_name)
        rows = list(sheet.rows)[1:]
        data = []
        for row in rows:
            row_data = []
            for cell in row:
                row_data.append(cell.value)
            data.append(row_data)
        return data
???? def read_key_value(self,sheet_name):
 ? ? ? ?'''獲取所有數(shù)據(jù)，將表頭與內(nèi)容結合整....'''????????????????...
 ? ? ? ?return data

3. 封裝 requests 請求

為了方便地發(fā)送 HTTP 請求，我們需要封裝一些 requests 請求的函數(shù)，例如：

• 初始化 session

• 發(fā)送請求

• 關閉 session

我們可以將這些函數(shù)定義在一個類中，并保存為 requests_handler.py，并放在項目的 Lib 目錄下。具體代碼如下：

import requests

class HTTPHandler:
    # 初始化
    def __init__(self):
        self.session = requests.Session()
    
    # 定義一個方法，接收訪問http請求的方式
    def visit(self, url, method, params=None, data=None, json=None, **kwargs):
        res = self.session.request(method, url, params=params, data=data, json=json, **kwargs)
        try:
            return res.json()
        except ValueError:
            print('return not json')
    
    # 關閉session會話
    def close_session(self):
        self.session.close()

三、執(zhí)行測試

1. 編寫測試用例

我們需要使用 unittest 和 ddt 來編寫和執(zhí)行測試用例，具體步驟如下：

• 導入相關庫和模塊

• 定義一個測試類，繼承 unittest.TestCase

• 定義一個類方法，用來初始化 session 和讀取 excel 文件中的數(shù)據(jù)

• 使用 @ddt.data 裝飾器，傳入 excel 文件中的數(shù)據(jù)

• 定義一個測試方法，用來發(fā)送請求，并斷言響應結果是否符合預期

我們可以將這些代碼保存為 test_api.py，并放在項目的 test_cases 目錄下。具體代碼如下：

from Lib.excel_handler import ExcelHandler
from Lib.requests_handler import HTTPHandler
import requests
import ddt
import unittest

data = ExcelHandler('test_datas/test_data.xlsx').read_key_value('Sheet1')

@ddt.ddt
class TestAPI(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.s = HTTPHandler()
    
    @ddt.data(*data)
    def test_api(self,params):
        print('params:%s' % params)
        case_name = params.get('接口名稱')
        url = params.get('請求地址')
        args = eval(params.get('請求參數(shù)')) if isinstance(params.get('請求參數(shù)'), str) else params.get('請求參數(shù)')
        method = params.get('請求方法')
        expct_res1 = params.get('預期結果')
        
        # 發(fā)起請求，獲取返回數(shù)據(jù)
        result = self.s.visit(url, method, params=args)

        # 分析返回數(shù)據(jù)
        response_data = result['name']

        
        # 斷言響應結果是否符合預期
        self.assertEqual(expct_res1, response_data)

    @classmethod
    def tearDownClass(cls):
       cls.s.close_session()

2. 生成測試報告

我們需要使用 HTMLTestRunner_api 來生成 HTML 格式的測試報告，具體步驟如下：

• 導入相關庫和模塊

• 定義一個測試套件，添加測試用例

• 定義一個測試報告的文件名和路徑

• 定義一個測試運行器，傳入測試報告的文件對象和相關參數(shù)

• 使用測試運行器來運行測試套件

我們可以將這些代碼保存為 run.py，并放在項目的根目錄下。具體代碼如下：

import unittest
from HTMLTestRunner_api import HTMLTestRunner
from test_cases.test_api import TestAPI

# 定義一個測試套件
suite = unittest.TestSuite()
# 添加測試用例
suite.addTest(unittest.makeSuite(TestAPI))

# 定義一個測試報告的文件名和路徑
report_file = 'reports/test_report.html'

# 定義一個測試運行器，傳入測試報告的文件對象和相關參數(shù)
with open(report_file, 'wb') as f:
    runner = HTMLTestRunner(f, title='接口自動化測試報告', description='用例執(zhí)行情況')
    # 使用測試運行器來運行測試套件
    runner.run(suite)

四、查看結果

運行 run.py 文件后，我們可以在 reports 目錄下看到生成的 test_report.html 文件，打開它，我們可以看到以下內(nèi)容：

我們可以看到，共有 11 個測試用例，通過 6 個，失敗 5 個。我們還可以看到每個測試用例的詳細信息，包括用例名稱、請求地址、請求參數(shù)、預期結果、實際結果等。我們還可以看到一些圖表、日志、截圖等。

這樣，我們就完成了用 python +ddt+excel 實現(xiàn)接口自動化測試的過程。

?