主张： 模拟函数：我遇到了返回值的困扰（尽管我已经定义了模拟函数和返回语句）

大家好，今天本人给大家带来文章《主张： 模拟函数：我遇到了返回值的困扰（尽管我已经定义了模拟函数和返回语句）》，文中内容主要涉及到，如果你对Golang方面的知识点感兴趣，那就请各位朋友继续看下去吧~希望能真正帮到你们，谢谢！

我使用 testify 为我的 golang 应用程序创建单元测试。我需要为此函数创建一个单元测试，其中它调用可变参数函数（带有尾随参数的函数）。我测试的时候遇到了一个错误。我实际上不确定错误是否是由于尾随参数本身引起的，但我觉得模拟有问题。

// svc/callthisfunction.go
// data type of args is []sqkit.selectoption
func callthisfunction(ctx context.context, args ...sqkit.selectfunctiom) (result string, err error) {
  return result, nil
}

// svc/functiontotest.go
// this is the function that i wanna test
func functiontotest(ctx context.context, id int64) (result string, err error) {
  args := []sqkit.selectoption{
    sqkit.where{
      fmt.sprintf("id = %d", id),
    },
  }

  newresult, err := callthisfunctionservice.callthisfunction(ctx, args)
  if err != nil {
    return newresult, err
  }

  return newresult, nil
}

// svc/functiontotest_test.go

func test_functiontotest(t *testing.t) {
  testcase := []struct {
    name string
    id int64
    oncallthisfunctionmock func(callthisfunctionsvc *mocks.callthisfunctionsvc)
    expectedresult string
    wanterror bool
    expectederror error  
  }{
      {
        name: "success",
        id: 1,
        oncallthisfunctionmock: func(callthisfunctionsvc *mocks.callthisfunctionsvc) {
          // note: i've created 2 different versions (used separately, not at the same), using mock.anything() and using actual arguments
          // both of these give the same errors
          // using actual arguments          
          args := []sqkit.selectoption{
            sqkit.where{
              fmt.sprintf("id = %d", 1},
            },
          }
          callthisfunctionsvc.on("callthisfunction", context.background(), args).return("success", nil)

          // using mock.anything
          callthisfunctionsvc.on("callthisfunction", context.background(), mock.anything).return("success", nil)
        }
      }
   }

   for _, tc := range testcases {
     var callthisfunctionsvc = new(mocks.callthisfunctionsvc)

     tc.oncallthisfunctionmock(callthisfunctionsvc)

     svc := &svc.functiontotest{
       callthisfunction: callthisfunctionsvc,
     }

     actualresult, actualerror := svc.functiontotest(context.background(), tc.id)

     if tc.wanteror {
       require.error(t, actualerror, tc.expectederror)
     } else {
       require.noerror(t, actualerror)
     }

     require.equal(t, tc.expectedresult, actualresult)
   }
}

这是它给出的错误

=== RUN   Test_GenerateDocument
--- FAIL: Test_GenerateDocument (0.00s)
panic: 
assert: mock: I don't know what to return because the method call was unexpected.
        Either do Mock.On("CallThisFunction").Return(...) first, or remove the GetTemplates() call.
        This method was unexpected:
                CallThisFunction(*context.emptyCtx,sqkit.Where)
                0: (*context.emptyCtx)(0xc0000a4010)
                1: sqkit.Where{"id = 1"}

通常，当我遇到这样的错误时，是因为我没有在我想要测试的函数中定义函数调用的返回值。但这一次我创建了它，但它不知何故无法读取返回值。知道为什么吗？

正确答案

该错误表明您使用参数 (context.context, sqkit.where) 调用了 callthisfuncion，但您的示例正在使用并设置 (context.context, []sqkit.option) 的期望。 mock.anything 的示例应该可以工作，但我相信它由于上下文而失败。您需要使用您传递的相同上下文来设置期望。如果 functiontotest 要更改上下文，我相信您需要使用 mock.anything 代替。

func test_functiontotest(t *testing.t) {
  testcase := []struct {
    name string
    id int64
    oncallthisfunctionmock func(context.context, *mocks.callthisfunctionsvc)
    expectedresult string
    wanterror bool
    expectederror error  
  }{
      {
        name: "success",
        id: 1,
        oncallthisfunctionmock: func(ctx context.context, callthisfunctionsvc *mocks.callthisfunctionsvc) {
          args := []sqkit.selectoption{
            sqkit.where{
              fmt.sprintf("id = %d", 1},
            },
          }
          callthisfunctionsvc.on("callthisfunction", ctx, args).return("success", nil)
        }
      }
  }

  for _, tc := range testcases {
    var callthisfunctionsvc = new(mocks.callthisfunctionsvc)
    var ctx = context.background()

    tc.oncallthisfunctionmock(ctx, callthisfunctionsvc)

    svc := &svc.functiontotest{
      callthisfunction: callthisfunctionsvc,
    }

    actualresult, actualerror := svc.functiontotest(ctx, tc.id)

    if tc.wanteror {
      require.error(t, actualerror, tc.expectederror)
    } else {
      require.noerror(t, actualerror)
    }

    require.equal(t, tc.expectedresult, actualresult)
  }
}

如果您想确保 context.context 作为第一个参数传递，但不关心上下文是什么，您可以使用 anythingoftype。

callThisFunctionSvc.On("CallThisFunction", mock.AnythingOfType("context.Context"), args).Return("Success", nil)

终于介绍完啦！小伙伴们，这篇关于《主张： 模拟函数：我遇到了返回值的困扰（尽管我已经定义了模拟函数和返回语句）》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！