在一个函数中定义的全局变量不会在另一个函数中保留定义

plicqrtu  于 2021-06-21  发布在  Mysql
关注(0)|答案(1)|浏览(424)

读完之后,我想我需要这个 TestMain(m *testing.M) 构造来设置我的数据库。然而,在运行测试时,db总是为零。我该怎么解决这个问题?

var db *sql.DB

func TestMain(m *testing.M) {

        db, err := sql.Open("mysql", os.Getenv("DSN"))
        if err != nil {
                log.Fatal("error opening database")
        }

        defer db.Close()
        log.Printf("here testing with %v", db)
        code := m.Run()
        log.Printf("finished test")
        os.Exit(code)

}

func Test_getRole(t *testing.T) {
        if db == nil {
                t.Fatalf("db is nil")
        }
}

输出为:

2018/05/02 19:10:14 here testing with &{{bugzilla:SECRET@tcp(example.com:3306)/bugzilla?multiStatements=true 0x7aba40} 0 {0 0} [] map[] 0 0 0xc42001e180 0xc4200740c0 false map[] map[] 0 0 0 <nil> 0x4e9850}
--- FAIL: Test_getRole (0.00s)
        main_test.go:32: db is nil

https://github.com/unee-t/processinvitations/blob/testfail/main_test.go#l43 是指向完整代码段的链接。

wfauudbj

wfauudbj1#

你的台词 db, err := sql.Open("mysql", os.Getenv("DSN")) 创建局部变量db(函数testmain的局部变量)。而是使用:

err := nil
db, err = sql.Open("mysql", os.Getenv("DSN"))

相关问题