环境准备

本次选用的是Python+Pytest+requests来搭建自动化框架，需要有 Python 环境（3.x 版本），安装pytest和requests，不会安装的自行去网上搜教程。

关于 Pytest 的基本使用，这里不作讲解，不清楚的可以看我之前的文章。

简单 Demo

我们先创建一个项目，取名api-test，然后在下面创建一个test_api.py的文件

# test_api.pyimport pytestimport requestsdef test_01():response = requests.get('http://127.0.0.1:5000/login').json()print(response)assert response.get('success') == 'ok'if __name__ == '__main__':pytest.main()

直接运行，结果如下

============================= test session starts =============================platform win32 -- Python 3.7.1, pytest-6.2.1, py-1.10.0, pluggy-0.13.1rootdir: D:\study\api-testplugins: allure-pytest-2.8.31, dependency-0.5.1, forked-1.3.0, ordering-0.6, rerunfailures-9.1.1, xdist-2.2.1collected 1 itemtest_api.py .{'success': 'ok', 'token': 'sdkhasgdhgasjhdgasgda'}[100%]============================== 1 passed in 0.16s ==============================Process finished with exit code 0

一个简单的测试用例就完成了。

自动化测试工作中，往往会有很多测试文件，每次都需要导入 requests 库吗？

域名都是写在用例一起吗？

如果要记录日志，每次都在用例里面记录吗？

测试数据也写在用例里吗？

等等，答案显然是否定的，下面就让我们一起踏上框架封装之路。

首先我们创建一个包：base，用于存放最基础的封装方法，在包下创建一个模块http_client.py，此处用于封装 requests 发送请求的方法。

封装思路：

创建 HttpClient 类，方便后续 api 继承后直接调用类下面的 get 和 post 方法创建__request 私有方法，作为发送请求公共方法，方便后面日志搜集请求信息和返回信息使用 requests.session()，requests 库的 session 会话对象可以跨请求保持某些参数

# base/http_client.pyimport requestsclass HttpClient():def __init__(self):self.__session = requests.session()def get(self, path, **kwargs):return self.__request(path, 'GET', **kwargs)def post(self, path, data=None, json=None, **kwargs):return self.__request(path, 'POST', data, json, **kwargs)def __request(self, url, method, data=None, json=None, **kwargs):resp = Noneif method == "GET":resp = self.__session.get(url, **kwargs)elif method == "POST":resp = requests.post(url, data, json, **kwargs)return resp

然后我们使用封装好的 requests 替换用例中的 requests

# test_api.pyimport pytestfrom base.http_client import HttpClienthttp_client = HttpClient()def test_01():response = http_client.get('http://127.0.0.1:5000/login').json()print(response)assert response.get('success') == 'ok'if __name__ == '__main__':pytest.main()

同样能够运行成功。

那么问题来了，如果多个 py 文件中都需要调用到同一个接口，比如登录，是不是所有文件中都需要写一遍呢？哪天接口路径改了，那修改起来是一件很麻烦的事情，还容易出错。

那我的想法是把这个接口提取到一个公共的地方封装起来，那么就可以给其他用例文件导入使用，即使有修改，那我修改公共的就可以解决问题了。

我们在项目下创建名为api的包，下面创建login.py

# login.pyfrom base.http_client import HttpClient# 继承HttpClientclass Auth(HttpClient):def login(self, **kwargs):return self.get('http://127.0.0.1:5000/login', **kwargs)# 实例化Auth类auth = Auth()

那么，test_api.py就需要做下修改：

# test_api.pyimport pytestfrom api.login import authdef test_01():response = auth.login().json()print(response)assert response.get('success') == 'ok'if __name__ == '__main__':pytest.main()

运行测试成功。

由于某天登录出问题，报错了：

============================= test session starts =============================......def test_01():response = auth.login().json()> assert response.get('success') == 'ok'E AssertionError: assert 'ko' == 'ok'E - okE + kotest_api.py:7: AssertionError=========================== short test summary info ===========================FAILED test_api.py::test_01 - AssertionError: assert 'ko' == 'ok'============================== 1 failed in 0.22s ==============================

我们需要定位问题，那就要拿到当时的请求明细，故而我们需要加入一个日志功能，记录发送的请求和接收的响应。

要想记录请 i 去和返回的信息，只需要在之前创建的http_client.py模块做修改就行。那么先封装一个日志工具。

在项目下创建包名为utils和logs目录，紧接着utils包下创建logger.py。

# utils/logger.pyimport osimport timeimport loggingimport json as _jsonroot_path = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))logger = logging.getLogger()logger.setLevel(logging.DEBUG)# 建立一个filehandler来把日志记录在文件里，级别为debug以上fh = logging.FileHandler("{}/logs/{}.log".format(root_path, time.strftime('%Y%m%d', time.localtime())))fh.setLevel(logging.DEBUG)# 建立一个streamhandler来把日志打在CMD窗口上，级别为error以上ch = logging.StreamHandler()ch.setLevel(logging.ERROR)# 设置日志格式formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")ch.setFormatter(formatter)fh.setFormatter(formatter)# 将相应的handler添加在logger对象中logger.addHandler(ch)logger.addHandler(fh)

