首页 > 其他 > 详细

UI自动化测试常用操作函数(2)

时间:2020-12-14 10:29:12      阅读:35      评论:0      收藏:0      [点我收藏+]

在做断言的时候,有时候需要查询数据库

创建数据库连接的时候,可以将连接信息参数化

import pymysql
class Service:
    # 创建数据库连接
    @classmethod
    def ConnectMysql(cls,host,uname,pwd,database):
        db = pymysql.connect(host=host, user=uname, password=pwd, database=database)
        cur = db.cursor()
        return db, cur

    # 断开数据库连接
    @classmethod
    def connectOff(cls, db, cur):
        db.commit()
        cur.close()
        db.close()

    # mysql查询得到的结果函数,供调用,这里的sql参数是sql语句
    @classmethod
    def select(cls, sql):
        db, cur = cls.ConnectMysql()
        try:
            cur.execute(sql)
            result = cur.fetchall()
            return result
        except Exception as e:
            print(str(e))
        finally:
            cls.connectOff(db, cur)

 

UI自动化测试常用操作函数(2)

原文:https://www.cnblogs.com/qianjunjun/p/14131653.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!