慕桂英4014372
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 。