我在Golang的单元测试中使用了模拟。但是如何在Golang的实现代码中获得存根和模拟之间的区别?
答案 0 :(得分:1)
GO中的模拟和存根的意图与不同的编程语言相同:
让我们检查一下它在示例中如何工作:
在我们的例子中,我们有一个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。