在go-sqlmock中模拟多条件查询的最佳实践

在使用go-sqlmock进行数据库单元测试时，经常会遇到需要模拟不同查询条件返回不同结果的场景。本文深入探讨如何高效地使用go-sqlmock来模拟多条件查询的情况。

理解ExpectQuery的工作原理

go-sqlmock中的ExpectQuery方法用于设置对Query或QueryRow调用的预期。每次调用ExpectQuery都会创建一个新的预期对象，这个对象只能对应一次数据库查询操作。这是设计上的有意为之，因为每次数据库查询都是独立的操作。

多条件查询模拟方案

对于需要模拟不同查询参数返回不同结果的场景，正确的做法是为每个查询条件单独设置预期：

query := "SELECT (.+) FROM products WHERE (.+)"

// 模拟ID为1的查询
mock.ExpectQuery(query).
    WithArgs("1").
    WillReturnRows(sqlmock.NewRows(columns).FromCSVString("1,USD,100"))

// 模拟ID为2的查询(空结果)
mock.ExpectQuery(query).
    WithArgs("2").
    WillReturnRows(emptyRows)

// 模拟ID为3的查询
mock.ExpectQuery(query).
    WithArgs("3").
    WillReturnRows(sqlmock.NewRows(columns).FromCSVString("3,USD,200"))

实际应用建议

1. 查询模式重用：将常用的查询模式定义为常量或变量，便于维护和修改

2. 执行顺序验证：go-sqlmock会按照ExpectQuery的调用顺序验证实际查询的执行顺序

3. 参数匹配：WithArgs支持精确匹配参数值，确保测试的准确性

4. 结果集构建：使用FromCSVString可以快速构建测试数据行

常见误区

许多开发者误以为可以通过单个ExpectQuery调用来模拟多种查询情况，这是不正确的。每个查询预期都需要独立的ExpectQuery调用，即使它们使用相同的SQL模式。

通过正确使用go-sqlmock的多查询预期设置，可以构建出精确模拟各种数据库查询场景的测试用例，确保业务逻辑在各种数据情况下都能正确工作。