一、数据库封装并在用例中使用
1、数据库封装
import pymysql
class DBHandler:
def __init__(self,
host='api.lemonban.com',
port=3306,
username='future',
password='123456',
db_name='futureloan'):
self.connection = pymysql.connect(host=host,
port=port,
user=username,
password=password,
db=db_name)
def query_one(self,sql):
cursor = self.connection.cursor()
self.connection.commit()
cursor.execute(sql)
data = cursor.fetchone()
cursor.close()
return data
def query_all(self,sql):
cursor = self.connection.cursor()
self.connection.commit()
cursor.execute(sql)
data = cursor.fetchall()
cursor.close()
return data
def close(self):
self.connection.close()
2、测试用例中使用(一般用于比较余额)
import unittest
from common.db_package import DBHandler
from ddt import