如何在Golang中实现存根?存根和模拟之间有什么区别?

时间:2018-11-18 11:19:03

标签: unit-testing go

我在Golang的单元测试中使用了模拟。但是如何在Golang的实现代码中获得存根和模拟之间的区别?

1 个答案:

答案 0 :(得分:1)

GO中的模拟和存根的意图与不同的编程语言相同:

  • 存根替代了代码中将在测试执行期间使用的某些依赖性。它通常是为一种特定的测试而构建的,不太可能再用于另一项测试,因为它具有硬编码的期望和假设。
  • 模拟将存根提升到新的水平。它增加了配置手段,因此您可以为不同的测试设置不同的期望。这使模拟更加复杂,但可用于不同的测试。

让我们检查一下它在示例中如何工作:

在我们的例子中,我们有一个http处理程序,在内部对另一个Web服务进行http调用。为了测试处理程序,我们希望将处理程序代码与不受控制的依赖区分开(外部Web服务)。我们可以使用stubmock来做到这一点。

我们的处理程序代码与stubmock相同。我们应该注入http.Client依赖项,以便能够在单元测试中将其隔离:

func New(client http.Client) http.Handler {
    return &handler{
        client: client,
    }
}

type handler struct {
    client http.Client
}

func (h handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    ...
    // work with external web service that cannot be executed in unit test
    resp, err := h.client.Get("http://example.com")
    ...
}

我们在http.Client中直接替换了运行时stub

func TestHandlerStub(t *testing.T) {
    mux := http.NewServeMux()
    mux.Handle("/", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // here you can put assertions for request

        // generate response
        w.WriteHeader(http.StatusOK)
    }))
    server := httptest.NewServer(mux)

    r, _ := http.NewRequest(http.MethodGet, "https://some.com", nil)
    w := httptest.NewRecorder()
    sut := New(server.Client())
    sut.ServeHTTP(w, r)
    //assert handler response
}

模拟故事更加复杂。我正在跳过模拟实现的代码,但这是其接口的外观:

type Mock interface {
    AddExpectation(path string, handler http.HandlerFunc)
    Build() *http.Client
}

这是使用Mock进行测试的代码:

func TestHandlerMock(t *testing.T) {
    mock := NewMock()
    mock.AddExpectation("/", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // here you can put assertions for request

        // generate response
        w.WriteHeader(http.StatusOK)
    }))

    r, _ := http.NewRequest(http.MethodGet, "https://some.com", nil)
    w := httptest.NewRecorder()
    sut := New(mock.Build())
    sut.ServeHTTP(w, r)
    //assert handler response
}

对于这个简单的示例,它没有太大的价值。但是考虑更复杂的情况。您可以构建更简洁的测试代码,并以更少的行覆盖更多案例。

如果我们必须调用2个服务并稍微改进了模拟,这就是测试设置的样子:

mock.AddExpectation("/first", firstSuccesfullHandler).AddExpectation("/second", secondSuccesfullHandler)
mock.AddExpectation("/first", firstReturnErrorHandler).AddExpectation("/second", secondShouldNotBeCalled)
mock.AddExpectation("/first", firstReturnBusy).AddExpectation("/first", firstSuccesfullHandler)AddExpectation("/second", secondSuccesfullHandler)

您可以想象,如果我们没有小型的模拟助手,则必须在测试中复制粘贴处理程序逻辑多少次。复制粘贴的代码使我们的测试变得脆弱。

但是构建自己的模拟并不是唯一的选择。您可以依赖现有的模拟SQL的模拟程序包,例如DATA-DOG/go-sqlmock