当前位置：首页 > 文章列表 > Golang > Go教程 > 使用 STRETCHR/TESTIFY 和 MOCKERY 进行 GOL 测试

使用 STRETCHR/TESTIFY 和 MOCKERY 进行 GOL 测试

来源：dev.to 2024-10-17 16:07:01 0浏览收藏

从现在开始，努力学习吧！本文《使用 STRETCHR/TESTIFY 和 MOCKERY 进行 GOL 测试》主要讲解了等等相关知识点，我会在golang学习网中持续更新相关的系列文章，欢迎大家关注并积极留言建议。下面就先一起来看一下本篇正文内容吧，希望能帮到你！

让我们看一个全面的示例，其中涵盖了stretchr/testify库的常见功能以及golang中的mockery。此示例将包括使用断言进行测试、使用 require 包进行严格断言、测试 http 处理程序以及使用 mockery 模拟依赖项。

设想

假设我们有一个从外部 api 获取用户信息的服务。我们想要测试：

服务的功能。
它与外部客户端的集成。
嘲笑外部客户端。

项目结构

/project
│
├── main.go
├── service.go
├── service_test.go
├── user_client.go
├── mocks/
│   └── userclient.go (generated by mockery)
└── go.mod

代码概述

user_client.go

该文件定义了与外部用户 api 交互的接口。

package project

type user struct {
    id   int
    name string
}

type userclient interface {
    getuserbyid(id int) (*user, error)
}

service.go

该文件包含一个使用 userclient 获取用户详细信息的服务。

package project

import "fmt"

type userservice struct {
    client userclient
}

func newuserservice(client userclient) *userservice {
    return &userservice{client: client}
}

func (s *userservice) getuserdetails(id int) (string, error) {
    user, err := s.client.getuserbyid(id)
    if err != nil {
        return "", fmt.errorf("failed to get user: %w", err)
    }

    return fmt.sprintf("user: %s (id: %d)", user.name, user.id), nil
}

使用嘲讽生成模拟

您可以使用 mockery 为 userclient 生成模拟：
```
mockery --name=userclient --output=./mocks
```
这将在mocks/userclient.go 中生成一个模拟。

service_test.go

现在，让我们使用 testify 断言和模拟生成的模拟为 userservice 编写一个测试。

    package project_test

    import (
        "errors"
        "testing"

        "github.com/stretchr/testify/assert"
        "github.com/stretchr/testify/require"
        "github.com/stretchr/testify/mock"
        "project"
        "project/mocks"
    )

    func TestUserService_GetUserDetails_Success(t *testing.T) {
        // Create a new mock client
        mockClient := new(mocks.UserClient)

        // Define what the mock should return when `GetUserByID` is called
        mockClient.On("GetUserByID", 1).Return(&project.User{
            ID:   1,
            Name: "John Doe",
        }, nil)

        // Create the UserService with the mock client
        service := project.NewUserService(mockClient)

        // Test the GetUserDetails method
        result, err := service.GetUserDetails(1)

        // Use `require` for error checks
        require.NoError(t, err)
        require.NotEmpty(t, result)

        // Use `assert` for value checks
        assert.Equal(t, "User: John Doe (ID: 1)", result)

        // Ensure that the `GetUserByID` method was called exactly once
        mockClient.AssertExpectations(t)
    }

    func TestUserService_GetUserDetails_Error(t *testing.T) {
        // Create a new mock client
        mockClient := new(mocks.UserClient)

        // Define what the mock should return when `GetUserByID` is called with an error
        mockClient.On("GetUserByID", 2).Return(nil, errors.New("user not found"))

        // Create the UserService with the mock client
        service := project.NewUserService(mockClient)

        // Test the GetUserDetails method
        result, err := service.GetUserDetails(2)

        // Use `require` for error checks
        require.Error(t, err)
        assert.Contains(t, err.Error(), "user not found")

        // Ensure that the result is empty
        assert.Empty(t, result)

        // Ensure that the `GetUserByID` method was called exactly once
        mockClient.AssertExpectations(t)
    }

本例的要点

带有的断言证明：
- assert 和 require 包用于不同类型的检查。
- require 用于检查如果失败则应立即使测试失败（例如，检查 nil 错误）。
- 断言用于即使失败也可以继续的检查（例如，比较值）。
用嘲笑来嘲笑：
- mockery 生成 userclient 接口的模拟。
- 在测试中，模拟配置为使用 .on() 来指定预期输入，并使用 .return() 来指定输出。
- assertexpectations 验证是否使用预期输入调用了模拟方法。
测试错误处理：
- 一个测试检查成功场景，而另一个测试则测试服务如何处理来自 userclient 的错误。