数据库连接、配置文件、日志封装

wzcssq

于 2021-11-15 19:12:36 发布

阅读量674

点赞数

分类专栏： python基础 python自动化测试-接口测试文章标签：数据库 sql database

本文链接：https://blog.youkuaiyun.com/wzcssq/article/details/121339616

版权

本文详细介绍了如何进行数据库连接的封装，包括在测试用例中的应用，以及配置文件和日志系统的封装。通过yaml文件管理配置，利用log封装实现日志记录，并在测试中整合使用这些封装技术。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、数据库封装并在用例中使用

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

最低0.47元/天解锁文章