用测试驱动开发验证数据库查询方法

哈喽！今天心血来潮给大家带来了《用测试驱动开发验证数据库查询方法》，想必大家应该对Golang都不陌生吧，那么阅读本文就都不会很困难，以下内容主要涉及到，若是你正在学习Golang，千万别错过这篇文章~希望能帮助到你！

我想使用 Golang 创建一个数据库驱动的应用程序。我正在尝试以 TDD 方式做到这一点。 当我尝试测试进行 Sql 查询的方法时，可用的包有哪些？

• 我不想连接到用于开发的默认数据库。我可以编写代码在运行测试时占用另一个测试数据库，但是是否有任何 go 库已经做到了这一点。

• 是否有任何库可以在不连接数据库的情况下进行数据库测试？

使用 golang 进行数据库测试的标准方法是什么？

解决方案

不久前，我在重构自己的一些测试时遇到了类似的问题，有几种方法可以做到这一点：

a) 提供导出类型和返回它的 open 或 connect 函数 - 例如

type db struct {
    db *sql.db
}

// using http://jmoiron.github.io/sqlx/ for this example, but
// it has the same interface as database/sql
func open(opts *options) (*db, error) {
    db, err := sqlx.connect(opts.driver, fmt.sprintf("host=%s user=%s dbname=%s sslmode=%s", opts.host, opts.user, opts.name, opts.ssl))
    if err != nil {
        return nil, err
    }

    return &db{db}, nil
}

...然后每个测试，编写设置和拆卸函数，这些函数返回您在其上定义数据库函数的 *db 实例（作为方法 - 即 func (db *db ）getuser（用户*用户）（布尔，错误））：

// setup the test environment.
func setup() (*db, error) {
    err := withtestdb()
    if err != nil {
        return nil, err
    }

    // testoptions is a global in this case, but you could easily
    // create one per-test
    db, err := open(testoptions)
    if err != nil {
        return nil, err
    }

    // loads our test schema
    db.mustload()
    return db, nil
}

// create our test database.
func withtestdb() error {
    db, err := open()
    if err != nil {
        return err
    }
    defer db.close()

    _, err = db.exec(fmt.sprintf("create database %s;", testoptions.name))
    if err != nil {
        return err
    }

    return nil
}

请注意，这有点“集成”测试，但我强烈喜欢针对“真实”数据库进行测试，因为模拟接口不会帮助您捕获查询/查询语法的问题。

b) 另一种方法是在测试中使用 init() 初始化的全局 db *sql.db 变量，因为测试没有保证顺序，因此您需要使用 init()，然后从那里运行测试。即

var db *sql.DB

func init() {
    var err error
    // Note the = and *not* the assignment - we don't want to shadow our global
    db, err = sqlx.Connect(...)
    if err != nil {
        ...
    }

    err := db.loadTestSchema
    // etc.
}

func TestGetUser(t *testing.T) {
   user := User{}
   exists, err := db.GetUser(user)
   ...
}

你可以在drone.io's GitHub repo找到一些实际的例子，我也推荐this article on structuring Go applications（尤其是db的东西）。

我使用一个全局变量来存储当前的数据源（或连接字符串）数据库并在测试函数中设置为不同的值。由于我只需要操作一个数据库，所以我选择了最简单的方法。

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~