学习Go语言测试：深入理解依赖注入-优快云博客

本文链接：https://blog.youkuaiyun.com/gitblog_00233/article/details/148361776

学习Go语言测试：深入理解依赖注入

learn-go-with-tests Learn Go with test-driven development 项目地址: https://gitcode.com/gh_mirrors/le/learn-go-with-tests

依赖注入(Dependency Injection, DI)是软件开发中一个非常重要的概念，但很多开发者对它存在误解。本文将通过Go语言测试驱动开发的方式，带你真正理解依赖注入的本质和优势。

什么是依赖注入

依赖注入是一种设计模式，它允许我们将对象的依赖关系从内部创建改为外部传入。简单来说，就是"不要自己创建需要的对象，让别人传给你"。

为什么需要依赖注入

让我们从一个简单的问候函数开始：

func Greet(name string) {
    fmt.Printf("Hello, %s", name)
}

这个函数看起来很简单，但它有一个问题：难以测试。因为它直接调用了fmt.Printf，将输出写到了标准输出(stdout)，在测试中我们无法捕获这个输出。

使用接口解耦

Go语言中的接口提供了一种优雅的解决方案。我们可以修改函数，让它接受一个io.Writer接口而不是直接使用fmt.Printf：

func Greet(writer io.Writer, name string) {
    fmt.Fprintf(writer, "Hello, %s", name)
}

io.Writer是Go标准库中一个非常常用的接口，定义如下：

type Writer interface {
    Write(p []byte) (n int, err error)
}

测试驱动开发实践

第一步：编写测试

func TestGreet(t *testing.T) {
    buffer := bytes.Buffer{}
    Greet(&buffer, "Chris")
    
    got := buffer.String()
    want := "Hello, Chris"
    
    if got != want {
        t.Errorf("got %q want %q", got, want)
    }
}

这里我们使用了bytes.Buffer，它实现了io.Writer接口，可以作为我们测试中的"模拟"输出目标。

第二步：实现功能

func Greet(writer io.Writer, name string) {
    fmt.Fprintf(writer, "Hello, %s", name)
}

第三步：重构和扩展

现在我们的Greet函数更加灵活了，它不仅可以用于测试，还可以用于各种场景：

命令行应用：

func main() {
    Greet(os.Stdout, "Elodie")
}

Web服务器：

func MyGreeterHandler(w http.ResponseWriter, r *http.Request) {
    Greet(w, "world")
}

func main() {
    http.ListenAndServe(":5001", http.HandlerFunc(MyGreeterHandler))
}