Go语言TestMain初始化全局db后测试函数中db为nil的解决方法

你遇到的问题核心是变量作用域的混淆——在TestMain里你用了短变量声明:=，这会创建一个局部的db变量，完全覆盖了全局的db，导致全局变量根本没被赋值，始终是初始的nil。

修复方案 #

把TestMain里的短变量声明改成对全局变量的赋值操作，同时单独处理err变量：

var db *sql.DB
func TestMain(m *testing.M) {
    var err error // 单独声明err变量
    db, err = sql.Open("mysql", os.Getenv("DSN")) // 直接给全局db赋值，不用:=
    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, initialization failed")
    }
}

为什么会出现这个问题？ #

Go语言里的:=是短变量声明运算符，它的作用是在当前函数作用域内创建新的变量（如果该作用域中没有同名变量的话）。在你的原始代码里，db, err := sql.Open(...)这行代码，会在TestMain函数内部创建一个局部的db变量，这个变量和全局的db是完全独立的两个变量。

当TestMain执行时，你操作的都是这个局部db，全局的db从来没被修改过，依然保持初始的nil值。所以当Test_getRole执行时，它访问的全局db自然还是nil。

简单来说：局部作用域的变量会屏蔽同名的全局变量，短变量声明只会在当前作用域创建新变量，不会修改全局变量。

内容的提问来源于stack exchange，提问作者hendry