go - 在传递正确的参数时证明模拟的断言失败

Question

Golang初学者在这里。我正在尝试为我的结构编写测试，最终模拟统计接口。就测试而言，我的代码的相关部分如下所示。

// Test code
type MockStats struct {
    mock.Mock
}

func (m *MockStats) Mean(input []float64) (mean float64, err error) {
    args := m.Called(input)
    return (args.Get(0)).(float64), args.Error(1)
}

func TestCalculateWhenErrorCalculatingMean(t *testing.T) {

    m := new(MockStats)

    data := []float64{0.0, 0.1}
    r := ResponseAggregate{ExecutionTimes: data}

    // Expectations
    m.On("Mean", []float64{0.0, 0.1}).Return(float64(0.1), errors.New("Oops!")).Once()
    m.AssertExpectations(t)

    // Function call

    err := r.Calculate(m)

    // Data assertion
    assert.NotNil(t, err)
    if assert.Error(t, err) {
        assert.Equal(t, errors.New("error calculating average. Oops!"), err)
    }

}

我的调用代码看起来像

type ResponseAggregate struct {
    ExecutionTimes             []float64
    AverageTime                float64
}


func (response *ResponseAggregate) Calculate(stats Stats) error {

    average, err := stats.Mean(response.ExecutionTimes)
    if err != nil {
        return fmt.Errorf("error calculating average. %v", err)
    }

    return nil

当我运行我的测试时，我得到了错误

            The code you are testing needs to make 1 more call(s).

你能帮我解决我在这里缺少的东西吗？谢谢。顺便说一句，使用 Go 1.16.3

Answer 1

您实际上是在拨打电话m.AssertExpectation(t)之前拨打电话r.Calculate(m)。因此，您尝试在调用方法之前检查模拟的方法调用。

只需将断言放在后面r.Calculate(m)，就可以了。