嘿，我懂你的困扰——main函数没被测试覆盖导致整体覆盖率只有50%对吧？其实Go的测试默认不会自动执行main函数，得我们主动去测试它。下面给你两种可行的方法，结合你的代码示例来讲解：

方法1：直接测试main函数（处理os.Exit问题） #

因为main函数执行完会调用os.Exit(0)，直接在测试里调用的话会导致测试进程提前退出，所以我们需要先替换os.Exit的行为，避免程序真的退出。修改你的main_test.go如下：

package main

import (
	"os"
	"testing"
)

func TestSum(t *testing.T) {
	total := Sum(5, 5)
	if total != 10 {
		t.Fail()
	}
}

func TestMainFunction(t *testing.T) {
	// 保存原始的os.Exit函数
	originalExit := os.Exit
	defer func() {
		// 测试结束后恢复原始函数
		os.Exit = originalExit
		// 捕获main函数触发的panic，避免测试崩溃
		if r := recover(); r != nil {
			if exitMsg, ok := r.(string); ok && exitMsg == "main exited" {
				t.Log("Main function executed successfully")
				return
			}
			// 如果是其他panic，重新抛出
			panic(r)
		}
	}()

	// 替换os.Exit为自定义函数，触发panic而不是真的退出
	os.Exit = func(code int) {
		if code != 0 {
			t.Errorf("Main function exited with non-zero code: %d", code)
		}
		panic("main exited")
	}

	// 调用main函数，此时会执行里面的Sum(5,5)
	main()
}

现在运行测试并查看覆盖率：

go test -coverprofile=coverage.out && go tool cover -html=coverage.out

你会看到main函数已经被完全覆盖了。

方法2：重构代码（更优雅的长期方案） #

如果你的main函数以后会变得复杂，建议把业务逻辑从main里抽离出来，这样既方便测试，也符合单一职责原则。比如修改你的main.go：

package main

func Sum(x int, y int) int { return x + y }

// 把main里的逻辑放到run函数中
func run() {
	Sum(5, 5)
}

func main() {
	run()
}

然后在测试里测试run函数：

func TestRun(t *testing.T) {
	run()
	// 这里如果有需要可以添加断言，比如验证Sum的调用情况（可以用mock库）
}

不过这种方法下，main函数本身还是没被测试覆盖，但因为它只是调用了run函数，逻辑非常简单，很多时候我们可以接受这种情况。如果一定要覆盖main函数，还是得结合方法1的方式去调用它。

补充说明 #

• 运行测试时，确保使用-cover参数查看覆盖率，或者生成覆盖率报告后用go tool cover查看细节。
• 如果你的main函数涉及命令行参数、环境变量等，可以在测试里通过设置os.Args来模拟。

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