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

Question

我在 Golang 的单元测试中使用模拟。但是如何在Golang的实现代码中得到stub和mock的区别呢？

Answer 1

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

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

让我们检查一下它是如何工作的：

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

我们的处理程序代码与stub和相同mock。我们应该注入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。