如何 Mock Client 调用

go install go.uber.org/mock/mockgen@latest

mockgen -source=kitex_gen/xxxx/xxxservice/client.go -destination=xxx/client_mock.go -package=xxx

// MockHello 实现你想要的 mock 方法
func MockHello(ctx context.Context, req *hello.MyReq, callOptions ...callopt.Option) (*hello1.MyResp, error) {
   return &hello.MyResp{Message: "hello:" + req.Name}, nil
}

// TestClient 测试 mock 函数
func TestClient(t *testing.T) {

   ctrl := gomock.NewController(t)
   defer ctrl.Finish()

   // 获取 go mock 生成的 client
   client := NewMockClient(ctrl)

   // 添加 mock 函数
   client.EXPECT().Hello(gomock.Any(), gomock.Any(), gomock.Any()).DoAndReturn(MockHello).AnyTimes()

   // 发起 mock call
   resp, err := client.Hello(context.Background(), &hello.MyReq{Name: "bd"})
   if err == nil {
      fmt.Println(resp.Message)
   } else {
      fmt.Println(err)
   }
}