通用的UI自动化测试框架(3)----数据库操作封装

介绍

    在上一篇文章链接🔗里我们讲解了如今将一些常用的公共方法进行封装，但是我们平时的一些UI自动化操作会需要对后代数据库进行操作，如造数据或修改数据去完成一些UI的操作。

    这篇文章我会举例子，在UI自动化框架里，如何对一些数据库的操作进行封装，减少代码的冗余度。

正文

    我们这里的例子将是使用MySQL,如果是其他类型数据库，具体操作也差不多，原理都是一样。如下：

import pymysql

class operate_mysql():

    def __init__(self):

        self.connect = pymysql.connect(host='xx.xx.xxx.xxx',port=3306,user='XXX',password='YYY',db='123',autocommit=True)

    def sql_operate(self,name):

    #参数可以自己定义，name为举例参数

    #print('数据库连接成功')

    cursor = self.connect.cursor()

    #创建游标

    sql = "UPDATE table1 SET 'age' = '%d' WHERE name = '%s'"%(20,name)

    #需要执行的sql

    try:

        cursor.execute(sql)

        #print(1)

        self.connect.commit()

        #print(2)

    except:

        #print('执行失败，回滚')

        self.connect.rollback()

    finally:

        cursor.close()

        self.connect.close()

if __name__ =='__main__':

    OM = operate_mysql()

    name = '小王'

    OM.sql_operate(name)

    这个简单例子是把数据库的连接和sql的执行封装成一个类。