【轉】golang mock測試？ mock掉數據庫操作？

 

原文： https://www.jianshu.com/p/19089baf79ca

 

 

 

 

 a.go

 package main
 import (
   "fmt"
   "bou.ke/monkey"
 )
 
 func foo1(s string) string {
     return fmt.Sprintf("i am %s, calling function foo1", s)
 }
 
 func foo2(s string) string {
     return fmt.Sprintf("i am %s, calling function foo2", s)
 }
 func main() {
   monkey.Patch(foo1, foo2)
   fmt.Println(foo1("apple"))  //輸出：i am apple, calling function foo2
 }

　　

------------------

 

背景：項目進行 mock 操作，選擇了 monkey patch 的方式。

在 github 上找了一個開源項目 https://github.com/bouk/monkey

使用如下：

下載： go get github.com/bouk/monkey

package github.com/bouk/monkey: code in directory /xxxx/src/github.com/bouk/monkey expects import "bou.ke/monkey"

查看它的源碼：github.com/bouk/monkey/monkey.go 第一行顯示 package monkey // import "bou.ke/monkey"
正常的 import 方式為 import "github.com/bouk/monkey", 源碼指定了 import 方式，
因此需要：mv github.com/bouk bou.ke

下載完后使用結果：

func foo1(s string) string { return fmt.Sprintf("i am %s, calling function foo1", s) } func foo2(s string) string { return fmt.Sprintf("i am %s, calling function foo2", s) } // result fmt.Println(foo1("apple")) // 輸出：i am apple, calling function foo1 fmt.Println(foo2("pear")) // 輸出：i am pear, calling function foo2 

結果是符合預期的，場景是有時候我需要對函數 mock 操作，進行測試（例如 mock 掉數據庫操作），代碼如下：

func foo1(s string) string { return fmt.Sprintf("i am %s, calling function foo1", s) } func foo2(s string) string { return fmt.Sprintf("i am %s, calling function foo2", s) } monkey.Patch(foo1, foo2) fmt.Println(foo1("apple")) //輸出：i am apple, calling function foo2 

這也是符合預期的，因為 foo1 函數被 foo2 給替換了。

下面看另外的例子：

func foo3() bool { return true } func foo4() bool { return false } monkey.Patch(foo3, foo4) fmt.Println(foo3()) //實際輸出：true 

問題是我們期待的結果是應該是輸出 false,因為 foo3 進行了 monkey patch 操作。看一下說明：

Monkey sometimes fails to patch a function if inlining is enabled. Try running your tests with inlining disabled, for example: go test -gcflags=-l. The same command line argument can also be used for build.

大概原因就是此 monkey patch 方式對 inline function 無效，因此需要 disable inlining ,添加參數 -gcflags=-l, 因為上述的例子執行 go run -gcflags=-l main.go 得到符合預期的和正確的輸出結果 false。

上面是使用 https://github.com/bouk/monkey 的一些心得。

另： 開源項目的原作者實現的原理： https://bou.ke/blog/monkey-patching-in-go/

作者：七秒鍾回憶待續
鏈接：https://www.jianshu.com/p/19089baf79ca
來源：簡書
著作權歸作者所有。商業轉載請聯系作者獲得授權，非商業轉載請注明出處。