数据库连接get的单元测试策略

一分耕耘，一分收获！既然都打开这篇《数据库连接get的单元测试策略》，就坚持看下去，学下去吧！本文主要会给大家讲到等等知识点，如果大家对本文有好的建议或者看到有不足之处，非常欢迎大家积极提出！在后续文章我会继续更新Golang相关的内容，希望对大家都有所帮助！

正在测试的代码中有一个方法，它只是尝试获取数据库连接，如果无法连接则返回错误。

它以及涉及的结构体定义如下：

type databasecontext struct {
    context
    database databasest
}


// //getinfo returns the context.
// func (c *databasecontext) getinfo() *context {
//  return &c.context
// }

//getdb gets the database connection from the connection string.
func (c *databasecontext) getdb() (*sql.db, *errors.errorst) {
    var errst *errors.errorst
    if c.database.dbconnection == nil {
        c.database.dbconnection, errst = c.opendb()
        if errst != nil {
            return nil, errst
        }
        c.database.dbconnection.setmaxopenconns(50)
    }
    return c.database.dbconnection, nil
}

同一文件中可能遇到的其他方法如下：

//openDB opens the database with the connection string.
func (c *DatabaseContext) openDB() (*sql.DB, *errors.ErrorSt) {
    if c.Database.DBConnectionStr == "" {
        c.GetDatabase()
    }
    return db.OpenConnection(c.Database.DBConnectionStr, c.Database.InterpolateParams)

}

//CloseDB Closes the database.
func (c *DatabaseContext) CloseDB() {
    if c.Database.dbConnection != nil {
        c.Database.dbConnection.Close()
    }
}

//SetDatabaseString Sets the database string into the session.
func (c *DatabaseContext) SetDatabaseString(str string) {
    c.Database.DBConnectionStr = str
    i := strings.Index(str, ")/") + 2
    c.Database.DBName = str[i:]
    c.SetDatabase()
}

//GetDatabaseString Gets the database string from the session.
func (c *DatabaseContext) GetDatabase() {
    if dbIntf := c.GetFromSession("Database"); dbIntf != nil {
        c.Database = dbIntf.(DatabaseSt)
    }
}

//SetDatabaseString Sets the database string into the session.
func (c *DatabaseContext) SetDatabase() {
    c.SetToSession("Database", c.Database)
}

幸运的是，databasecontext 实现了 databasecontextintf，我想用它来进行测试。我的本能是直接模拟 databasecontext，但这行不通，因为它不是一个接口（在 golang 中，你只能模拟接口）。

我将如何在不访问真实数据库的情况下进行测试，因为真实数据库可能会超出我的控制范围（从而在测试中创建错误的失败）？

更新我的问题与可疑的重复问题不同，因为他们的问题是关于数据库条目，而不是连接。标记的重复项引用该库作为答案，但是，为了测试的目的，其中没有方法返回为零的“连接”。它最好的是 new ，它创建一个测试双连接，并且无法控制返回值的状态（我需要它在一个测试中为 nil （“无连接”），而在另一个测试中为非零（“理智”）测试”))

解决方案

我最终使测试的包与被测代码的包相同（这允许 visual studio code 中的测试生成器将生成的测试放在测试文件中，而不会感到困惑，以及让我访问我使用的未导出字段），然后直接制作了一个假的 databasecontext

我的测试用例如下所示：

t.Run("SanityTest", func(t *testing.T) {
    c := new(DatabaseContext)
    assert.Nil(t, c.Database.dbConnection)
    database, err := c.GetDB()
    defer database.Close()
    assert.NotNil(t, database)
    if !assert.Nil(t, err) {
        t.Error(err.ToString(false))
    }
})

今天关于《数据库连接get的单元测试策略》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！