简单几步教你学会Python接口自动化测试

Hacker_Fuchen

403人浏览 · 2026-03-14 20:10:24

Hacker_Fuchen · 2026-03-14 20:10:24 发布

一、简介

本文从一个简单的登录接口测试入手，一步步调整优化接口调用姿势；

然后简单讨论了一下接口测试框架的要点；

最后介绍了一下我们目前正在使用的接口测试框架 pithy。

期望读者可以通过本文对接口自动化测试有一个大致的了解。

二、引言

为什么要做接口自动化测试？

在当前互联网产品迭代频繁的背景下，回归测试的时间越来越少，很难在每个迭代都对所有功能做完整回归。

但接口自动化测试因其实现简单、维护成本低，容易提高覆盖率等特点，越来越受重视。

为什么要自己写框架呢？

使用 requets + unittest 很容易实现接口自动化测试，而且 requests 的api 已经非常人性化，非常简单。

但通过封装以后（特别是针对公司内特定接口），再加上对一些常用工具的封装，可以进一步提高业务脚本编写效率。

三、环境准备

确保本机已安装 python2.7 以上版本，然后安装如下库：

1	`pip` `install` `flaskpip` `install` `requests`

后面我们会使用 flask 写一个用来测试的接口，使用requests去测试。

四、测试接口准备

下面使用 flask 实现两个 http 接口，一个登录，另外一个查询详情，但需要登录后才可以，新建一个 demo.py 文件（注意，不要使用windows记事本），把下面代码 copy 进去，然后保存、关闭。

接口代码

#!/usr/bin/python# coding=utf-8from flask import Flask, request, session, jsonify

USERNAME = 'admin'PASSWORD = '123456'app = Flask(__name__)

app.secret_key = 'pithy'@app.route('/login', methods=['GET', 'POST'])def login():

error = None

if request.method == 'POST': if request.form['username'] != USERNAME:

error = 'Invalid username'

elif request.form['password'] != PASSWORD:

error = 'Invalid password'

else:

session['logged_in'] = True

return jsonify({'code': 200, 'msg': 'success'}) return jsonify({'code': 401, 'msg': error}), 401@app.route('/info', methods=['get'])def info():

if not session.get('logged_in'): return jsonify({'code': 401, 'msg': 'please login !!'}) return jsonify({'code': 200, 'msg': 'success', 'data': 'info'})if __name__ == '__main__':

app.run(debug=True)

最后执行如下命令：

1	`python demo.py`

响应如下：

* Running on http://127.0.0.1:5000/ (Press CTRL+C to quit)
* Restarting with sta

大家可以看到服务已经起来了。

接口信息

登录接口

请求url

/login

请求方法

post

请求参数

| 参数名称 | 参数类型 | 参数说明 |

| :————: | :——-: | :——: |

| username | String | 登录名称 |

| password | String | 登录密码 |

响应信息

| 参数名称 | 参数类型 | 参数说明 |

| :————: | :——-: | :——: |

| code | Integer | 结果code |

| msg | String | 结果信息 |

详情接口

请求url

/info

请求方法

get

请求 cookies

| 参数名称 | 参数类型 | 参数说明 |

| :————: | :——-: | :——: |

| session | String | session |

响应信息

| 参数名称 | 参数类型 | 参数说明 |

| :————: | :——-: | :——: |

| code | Integer | 结果code |

| msg | String | 结果信息 |

| data | String | 数据信息 |

五、编写接口测试

测试思路

使用 requests库模拟发送 HTTP 请求。
使用 python 标准库里 unittest 写测试 case。

脚本实现

#!/usr/bin/python# coding=utf-8import requestsimport unittestclass TestLogin(unittest.TestCase):

@classmethod

def setUpClass(cls):

cls.login_url = 'http://127.0.0.1:5000/login'

cls.info_url = 'http://127.0.0.1:5000/info'

cls.username = 'admin'

cls.password = '123456'

def test_login(self):

"""

测试登录

"""

data = { 'username': self.username, 'password': self.password

}

response = requests.post(self.login_url, data=data).json() assert response['code'] == 200

assert response['msg'] == 'success'

def test_info(self):

"""

测试info接口

"""

data = { 'username': self.username, 'password': self.password

}

response_cookies = requests.post(self.login_url, data=data).cookies

session = response_cookies.get('session') assert session

info_cookies = { 'session': session

}

response = requests.get(self.info_url, cookies=info_cookies).json() assert response['code'] == 200

assert response['msg'] == 'success'

assert response['data'] == 'info'

六、优化

封装接口调用

写完这个测试登录脚本，你或许会发现，在整个项目的测试过程，登录可能不止用到一次，如果每次都这么写，会不会太冗余了？

对，确实太冗余了，下面做一下简单的封装，把登录接口的调用封装到一个方法里，把调用参数暴漏出来，示例脚本如下：

#!/usr/bin/python# coding=utf-8import requestsimport unittesttry: from urlparse import urljoinexcept ImportError: from urllib.parse import urljoinclass DemoApi(object):

def __init__(self, base_url):

self.base_url = base_url def login(self, username, password):

"""

登录接口

:param username: 用户名

:param password: 密码

"""

url = urljoin(self.base_url, 'login')

data = { 'username': username, 'password': password

} return requests.post(url, data=data).json() def get_cookies(self, username, password):