接着修改http_client.py

import json as _jsonimport requestsfrom utils.logger import api_loggerclass HttpClient():# 接口做post一般都是采用json格式进行提交__headers = {"content-type": "application/json;charset=UTF-8"}def __init__(self):self.__session = requests.session()def get(self, path, **kwargs):return self.__request(path, 'GET', **kwargs)def post(self, path, data=None, json=None, **kwargs):return self.__request(path, 'POST', data, json, **kwargs)def __request(self, url, method, data=None, json=None, **kwargs):headers = kwargs.get("headers")params = kwargs.get("params")# 如果传入header不为空，那么更新headerif headers:self.__headers.update(headers)self.__request_log(url, method, data, json, params, self.__headers)resp = Noneif method == "GET":resp = self.__session.get(url, **kwargs)elif method == "POST":resp = requests.post(url, data, json, **kwargs)self.__response_log(resp)return respdef __request_log(self, url, method, data=None, json=None, params=None, headers=None):api_logger.info("接口请求地址: {}".format(url))api_logger.info("接口请求方式: {}".format(method))api_logger.info("接口请求头: {}".format(_json.dumps(headers, indent=4, ensure_ascii=False)))api_logger.info("接口请求 params 参数: {}".format(_json.dumps(params, indent=4, ensure_ascii=False)))api_logger.info("接口请求体 data 参数 : {}".format(_json.dumps(data, indent=4, ensure_ascii=False)))api_logger.info("接口请求体 json 参数: {}".format(_json.dumps(json, indent=4, ensure_ascii=False)))def __response_log(self, resp):try:api_logger.info("返回信息 : {}".format(resp.text, ensure_ascii=False))except Exception as e:api_logger.error('系统错误：{}'.format(e))

我们再次执行测试，查看 logs 文件下的日志，可以看到记录了接口测试过程。

实际工作中的自动化场景复杂程度远远不止如此。

公共方法可能在多个场景被使用，那么就需要在抽离出一个业务层接口入参也需要单独抽离，方便统一维护用例层抽离，方便统一管理

yml 文件读取功能

直接创建：data目录作为数据层，operations包作为业务层，testcases作为用例层。

数据我们采用yaml格式，封装 yaml 文件读取功能，在utils包下创建read_data.py

# utils/read_data.pyimport osimport yamlroot_path = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))class ReadFileData:def load_yaml(self, file_path):try:data_file_path = os.path.join(root_path, "data", file_path)with open(data_file_path, encoding='utf-8') as f:data = yaml.safe_load(f)except Exception as ex:print(ex)else:return datadata = ReadFileData()

数据层

data下创建login.yml文件用于存储登录的用户名和密码

# data/login.ymllogin_data:json:username: 'admin'password: '123456'

不要忘记在api包下新增我们登录接口如下：

# api/login.pyfrom base.http_client import HttpClient# 继承HttpClientclass Auth(HttpClient):# ..此处省略..def token(self, **kwargs):return self.post('http://127.0.0.1:5000/user/token', **kwargs)auth = Auth()

业务层

在operaitons包下创建auth.py文件

from api.auth import authfrom utils.read_data import data# 读取测试数据data = data.load_yaml('login.yml')def get_token():test_case = data['login_data']# 由于yml文件读取出来是字典形式，可以直接用关键字传参方式调用api层的接口方法return auth.token(**test_case)

测试用例层

把test_api.py移动到testcases下，新增登录测试用例：

# testcases/test_api.pyimport pytestfrom api.login import authfrom operations.auth import get_tokendef test_01():response = auth.login().json()assert response.get('success') == 'ok'# 新增登录测试用例def test_token():resp = get_token().json()assert resp.get('msg') == '登录成功'if __name__ == '__main__':pytest.main()

执行用例成功

============================= test session starts =============================platform win32 -- Python 3.7.1, pytest-6.2.1, py-1.10.0, pluggy-0.13.1rootdir: D:\study\api-testplugins: allure-pytest-2.8.31, dependency-0.5.1, forked-1.3.0, ordering-0.6, rerunfailures-9.1.1, xdist-2.2.1collected 2 itemstestcases\test_api.py .. [100%]============================== 2 passed in 0.81s ==============================

至此，我们自动化框架基本完成了，更多功能还需要读者自己去扩展和补充。

福利

福利大放送，从入门到实战，从电子书到面试真题，这些资料，对于【软件测试】的朋友来说应该是最全面最完整的备战仓库，这个仓库也陪伴我走过了最艰难的路程，希望也能帮助到你。

如果需要这份学习资料，可以微信扫描下方CSDN官方认证公众号【100%免费获取】。