23

我正在使用 go lang 测试在 for 循环中运行具有多个参数的测试。

我遇到了每次调用模拟时都返回相同的返回值（和第一个集合）的情况。我想要做的是在输入相同时更改每个测试的返回值，即在循环中相同On但不同。Return

我正在使用stretchr/testify 进行模拟。看起来它不会在相On同时覆盖已经创建的模拟。

func TestUpdateContactWithNewActions(t *testing.T) {
    tests := []struct {
        testName  string
        getParams func() *activities.UpdateContactWithNewActionsActivity
        mockError error
    }{

        {"UpdateContactWithNewActions with error from contact service",
            func() *activities.UpdateContactWithNewActionsActivity {
                return fixtures.GetUpdateContactWithNewActionsActivity()
            }, fixtures.Err},
        {"UpdateContactWithNewActions valid",
            func() *activities.UpdateContactWithNewActionsActivity {
                return fixtures.GetUpdateContactWithNewActionsActivity()
            }, nil},
    }

    lib.LoadWithMockClients()

    for _, test := range tests {
        test := test
        t.Run(test.testName, func(t *testing.T) {
            lib.MockCSClient.On(
                "UpdateContactWithNewActions",
                mock.AnythingOfType("tchannel.headerCtx"),
                fixtures.UpdateContactWithNewActions).Return(test.mockError)

            returnedResult, err := test.getParams().Execute(fixtures.Ctx)
            if test.mockError == nil {
                // some assertion
            }
            assert.Error(t, err)
        })
    }
}
4

2 回答 2

31

我有一个类似的问题。

解决方案是方法Once()

在您的模拟中添加一个.Once()并使用您需要的每个结果重复模拟。

像这样的东西：

lib.Mock.On("method", arg).Return(test.mockError).Once()
lib.Mock.On("method", arg).Return(nil).Once()

每个模拟结果只会返回一次。

https://godoc.org/github.com/stretchr/testify/mock#Call.Once

于 2020-04-11T15:25:38.937 回答
3

当结果需要只返回一次时，@Marcos 提供的答案效果很好。
但是在每个返回值都需要多次（未知）返回的情况下，它就行不通了。

我解决它的方法是mock.ExpectedCalls直接操作。在我的例子中，mock 只包含一个方法，所以清理整个ExpectedCalls切片很简单，但如果有多个方法，ExpectedCalls切片可以迭代，只更新所需的调用。

这是一个简单案例的工作示例：

lib.Mock.On("method", arg).Return("1")

assert.Equal(t, lib.Mock.method(arg), "1")
assert.Equal(t, lib.Mock.method(arg), "1")
....
assert.Equal(t, lib.Mock.method(arg), "1")

lib.Mock.ExpectedCalls = nil // cleanup the previous return value
lib.Mock.On("method", arg).Return("2")
assert.Equal(t, lib.Mock.method(arg), "2")
assert.Equal(t, lib.Mock.method(arg), "2")
....
assert.Equal(t, lib.Mock.method(arg), "2")
于 2021-12-21T14:10:03.917 回答