requests接口自动化-数据库参数化

Posted 2020-12-11 uni-hoang

• 数据库参数化的场景
  部分接口，参数其他方式获取不到，可以去数据库去获取，如手机验证码

• 安装pymysql
  
  

• 配置文件

def sql_conf():
    # 定义数据库的配置
    host=‘127.0.0.1‘
    user=‘root‘
    password=‘123456‘
    database=‘employees‘
    port=3306
    charset=‘utf8‘
    return host,user,password,database,port,charset

• 定义get_mysql方法

import pymysql
from config.conf import sql_conf  #导入参数配置定义的函数
def get_sql(sql):
    # 建立一个连接对象
    host,user,password,database,port,charset=sql_conf() #引用参数配置的值
    db=pymysql.connect(host=host,user=user,database=database,port=port,charset=charset)
    # 建立一个游标
    cursor=db.cursor()
    # 运行sql语句
    cursor.execute(sql)
    # 保存查询的结果
    data=cursor.fetchall()

    # 关闭游标
    cursor.close()
    # 关闭数据库连接
    db.close()
    # 返回数据
    return data

• 调用get_mysql方法获取id

import requests
from common.get_mysql import get_sql
def test_get_mysql():
    url=‘http//:127.0.0.1:8080‘
    id=get_sql("SELECT dept_no FROM departments WHERE dept_name =‘Finance‘") # 从数据库获取数据
    data={‘name‘:‘Finance‘,‘id‘:id}
    r=requests.post(url=url,data=data)
    print(r.status_code)
    print(r.text)