unit-testing - 如何模拟函数以在golang中获得不同的返回值

Question

我正在为我在 golang 中开发的功能编写单元测试。这是代码片段:

func myFunc () {
    //there is some code
    err := json.Unmarshal (a, &b)
    if err != nil {
        //handle error    
    }

    //there is some more code

    err := json.Unmarshal (c, &d)
    if err != nil {
        //handle error    
    }
}

现在我想模拟第一个 unmarshal 返回成功，第二个 unmarshal 返回失败。在 python 中，我看到一篇类似的帖子: Python mock multiple return values

但是我找不到 golang 的。谁能帮我解决这个问题。

Answer 1

在这种情况下我会做的是导出一个包变量 Unmarshaller 并在 init() 中将其设置为 json.Unmarshal方法。

var Unmarshaller func(data []byte, v interface{}) error;

func init() {
    Unmarshaller = json.Unmarshal
}

然后，当你想强制错误时，你可以这样做

mypackage.Unmarshaller = func(data[] byte, v interface{}) error {
    return errors.New("It broke!")
}

然后，在您的代码中，不是直接调用 json.Unmarshal，而是调用您的包级别 `Unmarshaller，所以

err =: json.Unmarshal(jsonBytes, &test)

会变成

err =: Unmarshaller(jsonBytes, &test)

作为一个简短的例子:https://play.golang.org/p/CinLmprtp5