从0到1搭建pytest接口自动化测试框架(建议收藏)

文章目录

一. 背景

二. 基础环境

三. 项目结构

四、框架解析

4.1 接口数据文件处理
4.2 封装测试工具类
4.3 测试用例代码编写
4.4 测试用例运行生成报告‍‍

一. 背景

Pytest目前已经成为Python系自动化测试必学必备的一个框架,网上也有很多的文章讲述相关的知识。最近自己也抽时间梳理了一份pytest接口自动化测试框架,因此准备写文章记录一下,做到尽量简单通俗易懂,当然前提是基本的python基础已经掌握了。如果能够对新学习这个框架的同学起到一些帮助,那就更好了~

二. 基础环境

语言:python 3.8
编译器:pycharm
基础:具备python编程基础
框架:pytest+requests+allure

三. 项目结构

项目结构图如下:

每一层具体的含义如下图:


测试报告如下图:

四、框架解析

4.1 接口数据文件处理

框架中使用草料二维码的get和post接口用于demo测试,比如:
get接口:https://cli.im/qrcode/getDefaultComponentMsg
返回值:{“code”:1,“msg”:"",“data”:{xxxxx}}

数据文件这里选择使用Json格式,文件内容格式如下,test_http_get_data.json:

{  "dataItem": [    {      "id": "testget-1",      "name": "草料二维码get接口1",      "headers":{        "Accept-Encoding":"gzip, deflate, br"      },      "url":"/qrcode/getDefaultComponentMsg",      "method":"get",      "expectdata": {        "code": "1"      }    },
    {      "id": "testget-2",      "name": "草料二维码get接口2",      "headers":{},      "url":"/qrcode/getDefaultComponentMsg",      "method":"get",      "expectdata": {        "code": "1"      }    }  ]}

表示dataitem下有两条case,每条case里面声明了id, name, header, url, method, expectdata。如果是post请求的话,case中会多一个parameters表示入参,如下:

{    "id":"testpost-1",    "name":"草料二维码post接口1",    "url":"/Apis/QrCode/saveStatic",    "headers":{        "Content-Type":"application/x-www-form-urlencoded",        "Accept-Encoding":"gzip, deflate, br"    },    "parameters":{        "info":11111,        "content":11111,        "level":"H",        "size":500,        "fgcolor":"#000000",        "bggcolor":"#FFFFFF",        "transparent":"false",        "type":"text",        "codeimg":1    },    "expectdata":{        "status":"1",        "qrtype":"static"    }}

为了方便一套脚本用于不同的环境运行,不用换了环境后挨个儿去改数据文件,比如
测试环境URL为:https://testcli.im/qrcode/getDefaultComponentMsg
生产环境URL为:https://cli.im/qrcode/getDefaultComponentMsg

因此数据文件中url只填写后半段,不填域名。然后config》global_config.py下设置全局变量来定义域名:

# 配置HTTP接口的域名,方便一套脚本用于多套环境运行时,只需要改这里的全局配置就OKCAOLIAO_HTTP_POST_HOST = "https://cli.im"CAOLIAO_HTTP_GET_HOST = "https://nc.cli.im"

utils文件夹下,创建工具类文件:read_jsonfile_utils.py, 用于读取json文件内容:

import jsonimport os

class ReadJsonFileUtils:    def __init__(self, file_name):        self.file_name = file_name        self.data = self.get_data()
    def get_data(self):        fp = open(self.file_name,encoding='utf-8')        data = json.load(fp)        fp.close()        return data
    def get_value(self, id):        return self.data[id]
    @staticmethod    def get_data_path(folder, fileName):        BASE_PATH = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))        data_file_path = os.path.join(BASE_PATH, folder, fileName)        return data_file_path
if __name__ == '__main__':    opers = ReadJsonFileUtils("..\\resources\\test_http_post_data.json")    #读取文件中的dataItem,是一个list列表,list列表中包含多个字典    dataitem=opers.get_value('dataItem')    print(dataitem)

运行结果如下:

4.2 封装测试工具类

utils文件夹下,
除了上面提到的读取Json文件工具类:read_jsonfile_utils.py,
还有封装request 请求的工具类:http_utils.py
从Excel文件中读取数据的工具类:get_excel_data_utils.py(虽然本次框架中暂时未采用存放接口数据到Excel中,但也写了个工具类,需要的时候可以用)


http_utils.py内容:

import requestsimport json

