快捷導(dǎo)航

使用?gomonkey?Mock?函數(shù)及方法示例詳解

更新時間：2022年06月02日 14:20:47 作者：CG國斌

在 Golang 語言中，寫單元測試的時候，不可避免的會涉及到對其他函數(shù)及方法的 Mock，即在假設(shè)其他函數(shù)及方法響應(yīng)預(yù)期結(jié)果的同時，校驗被測函數(shù)的響應(yīng)是否符合預(yù)期，這篇文章主要介紹了使用?gomonkey?Mock?函數(shù)及方法,需要的朋友可以參考下

前言

在 Golang 語言中，寫單元測試的時候，不可避免的會涉及到對其他函數(shù)及方法的 Mock，即在假設(shè)其他函數(shù)及方法響應(yīng)預(yù)期結(jié)果的同時，校驗被測函數(shù)的響應(yīng)是否符合預(yù)期。

其中，在 Mock 其他函數(shù)及方法的時候，我們常用到的一個測試類庫是「gomonkey」。特別地，對于方法和函數(shù)的 Mock，略有差異，在這里我們就分別給出函數(shù)和方法 Mock 示例，方便大家參考。

函數(shù)

在 Golang 語言中，函數(shù)是沒有接受者的方法，其形式為

func function_name([parameter list]) [return_types] {
   函數(shù)體
}

對于函數(shù)的 Mock 相對來說比較簡單，假設(shè)我們對 A 函數(shù)進(jìn)行單元測試，且 A 函數(shù)里面又調(diào)用了 B 函數(shù)，例如

func A(ctx context.Context, str string) error {
   if len(str) == 0 {
	  return errors.New("str is empty")
   }
   return test_package_name.B(ctx, str)
}

為了將 A 函數(shù)的每一行代碼都覆蓋到，則其單元測試可以寫為：

func TestA(t *testing.T) {
	type args struct {
		ctx    context.Context
		str    string
	}
	tests := []struct {
		name    string
		args    args
		Setup   func(t *testing.T)
		wantErr error
	}{
		{
			name: "len(str) == 0",
			wantErr: errors.New("str is empty")
		},
		{
			name: "正常響應(yīng)",
			Setup: func(t *testing.T) {
				patches := gomonkey.ApplyFunc(test_package_name.B, func(_ context.Context, _ string) error {
					return nil
				})
				t.Cleanup(func() {
					patches.Reset()
				})
			},
			args: args{
				ctx:     context.Background(),
				str:     "test",
			},
			wantErr: nil,
		},
	}

	// 執(zhí)行測試用例
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if tt.Setup != nil {
				tt.Setup(t)
			}
			err := A(tt.args.ctx, tt.args.str)
			if err != nil {
				assert.EqualError(t, err, tt.wantErr.Error(), "error 不符合預(yù)期")
			}
		})
	}
}

其中，ApplyFunc函數(shù)是用來 Mock 函數(shù)的，其第一個參數(shù)為需要 Mock 的函數(shù)名稱（不需要寫參數(shù)列表），第二個參數(shù)為需要 Mock 的函數(shù)結(jié)果；特別地，在Setup里面，我們要記得顯式調(diào)用Cleanup對patches進(jìn)行Reset操作，防止該 Mock 影響其他測試用例。

方法

在 Golang 語言中，方法是含有接受者的函數(shù)，其形式為

func (variable_name variable_data_type) function_name([parameter list]) [return_type]{
   函數(shù)體
}

對于方法的 Mock 相對來說復(fù)雜一下，假設(shè)我們對 A 函數(shù)進(jìn)行單元測試，且 A 函數(shù)里面又調(diào)用了結(jié)構(gòu) C 的 B 方法，例如

func A(ctx context.Context, str string) error {
   if len(str) == 0 {
	  return errors.New("str is empty")
   }
   c := &test_package_name.C{}
   return c.B(ctx, str)
}

為了將 A 函數(shù)的每一行代碼都覆蓋到，則其單元測試可以寫為：

func TestA(t *testing.T) {
	// 初始化C結(jié)構(gòu)
	var c *test_package_name.C
	
	type args struct {
		ctx    context.Context
		str    string
	}
	tests := []struct {
		name    string
		args    args
		Setup   func(t *testing.T)
		wantErr error
	}{
		{
			name: "len(str) == 0",
			wantErr: errors.New("str is empty")
		},
		{
			name: "正常響應(yīng)",
			Setup: func(t *testing.T) {
				patches := gomonkey.ApplyMethod(reflect.TypeOf(c), "B", func(_ *test_package_name.C, _ context.Context, _ string) error {
					return nil
				})
				t.Cleanup(func() {
					patches.Reset()
				})
			},
			args: args{
				ctx:     context.Background(),
				str:     "test",
			},
			wantErr: nil,
		},
	}

	// 執(zhí)行測試用例
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if tt.Setup != nil {
				tt.Setup(t)
			}
			err := A(tt.args.ctx, tt.args.str)
			if err != nil {
				assert.EqualError(t, err, tt.wantErr.Error(), "error 不符合預(yù)期")
			}
		})
	}
}

其中，ApplyMethod函數(shù)是用來 Mock 方法的，其第一個參數(shù)為需要 Mock 的方法的接受者類型，第二個參數(shù)為需要 Mock 的方法名稱（字符串類型），第三個參數(shù)為需要 Mock 的方法的定義及 Mock 結(jié)果；特別地，第一個參數(shù)和第三個參數(shù)需要我們注意：

第一個參數(shù)，需要使用reflect.TypeOf獲取接受者的類型，初始化的接受者必須是真正的類型，如結(jié)構(gòu) C 組合了結(jié)構(gòu) D，而B方法是通過組合 D 得到的，則初始化的時候需要定義結(jié)構(gòu) D，而不是結(jié)構(gòu) C，否則會報空指針異常；
第三個參數(shù)，雖然B方法的聲明是func(ctx context.Context, str string)，但是在使用ApplyMethod的時候，需要將B方法的聲明修改為func(c *test_package_name.C, ctx context.Context, str string)，即需要將方法的接受者置為方法的第一個參數(shù)。