在 go 中测试私有函数可以通过以下方法:1. 使用反射获取私有函数对象并直接调用;2. 使用 testing 包中的 internaltest 禁用私有成员检查,然后直接访问和测试私有函数。在测试私有函数时,可以选择使用反射或 testing 包中的 internaltest,具体方法根据实际测试场景选择,并确保在测试完成后恢复正常模式。
如何在 Golang 中测试私有函数?
在 Golang 中,私有函数是指仅在定义的包内可见的函数。这意味着您无法直接从其他包访问或测试这些函数。但是,可以通过间接方法或工具来测试私有函数。
方法 1:使用反射
反射是 Golang 标准库中的一项强大功能,它允许您访问和操纵类型和值的信息。我们可以使用反射来获取私有函数并进行测试:
package main
import (
"fmt"
"reflect"
)
// 定义私有函数
type MyStruct struct{}
func (s *MyStruct) privateFunc() {
fmt.Println("I'm a private function")
}
func main() {
// 使用反射获取私有函数对象
v := reflect.ValueOf(new(MyStruct))
method := v.MethodByName("privateFunc")
// 调用私有函数
method.Call(nil)
}
方法 2:使用测试包
在 Go 1.17 中,引入了 testing 包,它包含用于辅助测试的额外功能。我们可以使用 testing.InternalTest 来暂时禁用私有成员检查,从而允许我们访问和测试私有函数:
立即学习“go语言免费学习笔记(深入)”;
package main
import (
"fmt"
"testing"
)
// 定义私有函数
type MyStruct struct{}
func (s *MyStruct) privateFunc() {
fmt.Println("I'm a private function")
}
func TestPrivate(t *testing.T) {
// 为了更好地显示,我们使用匿名类型
// 在实际测试中通常使用常量作为标志
t.Run("Enabled", func(t *testing.T) {
// 禁用私有成员检查
testing.InternalTest(t, func(t *testing.T) {
s := &MyStruct{}
s.privateFunc()
})
})
t.Run("Disabled", func(t *testing.T) {
s := &MyStruct{}
// 恢复正常模式
assert.Panics(t, func() { s.privateFunc() })
})
}
实战案例
假设我们有一个 util 包,其中包含一个将字符串转换为数字的私有辅助函数:
// util/util.go
package util
import (
"strconv"
)
// 私有函数将字符串转换为数字
func strToInt(s string) (int, error) {
return strconv.Atoi(s)
}
我们可以使用以上方法之一来测试 strToInt 函数:
// main_test.go
package main
import (
"testing"
"util"
)
func TestStrToInt(t *testing.T) {
tests := []struct {
input string
want int
err bool
}{
{"123", 123, false},
{"-456", -456, false},
{"abc", 0, true},
}
for _, tt := range tests {
got, err := util.strToInt(tt.input)
if tt.err && err == nil {
t.Errorf("Expected error, got: nil")
} else if !tt.err && err != nil {
t.Errorf("Unexpected error: %v", err)
} else if got != tt.want {
t.Errorf("Expected %v, got: %v", tt.want, got)
}
}
}