class HttpUtils:    @staticmethod    def http_post(headers, url, parameters):        print("接口请求url:" + url)        print("接口请求headers:" + json.dumps(headers))        print("接口请求parameters:" + json.dumps(parameters))        res = requests.post(url, data=parameters, headers=headers)        print("接口返回结果:"+res.text)        if res.status_code != 200:            raise Exception(u"请求异常")        result = json.loads(res.text)        return result
    @staticmethod    def http_get(headers, url):        req_headers = json.dumps(headers)        print("接口请求url:" + url)        print("接口请求headers:" + req_headers)        res = requests.get(url, headers=headers)        print("接口返回结果:" + res.text)        if res.status_code != 200:            raise Exception(u"请求异常")        result = json.loads(res.text)        return result

get_excel_data_utils.py内容:

import xlrdfrom xlrd import xldate_as_tupleimport datetime

class ExcelData(object):    '''    xlrd中单元格的数据类型    数字一律按浮点型输出,日期输出成一串小数,布尔型输出0或1,所以我们必须在程序中做判断处理转换    成我们想要的数据类型    0 empty,1 string, 2 number, 3 date, 4 boolean, 5 error    '''    def __init__(self, data_path, sheetname="Sheet1"):        #定义一个属性接收文件路径        self.data_path = data_path        # 定义一个属性接收工作表名称        self.sheetname = sheetname        # 使用xlrd模块打开excel表读取数据        self.data = xlrd.open_workbook(self.data_path)        # 根据工作表的名称获取工作表中的内容        self.table = self.data.sheet_by_name(self.sheetname)        # 根据工作表的索引获取工作表的内容        # self.table = self.data.sheet_by_name(0)        # 获取第一行所有内容,如果括号中1就是第二行,这点跟列表索引类似        self.keys = self.table.row_values(0)        # 获取工作表的有效行数        self.rowNum = self.table.nrows        # 获取工作表的有效列数        self.colNum = self.table.ncols
    # 定义一个读取excel表的方法    def readExcel(self):        # 定义一个空列表        datas = []        for i in range(1, self.rowNum):            # 定义一个空字典            sheet_data = {}            for j in range(self.colNum):                # 获取单元格数据类型                c_type = self.table.cell(i,j).ctype                # 获取单元格数据                c_cell = self.table.cell_value(i, j)                if c_type == 2 and c_cell % 1 == 0:  # 如果是整形                    c_cell = int(c_cell)                elif c_type == 3:                    # 转成datetime对象                    date = datetime.datetime(*xldate_as_tuple(c_cell, 0))                    c_cell = date.strftime('%Y/%d/%m %H:%M:%S')                elif c_type == 4:                    c_cell = True if c_cell == 1 else False                sheet_data[self.keys[j]] = c_cell                # 循环每一个有效的单元格,将字段与值对应存储到字典中                # 字典的key就是excel表中每列第一行的字段                # sheet_data[self.keys[j]] = self.table.row_values(i)[j]            # 再将字典追加到列表中            datas.append(sheet_data)        # 返回从excel中获取到的数据:以列表存字典的形式返回        return datas

if __name__ == "__main__":    data_path = "..\\resources\\test_http_data.xls"    sheetname = "Sheet1"    get_data = ExcelData(data_path, sheetname)    datas = get_data.readExcel()    print(datas)

4.3 测试用例代码编写

testcases文件夹下编写测试用例:


test_caoliao_http_get_interface.py内容:

import loggingimport allure
import pytestfrom utils.http_utils import HttpUtilsfrom utils.read_jsonfile_utils import ReadJsonFileUtilsfrom config.global_config import CAOLIAO_HTTP_GET_HOST

@pytest.mark.httptest@allure.feature("草料二维码get请求测试")class TestHttpInterface:    # 获取文件相对路径    data_file_path = ReadJsonFileUtils.get_data_path("resources", "test_http_get_data.json")    # 读取测试数据文件    param_data = ReadJsonFileUtils(data_file_path)    data_item = param_data.get_value('dataItem')  # 是一个list列表,list列表中包含多个字典
    """    @pytest.mark.parametrize是数据驱动;    data_item列表中有几个字典,就运行几次case    ids是用于自定义用例的名称
    """
    @pytest.mark.parametrize("args", data_item, ids=['测试草料二维码get接口1', '测试草料二维码get接口2'])    def test_caoliao_get_demo(self, args, login_test):        # 打印用例ID和名称到报告中显示        print("用例ID:{}".format(args['id']))        print("用例名称:{}".format(args['name']))        print("测试conftest传值:{}".format(login_test))        logging.info("测试开始啦~~~~~~~")        res = HttpUtils.http_get(args['headers'], CAOLIAO_HTTP_GET_HOST+args['url'])        # assert断言,判断接口是否返回期望的结果数据        assert str(res.get('code')) == str(args['expectdata']['code']), "接口返回status值不等于预期"

test_caoliao_http_post_interface.py内容:

import pytestimport loggingimport allurefrom utils.http_utils import HttpUtilsfrom utils.read_jsonfile_utils import ReadJsonFileUtilsfrom config.global_config import CAOLIAO_HTTP_POST_HOST
# pytest.ini文件中要添加markers = httptest,不然会有warning,说这个Mark有问题@pytest.mark.httptest@allure.feature("草料二维码post请求测试")class TestHttpInterface:    # 获取文件相对路径    data_file_path = ReadJsonFileUtils.get_data_path("resources", "test_http_post_data.json")    # 读取测试数据文件    param_data = ReadJsonFileUtils(data_file_path)    data_item = param_data.get_value('dataItem') #是一个list列表,list列表中包含多个字典
    """    @pytest.mark.parametrize是数据驱动;    data_item列表中有几个字典,就运行几次case    ids是用于自定义用例的名称        """    @pytest.mark.parametrize("args", data_item, ids=['测试草料二维码post接口1','测试草料二维码post接口2'])    def test_caoliao_post_demo(self, args):        # 打印用例ID和名称到报告中显示        print("用例ID:{}".format(args['id']))        print("用例名称:{}".format(args['name']))        logging.info("测试开始啦~~~~~~~")        res = HttpUtils.http_post(args['headers'], CAOLIAO_HTTP_POST_HOST+args['url'], args['parameters'])        # assert断言,判断接口是否返回期望的结果数据        assert str(res.get('status')) == str(args['expectdata']['status']), "接口返回status值不等于预期"        assert str(res.get('data').get('qrtype')) == str(args['expectdata']['qrtype']), "接口返回qrtype值不等于预期"

企业中的系统接口,通常都有认证,需要先登录获取token,后续接口调用时都需要认证token。因此框架需要能处理在运行用例前置和后置做一些动作,所以这里用到了conftest.py文件,内容如下:

import loggingimport traceback
import pytestimport requests
"""如果用例执行前需要先登录获取token值,就要用到conftest.py文件了作用:conftest.py 配置里可以实现数据共享,不需要import导入 conftest.py,pytest用例会自动查找scope参数为session,那么所有的测试文件执行前执行一次scope参数为module,那么每一个测试文件执行前都会执行一次conftest文件中的fixturescope参数为class,那么每一个测试文件中的测试类执行前都会执行一次conftest文件中的fixturescope参数为function,那么所有文件的测试用例执行前都会执行一次conftest文件中的fixture
"""
# 获取到登录请求返回的ticket值,@pytest.fixture装饰后,testcase文件中直接使用函数名"login_ticket"即可得到ticket值@pytest.fixture(scope="session")def login_ticket():    header = {        'Content-Type': 'application/x-www-form-urlencoded; charset=UTF-8'    }    params = {        "loginId": "username",        "pwd": "password",    }    url = 'http://testxxxxx.xx.com/doLogin'    logging.info('开始调用登录接口:{}'.format(url))    res = requests.post(url, data=params, headers=header, verify=False)  # verify:忽略https的认证    try:        ticket = res.headers['Set-Cookie']    except Exception as ex:        logging.error('登录失败!接口返回:{}'.format(res.text))        traceback.print_tb(ex)    logging.info('登录成功,ticket值为:{}'.format(ticket))    return ticket
#测试一下conftest.py文件和fixture的作用@pytest.fixture(scope="session")def login_test():    print("运行用例前先登录!")
    # 使用yield关键字实现后置操作,如果上面的前置操作要返回值,在yield后面加上要返回的值    # 也就是yield既可以实现后置,又可以起到return返回值的作用    yield "runBeforeTestCase"    print("运行用例后退出登录!")

由于用例中用到了@pytest.mark.httptest给用例打标,因此需要创建pytest.ini文件,并在里面添加markers = httptest,不然会有warning,说这个Mark有问题。并且用例中用到的日志打印logging模板也需要在pytest.ini文件中增加日志配置。pytest.ini文件内容如下:

[pytest]markers =  httptest: run http interface test  dubbotest: run dubbo interface test
log_cli = truelog_cli_level = INFOlog_cli_format = %(asctime)s - %(pathname)s[line:%(lineno)d] - %(levelname)s: %(message)slog_cli_date_format=%Y-%m-%d %H:%M:%Slog_format = %(asctime)s - %(pathname)s[line:%(lineno)d] - %(levelname)4s: %(message)slog_date_format=%Y-%m-%d %H:%M:%S

