使用接口模拟数据库返回类型

Question

如何正确模拟一些*sql.DB 使用接口进行单元测试的方法？例如，我想要一个 Store 结构，用作我的处理程序的数据源。它有一个conn 字段，即*sql.DB。但是在我的测试中，我想模拟数据库，所以我尝试返回的不是实际的*sql.Rows，而是*sql.Rows 应该满足的接口。代码如下：

type TestRows interface {
    Scan ()
}

type TestDB interface {
    Query (query string, args ...interface{}) (TestRows, error) 
}

type Rows struct {
    //..
}

func (r *Rows) Scan () {
    fmt.Println("Scanning")
}

这是*sql.DB

type DB struct {
    //...
}

func (d *DB) Query (query string, args ...inteface{}) (*Rows, error) {
    return &Rows{/* .... */}, nil
}

当我想实例化 Store 时，它会抛出错误

type Store struct {
    conn TestDB
}

func main() { 
    cl := Store {conn: &DB{}}
    fmt.Println("Hello, playground")
}

cannot use DB literal (type *DB) as type TestDB in field value:
    *DB does not implement TestDB (wrong type for Query method)
        have Query(string) (*Rows, error)
        want Query(string) (TestRows, error)

Answer 1

我建议模拟数据库是个坏主意。您可以验证是否使用特定的 SQL 字符串调用查询并返回虚假结果，但这确实没有用。这些测试几乎没有增加价值，并且会产生错误的信心。您的测试接受的内容与您的数据库中实际运行的内容之间绝对没有联系。

真正的危险之一是您的数据库架构可能会更改，但您的模拟不会更新。然后您的代码将在生产中失败但通过测试。

相反，实现某种数据抽象层，例如就像Data Access Object (DAO)。

您可以编写自动集成测试来针对您的数据库测试您的 DAO。这将确保您的 DAO 实际工作并捕获任何错误，例如架构更改后查询未更新。

然后您可以为您的其他代码编写单元测试。任何与数据库交互的东西都应该使用模拟/存根/假 DAO，具体取决于测试的要求。

Answer 2

Query函数的定义需要是：

func (d *DB) Query(query string, args ...interface{}) (TestRows, error) {
    return &Rows{count: 1}, nil
}

编辑：

如果如——你指出的——你不能改变Query函数的实现，因为你需要满足其他接口，那么你需要调整你自己的接口。在这种情况下

type TestDB interface {
    Query(query string, args ...interface{}) (*Rows, error)
}