python+pytest写测试用例后置清理数据操作

一、teardown_function 函数是为了在每个测试函数def执行后进行数据清理。

# 引入 DbConnect 类或者确保它已经被定义
from your_db_module import DbConnect

def teardown_function():
　　try:
　　　　print("后置操作-做数据清理，把批注通知删掉")
　　　　db = DbConnect(db_info) # 使用给定的 db_info 建立数据库连接
　　　　sql = 'delete from gy_daily WHERE relation_id = 0 AND type = 3;' # 要执行的删除语句
　　　　de = db.execute_sql(sql) # 执行 SQL 删除语句
　　　　print("删除通知信息", de) # 打印删除结果
　　except Exception as e:
　　　　print("清理通知信息时出错:", str(e))

二、teardown_module 函数在测试模块结束后执行数据清理。

# 引入 ConnectDb 类或者确保它已经被定义
from your_db_module import ConnectDb

def teardown_module():
　　db = ConnectDb(db_info) # 使用给定的 db_info 建立数据库连接
　　delete_user_sql = 'DELETE FROM gy_seto WHERE company_name ="11";' # 要执行的删除语句
　　d = db.execute_sql(delete_user_sql) # 执行 SQL 删除语句
　　print(d) # 打印删除结果