一、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) # 打印删除结果