在 Go 中测试常量

在 Go 中，常量是确保代码一致性和正确性的宝贵工具。然而，当涉及到单元测试时，使用常量可能会带来挑战。本文探讨了如何在 Go 中重新定义专门用于测试目的的常量。

常量的挑战

Go 中的常量是不可变的值，在程序执行期间无法修改。这种不变性确保了程序状态的完整性并防止意外更改。然而，在测试过程中，可能需要临时修改常量来测试特定场景。

测试友好的方法

考虑以下示例：

package main

const baseUrl = "http://google.com"

// in main_test.go
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
  ...
}))
const baseUrl = ts.URL // Error: const baseUrl already defined

在这个例子中，我们希望使用测试服务器而不是调用远程API进行测试。为此，我们需要使用测试服务器的 URL 重新定义 baseUrl。然而，Go 会抛出错误，因为常量 baseUrl 已经定义了。

解决方案：重构可测试性

为了让测试更容易，我们可以稍微重构一下代码，引入附加功能：

const baseUrl_ = "http://google.com"

func MyFunc() string {
    // Call other function passing the const value
    return myFuncImpl(baseUrl_)
}

func myFuncImpl(baseUrl string) string {
    // use baseUrl
    // Same implementation that was in your original MyFunc() function
}

在此重构代码中，原始 MyFunc 函数将其实现委托给 myFuncImpl，该函数采用 baseUrl 参数。这允许我们直接从测试函数调用 myFuncImpl 并传入任何所需的基本 URL。

保持生产代码安全

原始 MyFunc 函数保持安全且不受重构。它总是将常量baseUrl_传递给myFuncImpl，确保生产代码不会因修改常量而出现意外行为。

结论

这种技术允许开发人员重新定义常量用于测试目的，同时保持生产代码的完整性和安全性。通过引入提供与原始功能相同的功能的附加函数，测试可以修改常量而不影响实际代码的行为。