Golang单元测试配置文件路径怎么设置？

joker.zeng

已于 2024-06-19 15:19:32 修改

阅读量1.7k

点赞数 1

文章标签： golang 单元测试 log4j

于 2023-07-12 15:26:50 首次发布

本文链接：https://blog.youkuaiyun.com/qq_39272466/article/details/131682554

版权

文章描述了一个在Go语言项目中处理单元测试时配置文件路径错误的方法。通过获取`config.go`的绝对路径，确定项目根目录，动态调整工作目录以适应单元测试和正常运行。使用`godotenv`加载`.env`文件，并根据环境变量加载相应的yaml配置文件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

你们是不是单元测试的时候会报配置文件路径错误

在这里插入图片描述

这个错误其实很好解决，找到引入配置文件这一块代码做出如下修改

核心思想就是获取config.go的绝对路径然后推演出项目根目录

然后就是根据绝对路径引入配置文件了，不管单元测试还是正常运行都可以


/**
 * 读取配置文件,最先被初始化
 */
func init() {
	// 获取当前工作目录
	rootDir, err := os.Getwd()
	if err != nil {
		log.Fatalf("Failed to get current working directory: %s", err)
	}

	// 这是单元测试
	if strings.HasSuffix(rootDir, "service") {
		rootDir = filepath.Dir(rootDir)
	}

	// 设置工作目录
	if err := os.Chdir(rootDir); err != nil {
		log.Fatalf("Failed to change working directory: %s", err)
	}
	fmt.Println("工作目录：", rootDir)

	// 加载.env
	godotenv.Load(".env")

	// 设置当前环境
	mode := os.Getenv("Mode")
	fmt.Println("当前环境：", mode)
	if mode != "" {
		mode = "-" + mode
	}

	// 加载配置文件
	configPath := rootDir + "/config/application" + mode + ".yaml"
	yamlFile, err := ioutil.ReadFile(configPath)
	if err != nil {
		log.Panicf("Failed to read file , cause is %s", err.Error())
	}
	config := config.Config{}
	err = yaml.Unmarshal(yamlFile, &config)
	global.CONFIG = config
	if err != nil {
		fmt.Println(err.Error())
	}
}