Go单元测试中stub和mock有什么区别？

Question

我在 Go 的单元测试中使用模拟。Go的实现代码中如何理解stub和mock的区别？

Answer 1

GO 中 mocks 和 stubs 的意图与不同的编程语言相同：

• 存根替代了代码中将在测试执行期间使用的某些依赖项。它通常是为一个特定的测试而构建的，不太可能被重新用于另一个测试，因为它有硬编码的期望和假设。

• 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 。