时间戳

首页 > 解决方案 > 你如何在 GoLang 的模拟中模拟错误返回值？

go - 你如何在 GoLang 的模拟中模拟错误返回值？

问题描述

我正在使用https://github.com/vektra/mockery来生成我的模拟文件。

假设我有这个功能测试GetQualifyingListOfBooks。（模拟的）客户端方法的签名GetListOfBooks是：

func GetListOfBooks(ctx context.Context, in *GetListOfBooksRequest, opts ...grpc.CallOption) (*GetListOfBooksResponse, error)

目标：我想模拟GetListOfBooks返回错误的情况，所以我写了以下内容：

package whatever

import (
    "context"
    "testing"
    "github.com/stretchr/testify/assert"
    mock "github.com/stretchr/testify/mock"
    mockClients "github.com/MyOrg/MyApp/internal/app/myservice/mocks"
)

func TestGetQualifyingListOfBooks (t *testing.T) {
    client := &mockClients.BookClient{}
    client.On("GetListOfBooks", mock.Anything, mock.Anything).Return(nil, mock.AnythingOfType("error"))
    realInstance.bookClient = client
    b, err := GetQualifyingListOfBooks(realInstance, 10) // this calls `GetListOfBooks`
    assert.Nil(t, b)
    assert.NotNil(t, err)
}

但它对此感到恐慌：

--- FAIL: TestGetQualifyingListOfBooks (0.00s)
panic: assert: arguments: Error(1) failed because object wasn't correct type: Error(1) [recovered]
    panic: assert: arguments: Error(1) failed because object wasn't correct type: Error(1)

你如何模拟错误的返回值？

标签： gotestingmockingmockery

解决方案

更改此行

client.On("GetListOfBooks", mock.Anything, mock.Anything).Return(nil, mock.AnythingOfType("error"))

对此

client.On("GetListOfBooks", mock.Anything, mock.Anything).Return(nil, errors.New("hello world"))

为我做的。

推荐阅读