python接口测试4-数据库获取参数

首先确定需要传递的参数和接口，使用接口测试工具验证一下，接口和参数没有问题。

编写python接口脚本

import requests
import unittest
import json
import pymysql

conn = pymysql.connect(host='10.10.100.224',port=3840,user='kexin_admin',passwd='kexin@admin!',db='iot')
cursor = conn.cursor()
sql = "SELECT device_num FROM emergency  WHERE device_name = 'test2'"
cursor.execute(sql)
num = cursor.fetchone()#从数据库拿到想要的参数
print(num)

class Test_run(unittest.TestCase):
    def test_run_status(self):
        url = 'http://10.10.100.224:10001/api/emergencyService/doorlock/getBatterStatus'
        payload = {"deviceNum":num}

        re = requests.post(url,data=payload)
        da = json.loads(re.text)
        code = da['code']
        self.assertEqual(code,1)
        print(da)

if __name__ == "__main__":
    unittest.main()

运行结果：和接口测试工具的结果一致。