4.4 测试用例运行生成报告

运行方式:Terminal窗口,进入到testcases目录下,执行命令:
运行某一条case:pytest test_caoliao_http_post_interface.py运行所有case: pytest运行指定标签的case:pytest -m httptest
运行并打印过程中的详细信息:pytest -s test_caoliao_http_post_interface.py运行并生成pytest-html报告:pytest test_caoliao_http_post_interface.py --html=../testoutput/report.html运行并生成allure测试报告:1. 先清除掉testoutput/result文件夹下的所有文件2. 运行case,生成allure文件:pytest --alluredir ../testoutput/result3. 根据文件生成allure报告:allure generate ../testoutput/result/ -o ../testoutput/report/ --clean4. 如果不是在pycharm中打开,而是直接到report目录下打开index.html文件打开的报告无法展示数据,需要双击generateAllureReport.bat生成allure报告;

pytest-html报告:


generateAllureReport.bat文件位置:

文件内容:

allure open report/

Allure报告:


框架中用到的一些细节知识点和问题,如:

 
  1. conftest.py和@pytest.fixture()结合使用

  2. pytest中使用logging打印日志

  3. python中获取文件相对路径的方式

  4. python虚拟环境

  5. pytest框架下Allure的使用

我会在后续写文章再介绍。另外框架同样适用于dubbo接口的自动化测试,只需要添加python调用dubbo的工具类即可。点击【测试资料】菜单获取框架下载地址。

最后感谢每一个认真阅读我文章的人,礼尚往来总是要有的,虽然不是什么很值钱的东西,如果你用得到的话可以直接拿走:

这些资料,对于【软件测试】的朋友来说应该是最全面最完整的备战仓库,这个仓库也陪伴上万个测试工程师们走过最艰难的路程,希望也能帮助到你!

