在执行测试套件期间设置特定的环境变量

Question

我想在 Go 中运行与环境变量有关的单元测试 fxn。

在某个时候我会去（没有双关语）

    for test := range tests {
        for k, v := range test.envVars {
            os.Setenv(k, v)
        }

有没有办法确保在测试执行期间，通过上述对os.Setenv() 的调用设置的变量将是唯一可用于测试函数的环境变量？

Answer 1

您可以使用os.Environ获取当前环境，使用os.Unsetenv取消设置。

您可能还需要在测试完成后恢复这些变量，为此您可以在每个环境中使用subtest 和(*testing.T).Cleanup (https://play.golang.org/p/A5HGZwWm-NS)：

    for _, tc := range tests {
        t.Run(tc.name, func(t *testing.T) {
            // NOTE: do NOT call t.Parallel in this subtest or its parent,
            // or the environment WILL be corrupted.
            //
            // (However, *subtests of* t may be run in parallel.)

            prevEnv := os.Environ()
            for _, entry := range prevEnv {
                parts := strings.SplitN(entry, "=", 2)
                os.Unsetenv(parts[0])
            }
            for k, v := range tc.envVars {
                os.Setenv(k, v)
            }
            t.Cleanup(func() {
                for k := range tc.envVars {
                    os.Unsetenv(k)
                }
                for _, entry := range prevEnv {
                    parts := strings.SplitN(entry, "=", 2)
                    os.Setenv(parts[0], parts[1])
                }
            })

            …
        })
    }

---

还要注意，Go 1.17 添加了(*testing.T).Setenv 方法，它会自动清理修改。不过目前还没有对应的(*testing.T).Unsetenv。

---

如果您希望能够并行运行测试，您可以改为使用带有os/exec 和-test.run 标志的子测试来重新执行特定的子测试作为主测试进程的子进程，并具有自己独特的环境。