Question

我有一个测试套件，该套件使用从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的新手，因此强烈欢迎您对我的代码进行任何其他注释。

Answer 1

最好的方法通常是先使用事务，然后再使用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传递给污染者，以使您完全不影响数据库。

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

1 个答案: