做接口测试的时候，避免不了操作数据库。因为数据校验需要，测试数据初始化需要、一些参数化场景需要等。

数据库操作框架设计

这里主要操作mysql数据库，整体思路：

封装实现

具体代码实现：

　　import pymysql

　　import json

　　

　　

　　class OperateMysql(object):

　　 def __init__(self):

　　 # 数据库初始化连接

　　 self.connect_interface_testing = pymysql.connect(

　　 "localhost",

　　 "root",

　　 "123456",

　　 "test",

　　 charset='utf8mb4',

　　 cursorclass=pymysql.cursors.DictCursor

　　 )

　　

　　 # 创建游标操作数据库

　　 self.cursor_interface_testing = self.connect_interface_testing.cursor()

　　

　　 def select_first_data(self, sql):

　　 """

　　 查询第一条数据

　　 """

　　 try:

　　 # 执行 sql 语句

　　 self.cursor_interface_testing.execute(sql)

　　 except Exception as e:

　　 print("执行sql异常:%s"%e)

　　 else:

　　 # 获取查询到的第一条数据

　　 first_data = self.cursor_interface_testing.fetchone()

　　 # print(first_data)

　　 # 将返回结果转换成 str 数据格式，禁用acsii编码

　　 first_data = json.dumps(first_data,ensure_ascii=False)

　　 # self.connect_interface_testing.close()

　　 return first_data

　　

　　 def select_all_data(self,sql):

　　 """

　　 查询结果集

　　 """

　　 try:

　　 self.cursor_interface_testing.execute(sql)

　　 except Exception as e:

　　 print("执行sql异常:%s"%e)

　　 else:

　　 first_data = self.cursor_interface_testing.fetchall()

　　 first_data = json.dumps(first_data,ensure_ascii=False)

　　 # self.connect_interface_testing.close()

　　 return first_data

　　

　　 def del_data(self,sql):

　　 """

　　 删除数据

　　 """

　　 res = {}

　　 try:

　　 # 执行SQL语句

　　 result = self.cursor_interface_testing.execute(sql)

　　 # print(result)

　　 if result != 0:

　　 # 提交修改

　　 self.connect_interface_testing.commit()

　　 res = {'删除成功'}

　　 else:

　　 res = {'没有要删除的数据'}

　　 except:

　　 # 发生错误时回滚

　　 self.connect_interface_testing.rollback()

　　 res = {'删除失败'}

　　 return res

　　

　　 def update_data(self,sql):

　　 """

　　 修改数据

　　 """

　　 try:

　　 self.cursor_interface_testing.execute(sql)

　　 self.connect_interface_testing.commit()

　　 res = {'更新成功'}

　　 except Exception as e:

　　 self.connect_interface_testing.rollback()

　　 res = {'更新删除'}

　　 return res

　　

　　 def insert_data(self,sql,data):

　　 """

　　 新增数据

　　 """

　　

　　 try:

　　 self.cursor_interface_testing.execute(sql,data)

　　 self.connect_interface_testing.commit()

　　 res = {data,'新增成功'}

　　 except Exception as e:

　　 res = {'新增失败',e}

　　 return res

　　 def conn_close(self):

　　 # 关闭数据库

　　 self.cursor_interface_testing.close()

　　

　　

　　if __name__ == "__main__":

　　 # ()类的实例化

　　 om = OperateMysql()

　　

　　 # 新增

　　 data = [{'id': 1, 'name': '测试', 'age': 15}, {'id': 2, 'name': '老王', 'age': 10}, {'id': 3, 'name': '李四', 'age': 20}]

　　 for i in data:

　　 i_data = (i['id'],i['name'],i['age'])

　　 insert_res = om.insert_data(

　　 """

　　 INSERT INTO test_student (id,name,age) VALUES (%s,%s,%s)

　　 """,i_data

　　 )

　　 print(insert_res)

　　

　　 # 查询

　　 one_data = om.select_first_data(

　　 """

　　 SELECT * FROM test_student;

　　 """

　　 )

　　 all_data = om.select_all_data(

　　 """

　　 SELECT * FROM test_student;

　　 """

　　 )

　　 print(one_data)

　　 # all_data字符串类型的list转list

　　 print("查询总数据:%s",len(json.loads(all_data)),"分别是:%s",all_data)

　　

　　 # 修改

　　 update_data = om.update_data(

　　 """

　　 UPDATE test_student SET name = '王五' WHERE id = 1;

　　 """

　　 )

　　 print(update_data)

　　

　　 # 删除

　　 del_data = om.del_data(

　　 """

　　 DELETE FROM test_student WHERE id in (1,2,3);

　　 """

　　 )

　　 print(del_data)

　　

　　 # 关闭游标

　　 om.conn_close()

运行结果：

为了方便演示，先注释删除数据的sql，再执行程序：

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

​​​软件测试面试文档

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