时间戳

首页 > 解决方案 > 删除用于测试套件的所有数据库记录的最佳方法是什么？

sql - 删除用于测试套件的所有数据库记录的最佳方法是什么？

问题描述

我有一个测试套件，它使用从 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 != "myemail@gmail.com" {
        t.Errorf("AllUsers().Email = %s; want myemail@gmail.com", 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 的新手，因此对我的代码的任何其他评论或任何其他评论都非常感谢。

标签： sqlunit-testinggo

解决方案

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

该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, "bla@gmail.com", "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给您的污染者，这样您就不会影响您的数据库。

推荐阅读