Go语言：打造优雅数据库单元测试的实战指南

// init_test.go
package managers
import (
"testing"
"git.xxxxxx.com/xxx/xxx/config"
"git.xxxxxx.com/xxx/xxx//models"
_ "xxxx.com/xxx/mysql" // import your used driver
)
func TestMain(m *testing.M) {
config.GetConfig()
models.Init()
m.Run()
}

// conf.go
func getConfigFileName() string {
rootDir, err := os.Getwd()
if err != nil {
panic(err)
}
if strings.HasSuffix(rootDir, "services") {
rootDir = filepath.Dir(rootDir)
}
return rootDir + "/conf/config.yaml"
}
func GetConfig() *Config {
configFile := getConfigFileName()
bytes, err := ioutil.ReadFile(configFile)
if err != nil {
panic(fmt.Errorf("read config %s failed:%v", configFile, err))
}
config := Config{}
if err := yaml.Unmarshal(bytes, &config); err != nil {
panic(fmt.Errorf("parse config %s failed:%v", configFile, err))
}
return &config
}

// project_test.go
package managers


import (
"context"
"fmt"
"testing"
)


func TestProjectByID(t *testing.T) {
manager := NewProjectManager(context.TODO())
item, err := manager.ProjectByID(234)
if err != nil {
fmt.Println(err)
}
fmt.Println("item:", item)
}

func getConfigFileName() string {
rootDir, err := os.Getwd()
if err != nil {
panic(err)
}
if strings.HasSuffix(rootDir, "services") {
rootDir = filepath.Dir(rootDir)
}

return rootDir + "/conf/config.yaml"
}