unit-testing - 如何在 Golang 中实现 stub ？ stub 和模拟之间有什么区别？

Question

<分区>

Answer 1

GO 中 mock 和 stub 的意图与不同的编程语言相同:

• stub 用于替换代码中将在测试执行期间使用的某些依赖项。它通常是为一个特定测试构建的，不太可能重复用于另一个测试，因为它具有硬编码的期望和假设。
• mock 将 stub 提升到一个新的水平。它增加了配置方式，因此您可以为不同的测试设置不同的期望。这使得模拟更加复杂，但可重复用于不同的测试。

让我们检查一下它是如何工作的示例:

在我们的例子中，我们有 http 处理程序，它在内部对另一个网络服务进行 http 调用。为了测试处理程序，我们希望将处理程序代码与我们无法控制的依赖项(外部 Web 服务)隔离开来。我们可以通过使用 stub 或 mock 来做到这一点。

对于 stub 和 mock，我们的处理程序代码是相同的。我们应该注入(inject) 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")
    ...
}

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

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
}

对于这个简单的示例，它没有增加太多值(value)。但是想想更复杂的情况。您可以构建更简洁的测试代码，并用更少的行覆盖更多案例。

如果我们必须调用 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)

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

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