【Apifox】调用python脚本查询数据库并断言接口返回值是否满足条件

原创

已于 2023-04-18 15:59:02 修改 · 1.8k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#sql #测试工具 #学习 #python

于 2023-04-18 15:36:28 首次发布

文章讲述了在Apifox中通过调用Python脚本进行接口测试和数据库查询的场景。作者因接口查询的SQL语句需包含前一个接口的userId，而Apifox内无法直接处理，故采用Python脚本动态构建SQL并执行，再将结果与接口返回值做断言比较，确保数据一致性。

记录最近一次使用apifox调试接口，断言需要对比数据库与接口返回值是否一致的处理情况
本来没想用python脚本，但是由于查询的sql语句中，有一个userId是从前一个接口获取的，基于不想把自动化脚本写死的前提，发现apifox里面调用数据库操作时，好像没有可以调用变量的位置（也可能是我没发现），为了快速解决我遇到的问题，我只能曲线救国，调用python脚本，将上个接口返回的userId以传参的方式传入python脚本中，然后在脚本中实现数据库连接和查询。
第一步，编写python脚本：

import argparse
import pymysql

def test_auth(userId):
    conn = pymysql.connect(
    host='127.01.01.01',
    port=3306,
    user='root',
    passwd='******', # password也可以
    db='test',
    charset='utf8') # 如果查询有中文需要指定数据库编码
    # 2. 从连接建立游标（有了游标才能操作数据库）
    cur = conn.cursor()
    #简单粗暴的拼接
    sql1 ="SELECT COUNT(DISTINCT B.EMP_IDCARD) FROM `test`.`t_employee_project` B LEFT JOIN `test`.`t_employee_info` A