如何最有效地清空测试套件中的数据库记录？

积累知识，胜过积蓄金银！毕竟在Golang开发的过程中，会遇到各种各样的问题，往往都是一些细节知识点还没有掌握好而导致的，因此基础知识点的积累是很重要的。下面本文《如何最有效地清空测试套件中的数据库记录？》，就带大家讲解一下知识点，若是你对本文感兴趣，或者是想搞懂其中某个知识点，就请你继续往下看吧~

我有一个测试套件，它使用从 yaml 文件读取的种子来污染我的数据库。

我想知道在运行测试后是否有办法清理我的数据库（删除用于测试套件的所有记录）。

// Open db and returns pointer and closer func
func prepareMySQLDB(t *testing.T) (db *sql.DB, closer func() error) {
    db, err := sql.Open("mysql", "user:pass@/database")
    if err != nil {
        t.Fatalf("open mysql connection: %s", err)
    }
    return db, db.Close
}
// Pollute my database
func polluteDb(db *sql.DB, t *testing.T) {
    seed, err := os.Open("seed.yml")
    if err != nil {
        t.Fatalf("failed to open seed file: %s", err)
    }
    defer seed.Close()
    p := polluter.New(polluter.MySQLEngine(db))
    if err := p.Pollute(seed); err != nil {
        t.Fatalf("failed to pollute: %s", err)
    }
}

func TestAllUsers(t *testing.T) {
    t.Parallel()

    db, closeDb := prepareMySQLDB(t)
    defer closeDb()

    polluteDb(db, t)

    users, err := AllUsersD(db)
    if err != nil {
        t.Fatal("AllUsers() failed")
    }

    got := users[0].Email
    if got != "[email protected]" {
        t.Errorf("AllUsers().Email = %s; want [email protected]", got)
    }

    got1 := len(users)
    if got1 != 1 {
        t.Errorf("len(AllUsers()) = %d; want 1", got1)
    }
}


// Test I'm interested in
func TestAddUser(t *testing.T) {
    t.Parallel()

    db, closeDb := prepareMySQLDB(t)
    defer closeDb()

    polluteDb(db, t)

    user, err := AddUser(...)
    if err != nil {
        t.Fatal("AddUser() failed")
    }

    //how can I clean my database after this? 
}

我应该检索在 testadduser() 中插入的最后一个 id 并手动删除该行，还是有其他方法来保存我的数据库状态并在之后检索它？

正如我所说，我是 go 新手，因此非常感谢对我的代码或其他内容的任何其他评论。

解决方案

最好的方法通常是使用事务，然后回滚，因此它们从一开始就不会被提交。

github.com/DATA-DOG/go-txdb 包对此有很大帮助。

最终代码：

import (
    "database/sql"
    "os"
    "testing"

    txdb "github.com/DATA-DOG/go-txdb"

    "github.com/romanyx/polluter"
)

//mostly sql tests
func init() {
    txdb.Register("txdb", "mysql", "root:root@/betell_rest")
}

func TestAddUser(t *testing.T) {
    db, err := sql.Open("txdb", "root:root@/betell_rest")
    if err != nil {
        t.Fatal(err)
    }
    defer db.Close()

    users, _ := AllUsers(db)
    userscount := len(users)
    err = AddUser(db, "[email protected]", "pass")
    if err != nil {
        t.Fatal("AddUser() failed")
    }

    users, _ = AllUsers(db)
    if (userscount + 1) != len(users) {
        t.Fatal("AddUser() failed to write in database")
    }
}

注意：您也可以将 db 传递到您的污染器中，这样您就不会影响您的数据库。

以上就是《如何最有效地清空测试套件中的数据库记录？》的详细内容，更多关于的资料请关注golang学习网公众号！