三、封装接口层

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

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

# login.py

from base.http_client import HttpClient

# 继承HttpClient
class 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.py

import pytest
from api.login import auth


def 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         - ok
E         + ko

test_api.py:7: AssertionError
=========================== short test summary info ===========================
FAILED test_api.py::test_01 - AssertionError: assert 'ko' == 'ok'
============================== 1 failed in 0.22s ==============================

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