"""

获取登录cookies

"""

url = urljoin(self.base_url, 'login')

data = { 'username': username, 'password': password

} return requests.post(url, data=data).cookies def info(self, cookies):

"""

详情接口

"""

url = urljoin(self.base_url, 'info') return requests.get(url, cookies=cookies).json()class TestLogin(unittest.TestCase):

@classmethod

def setUpClass(cls):

cls.base_url = 'http://127.0.0.1:5000'

cls.username = 'admin'

cls.password = '123456'

cls.app = DemoApi(cls.base_url) def test_login(self):

"""

测试登录

"""

response = self.app.login(self.username, self.password) assert response['code'] == 200

assert response['msg'] == 'success'

def test_info(self):

"""

测试获取详情信息

"""

cookies = self.app.get_cookies(self.username, self.password)

response = self.app.info(cookies) assert response['code'] == 200

assert response['msg'] == 'success'

assert response['data'] == 'info'

OK，在这一个版本中，我们不但在把登录接口的调用封装成了一个实例方法，实现了复用，而且还把 host（self.base_url）提取了出来。

但问题又来了，登录之后，登录接口的 http 响应会把 session 以 cookie 的形式 set 到客户端，之后的接口都会使用此 session 去请求。

还有，就是在接口调用过程中，希望可以把日志打印出来，以便调试或者出错时查看。

好吧，我们再来改一版。

保持 cookies &增加 log 信息

使用 requests 库里的同一个 Session 对象 (它也会在同一个 Session 实例发出的所有请求之间保持 cookie )，即可解决上面的问题，示例代码如下：

#!/usr/bin/python# coding=utf-8import unittestfrom pprint import pprintfrom requests.sessions import Sessiontry: from urlparse import urljoinexcept ImportError: from urllib.parse import urljoinclass DemoApi(object):

def __init__(self, base_url):

self.base_url = base_url # 创建session实例

self.session = Session() def login(self, username, password):

"""

登录接口

:param username: 用户名

:param password: 密码

"""

url = urljoin(self.base_url, 'login')

data = { 'username': username, 'password': password

}

response = self.session.post(url, data=data).json()

print('\n*****************************************')

print(u'\n1、请求url: \n%s' % url)

print(u'\n2、请求头信息:')

pprint(self.session.headers)

print(u'\n3、请求参数:')

pprint(data)

print(u'\n4、响应:')

pprint(response) return response def info(self):

"""

详情接口

"""

url = urljoin(self.base_url, 'info')

response = self.session.get(url).json()

print('\n*****************************************')

print(u'\n1、请求url: \n%s' % url)

print(u'\n2、请求头信息:')

pprint(self.session.headers)

print(u'\n3、请求cookies:')

pprint(dict(self.session.cookies))

print(u'\n4、响应:')

pprint(response) return responseclass TestLogin(unittest.TestCase):

@classmethod

def setUpClass(cls):

cls.base_url = 'http://127.0.0.1:5000'

cls.username = 'admin'

cls.password = '123456'

cls.app = DemoApi(cls.base_url) def test_login(self):

"""

测试登录

"""

response = self.app.login(self.username, self.password) assert response['code'] == 200

assert response['msg'] == 'success'

def test_info(self):

"""

测试获取详情信息

"""

self.app.login(self.username, self.password)

response = self.app.info() assert response['code'] == 200

assert response['msg'] == 'success'

assert response['data'] == 'info'

大功告成，我们把多个相关接口调用封装到一个类中，使用同一个 requests Session 实例来保持 cookies，并且在调用过程中打印出了日志，我们所有目标都实现了。

但再看下脚本，又会感觉不太舒服，在每个方法里，都要写一遍print 1、2、3… 要拼url、还要很多细节等等。

但其实我们真正需要做的只是拼出关键的参数（url 参数、body 参数或者传入 headers 信息），可不可以只需定义必须的信息，然后把其它共性的东西都封装起来呢，统一放到一个地方去管理？

最后： 下方这份完整的软件测试视频教程已经整理上传完成，需要的朋友们可以自行领取【保证100%免费】

软件测试面试文档

我们学习必然是为了找到高薪的工作，下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料，并且有字节大佬给出了权威的解答，刷完这一套面试资料相信大家都能找到满意的工作。

在这里插入图片描述

CSDN-OPC开发者社区

这里是“一人公司”的成长家园。我们提供从产品曝光、技术变现到法律财税的全栈内容，并连接云服务、办公空间等稀缺资源，助你专注创造，无忧运营。

更多推荐

DeepSeek API 最高涨 11 倍！一人公司 AI 团队如何把 Token 成本压回原价

CSDN-OPC开发者社区

企业级AI的权限治理：既要能干，又要安全

CSDN-OPC开发者社区

DeepSeek Harness：重新定义 AI 智能体的“操作系统“

CSDN-OPC开发者社区

所有评论(0)

查看更多评论

Hacker_Fuchen

@Hacker_Fuchen

已为社区贡献2条内容

简单几步教你学会Python接口自动化测试

Hacker_Fuchen

一、简介

二、引言

三、环境准备

四、测试接口准备

接口代码

五、编写接口测试

六、优化

所有评论(0)

温馨提示：您尚未绑定手机号

Hacker_Fuchen