如何在 Golang 中测试私有函数？

在 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 来暂时禁用私有成员检查，从而允许我们访问和测试私有函数：

传媒公司模板(RTCMS)1.0

传媒企业网站系统使用热腾CMS(RTCMS),根据网站板块定制的栏目，如果修改栏目，需要修改模板相应的标签。站点内容均可在后台网站基本设置中添加。全站可生成HTML，安装默认动态浏览。并可以独立设置SEO标题、关键字、描述信息。源码包中带有少量测试数据，安装时可选择演示安装或全新安装。如果全新安装，后台内容充实后，首页才能完全显示出来。（全新安装后可以删除演示数据用到的图片，目录在https://

下载

立即学习“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)
        }
    }
}