使用MySQL事务时如何编写单元测试？是否需定义事务接口？

一、事务场景的单元测试写法 #

单元测试事务逻辑时，优先用内存数据库（如SQLite）或测试专用隔离数据库，既能保证测试真实性，又不会污染生产数据。具体步骤如下：

1. 改造原函数支持依赖注入
   如果原代码硬编码全局DB实例，改成参数传入形式，方便测试时替换为测试用实例：

func SomeFunc(db *gorm.DB) error {
    return db.Transaction(func(tx *gorm.DB) error {
        // 业务逻辑：查询、创建、删除操作
        tx.Select("xxx").Model(&Model{}).Where(...)
        tx.Create(&Model{XXX: xxx})
        tx.Delete(&Model{}, "xxx = ?", xxx)
        return nil
    })
}

2. 编写测试用例
   初始化内存DB，自动迁移表结构，调用目标函数后验证数据库状态：

func TestSomeFunc(t *testing.T) {
    // 初始化SQLite内存数据库
    testDB, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
    if err != nil {
        t.Fatalf("初始化测试DB失败: %v", err)
    }
    // 自动迁移对应表结构
    if err = testDB.AutoMigrate(&Model{}); err != nil {
        t.Fatalf("表迁移失败: %v", err)
    }

    // 调用待测试函数
    if err = SomeFunc(testDB); err != nil {
        t.Errorf("SomeFunc执行失败: %v", err)
    }

    // 验证事务执行结果：检查创建/删除的数据是否符合预期
    var createdCount int64
    testDB.Model(&Model{}).Where("xxx = ?", targetVal).Count(&createdCount)
    if createdCount != 1 {
        t.Errorf("期望创建1条数据，实际找到%d条", createdCount)
    }

    var deletedCount int64
    testDB.Model(&Model{}).Where("xxx = ?", deleteVal).Count(&deletedCount)
    if deletedCount != 0 {
        t.Errorf("期望删除数据，实际剩余%d条", deletedCount)
    }
}

二、是否需要为事务定义接口？ #

是否定义接口取决于业务需求和未来规划：

不需要定义接口的场景 #

如果业务逻辑深度依赖GORM特性，且短期内没有替换ORM的计划，没必要专门为测试定义接口。直接用测试数据库做真实测试，能更准确验证事务的ACID特性，避免mock带来的测试失真。

建议定义接口的场景 #

如果想降低业务逻辑与GORM的耦合度，或者需要在测试中完全隔离数据库操作（比如快速验证上层逻辑），可以定义事务执行接口：

1. 定义接口

type TxRunner interface {
    Transaction(f func(tx *gorm.DB) error) error
}

2. 改造业务函数依赖接口

func SomeFunc(runner TxRunner) error {
    return runner.Transaction(func(tx *gorm.DB) error {
        // 原业务逻辑不变
        tx.Select(xxx)
        tx.Create(xxx)
        tx.Delete(xxx)
        return nil
    })
}

3. Mock接口做单元测试
   使用mock框架（如gomock）或手动实现mock，验证事务是否被调用：

type MockTxRunner struct {
    mock.Mock
}

func (m *MockTxRunner) Transaction(f func(tx *gorm.DB) error) error {
    args := m.Called(f)
    return args.Error(0)
}

func TestSomeFuncWithMock(t *testing.T) {
    mockRunner := new(MockTxRunner)
    // 断言Transaction方法被调用，且返回nil
    mockRunner.On("Transaction", mock.Anything).Return(nil)

    err := SomeFunc(mockRunner)
    if err != nil {
        t.Errorf("SomeFunc执行失败: %v", err)
    }
    // 验证mock的调用是否符合预期
    mockRunner.AssertExpectations(t)
}

内容的提问来源于stack exchange，提问作者lee3164