如何在每次函数调用时都进行断言验证

本篇文章向大家介绍《如何在每次函数调用时都进行断言验证》，主要包括，具有一定的参考价值，需要的朋友可以参考一下。

我正在尝试使用stretchr/testify对如下代码进行单元测试：

func (c *MyClient) upsertData(data MyObject) {
    upsertToDatabase(data)
}

func doSomething(c *MyClient) {
    data1, data2 := getSomeData()
    c.upsertToDatabase(data1)
    c.upsertToDatabase(data2)
}

// Unit test.
func TestDoSomething(t *testing.T) {
    c := mock.MyClient{}
    doSomething(c)
    /* The following line checking for data1 upsert failed.
     * require.True(t, c.AssertCalled(t, "upsertToDatabase", mock.MatchedBy(func(data MyObject) bool { return data == MyObject{expectedObject1 /* data2 */}})) */
    require.True(t, c.AssertCalled(t, "upsertToDatabase", mock.MatchedBy(func(data MyObject) bool { return data == MyObject{expectedObject1 /* data2 */}}))
}

我想调用 assertcalled 并验证 data1 和 data2 确实使用预期函数调用。但我只能用函数的最后一次调用来断言，即使用 data2。有没有什么方法或者如何也可以使用 data1 来断言调用？

解决方案

the docs中的示例：

/*
  Actual test functions
*/

// TestSomething is an example of how to use our test object to
// make assertions about some target code we are testing.
func TestSomething(t *testing.T) {

  // create an instance of our test object
  testObj := new(MyMockedObject)

  // setup expectations
  testObj.On("DoSomething", 123).Return(true, nil)

  // call the code we are testing
  targetFuncThatDoesSomethingWithObj(testObj)

  // assert that the expectations were met
  testObj.AssertExpectations(t)


}

看起来您可以多次调用 .on 来记录任意数量的“以这种方式调用”期望。

我刚刚阅读了源代码，真的。我打赌它会比在 so 上发帖更快。

今天关于《如何在每次函数调用时都进行断言验证》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！