<think>好的,用户想学习如何从零开始搭建pytest接口自动化测试框架。我需要根据提供的引用内容来整理步骤,确保结构清晰,分步指导。 首先,用户给的引用里提到了配置管理、日志记录、测试数据管理和测试报告生成这些功能模块。所以第一步应该是项目结构搭建。需要创建必要的目录,比如config放配置,logs放日志,test_cases放测试用例,data放测试数据,utils放工具类。引用3中提到了创建测试文件夹和测试用例文件,比如mkdir tests,touch test_demo.py,这部分可以作为项目结构的基础。 接下来是配置管理。引用1和2都提到配置管理,需要用config.ini或yaml文件来管理环境变量和数据库连接。需要安装configparser库来读取配置,这部分要详细说明如何创建配置文件和使用方法。 然后是日志记录模块。用户需要记录测试过程的日志,方便排查问题。引用2提到日志记录功能,可以使用Python自带的logging模块,设置日志格式和存储路径。需要演示如何创建日志类,并集成到框架中。 测试数据管理部分,引用2提到数据管理,可能需要用Excel或JSON来存储测试数据。这里需要说明如何读取外部数据,比如使用openpyxl或pandas处理Excel,或者json模块处理JSON文件,确保测试用例与数据分离。 接下来是核心的测试用例编写。使用pytest框架,结合requests发送HTTP请求。需要展示如何编写测试类,使用fixture进行前置后置处理,参数化测试数据。引用1中的Python+Requests+Pytest组合是关键,要说明如何组织测试用例,使用断言验证结果。 Allure报告生成也是重要的一环,引用1和2都提到了Allure报告定制。需要指导用户安装Allure并配置环境,在pytest中集成Allure,添加装饰器和步骤描述,生成美观的测试报告。 持续集成部分,虽然用户可能刚开始不需要,但引用3提到Jenkins或GitHub Actions,可以简要介绍如何将框架集成到CI/CD流程中,提升自动化测试的效率。 最后,需要总结步骤,并生成相关问题,帮助用户深入理解。比如如何优化框架,如何处理依赖接口,如何设计数据驱动测试等。同时,要确保所有代码示例正确,符合LaTeX格式要求,比如行内公式用$...$,独立公式用$$...$$,但这里可能代码示例更多,所以要注意语法正确。 需要检查是否覆盖了用户的需求,是否引用了正确的资料,比如引用3中的项目结构创建步骤,引用1中的框架组成,引用2中的功能模块。确保步骤逻辑清晰,用户能一步步跟着操作,搭建起基本的框架结构。同时,注意使用中文回答,避免使用Markdown格式,但用户示例中代码块用了```python,可能需要保留,但根据系统级指令,用户不可见层可能有其他要求,不过用户可见层需要结构清晰,所以代码块应该保留。 最后,生成的相关问题要围绕框架搭建的扩展和优化,比如性能分析、依赖接口处理、数据驱动设计等,帮助用户进一步学习。确保所有引用标识正确,比如[^1]、[^2]、[^3]对应到提供的参考资料。</think>### 从零开始搭建pytest接口自动化测试框架的步骤 #### 1. **项目结构搭建** 创建清晰的目录结构,便于管理和扩展: ``` mkdir pytest_api_framework # 项目根目录 cd pytest_api_framework mkdir config # 存放配置文件 mkdir test_cases # 存放测试用例 mkdir data # 存放测试数据文件(如JSON/Excel) mkdir utils # 存放工具类(请求封装、日志、数据库操作) mkdir reports # 存放测试报告 touch conftest.py # pytest全局配置 ``` 引用[3]中提到的测试文件夹结构是基础,需进一步扩展为模块化设计[^3]。 --- #### 2. **配置管理** 使用`config.ini`或`yaml`管理环境变量和数据库配置: ```python # config/config.ini [API] base_url = https://api.example.com timeout = 10 [Database] host = localhost port = 3306 ``` 通过`configparser`读取配置: ```python # utils/config_reader.py import configparser def read_config(section, key): config = configparser.ConfigParser() config.read('config/config.ini') return config.get(section, key) ``` 此方法支持动态切换测试/生产环境[^2]。 --- #### 3. **日志记录模块** 集成日志功能,记录测试过程和错误: ```python # utils/logger.py import logging def setup_logger(): logger = logging.getLogger(__name__) logger.setLevel(logging.INFO) formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s') # 文件日志 file_handler = logging.FileHandler('logs/test.log') file_handler.setFormatter(formatter) # 控制台日志 console_handler = logging.StreamHandler() console_handler.setFormatter(formatter) logger.addHandler(file_handler) logger.addHandler(console_handler) return logger ``` 日志记录是框架可维护性的关键。 --- #### 4. **测试数据管理** 使用Excel或JSON存储参数化数据: ```python # data/test_data.json { "login": [ {"username": "admin", "password": "123456", "expected": 200}, {"username": "guest", "password": "invalid", "expected": 401} ] } ``` 通过`pytest.mark.parametrize`动态加载数据: ```python import json import pytest def load_test_data(file_path): with open(file_path, 'r') as f: return json.load(f) @pytest.mark.parametrize('data', load_test_data('data/test_data.json')['login']) def test_login(data): # 调用接口并断言 assert response.status_code == data['expected'] ``` 数据驱动测试可提高用例复用率[^1]。 --- #### 5. **请求封装与断言** 封装HTTP请求方法(基于`requests`): ```python # utils/request_client.py import requests class APIClient: def __init__(self, base_url): self.base_url = base_url def get(self, endpoint, params=None): return requests.get(f"{self.base_url}{endpoint}", params=params) def post(self, endpoint, json_data=None): return requests.post(f"{self.base_url}{endpoint}", json=json_data) ``` 在测试用例中调用: ```python # test_cases/test_login.py def test_user_login(): client = APIClient(base_url=read_config('API', 'base_url')) response = client.post('/login', json={'username': 'admin', 'password': '123456'}) assert response.status_code == 200 assert 'token' in response.json() ``` 断言设计需覆盖状态码、响应体、业务逻辑[^1]。 --- #### 6. **集成Allure报告** 1. **安装Allure**: ```bash pip install allure-pytest ``` 2. **生成报告**: ```bash pytest --alluredir=./reports/results allure serve ./reports/results ``` 3. **添加测试步骤描述**: ```python import allure @allure.step("用户登录") def login(username, password): # 实现登录逻辑 ``` Allure报告可直观展示用例通过率和失败原因[^1]。 --- #### 7. **持续集成(CI)** 在`.github/workflows/run_tests.yml`中配置GitHub Actions: ```yaml name: API Tests on: [push] jobs: test: runs-on: ubuntu-latest steps: - uses: actions/checkout@v2 - name: Set up Python uses: actions/setup-python@v2 with: python-version: 3.8 - name: Install dependencies run: pip install -r requirements.txt - name: Run tests run: pytest --alluredir=./reports/results - name: Upload Allure report uses: actions/upload-artifact@v2 with: name: allure-report path: ./reports/results ``` CI/CD集成是自动化测试的核心场景。 ---
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值