如何对 Golang 中的自定义类型进行单元测试？

苏承栈

2026-10-01 • 👁 0人浏览过

在 golang 中对自定义类型进行单元测试至关重要，方法包括：1. 使用 testing 包：创建 test 函数并使用 t.error() 报告错误；2. 使用 mocking 框架（如 gomock 和 mockery）：创建模拟类型进行测试；3. 使用辅助函数：创建一个辅助函数来测试类型，并将其用于单元测试中。如何在 Golang 中对自定义类型进行单元测试对自定义类型进行单元测试至关重要，因为它可以确保类型的行为符合预期。在 Golang 中，有几种对自定义类型进行单元测试的方法。 1. 使用标准的 testing 包立即学习 “ go语言免费学习笔记（深入） ”；

testing

包提供了用于编写和运行单元测试的工具。对于自定义类型，可以使用

Test

函数来定义测试用例，并使用

t.Error()

函数报告错误： Git 程序猿必备版本控制工具下载

package mypackage

import "testing"

type MyType struct {
    value int
}

func TestSum(t *testing.T) {
    myType := MyType{1}
    if myType.Sum(2) != 3 {
        t.Error("Expected 3, got", myType.Sum(2))
    }
}

2. 使用 Mocking 框架 Mocking 框架允许您创建自定义类型的模拟，以便更轻松地测试您的代码。流行的 Mocking 框架包括

gomock

和

mockery

：使用

gomock

：

package mypackage

import (
    "testing"
    "github.com/golang/mock/gomock"
)

type MyInterface interface {
    DoSomething(value int)
}

func TestMyFunction(t *testing.T) {
    ctrl := gomock.NewController(t)
    defer ctrl.Finish()

    mockMyInterface := gomock.NewMockFrom(ctrl, "MyInterface")
    mockMyInterface.EXPECT().DoSomething(1).Return(nil)

    myFunction(mockMyInterface)
}

3. 使用辅助函数创建一个辅助函数来测试您的自定义类型，可以帮助您保持代码的整洁性。然后，您可以在单元测试中使用此辅助函数：

package mypackage

import (
    "testing"
)

type MyType struct {
    value int
}

func TestSum(t *testing.T) {
    myType := MyType{1}
    if assertEqual(myType.Sum(2), 3, t) {
        t.Error("Expected 3, got", myType.Sum(2))
    }
}

func assertEqual(actual, expected int, t *testing.T) bool {
    if actual != expected {
        t.Errorf("Expected %d, got %d", expected, actual)
    }
    return actual == expected
}

相关文章