Question

如果我有一个像下面那样的handlerfunc。什么是＆＃34; mock＆＃34;或者注入一个包装某个对象进行测试的接口？

func GetOrCreateUser(w http.ResponseWriter, r *http.Request) {
    user := GetUserFromContext(r.Context())
    if created :=user.GetOrCreate(); created {
        smtp.SendEmail(...)
        return
    } else {
        w.Write([]byte("Hello User!"))
    }
}

我来的唯一方法似乎是这样做：

type Mailer interface { SendMail() }

func HandlerWithMailer(m Mailer) http.handlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        user := GetUserFromContext(r.Context())
        if created := user.GetOrCreate(); created {
            m.SendEmail(...)
            return
        } else {                
            w.Write([]byte("Hello User!"))
        }
    }
}

然后在路由器中调用这样的邮件程序（使用httprouter）：

m := mailer.New() // assuming this returns a new mailer
r := httprouter.New()
r.GET("/mailer", HandlerWithMailer(m))

然后可以通过创建实现接口的结构并返回我想要的任何内容来测试，这可以在测试中调用。我知道这有效，但我想知道这是否是Go中首选的测试方法，以及是否有其他方法可以实现此目的。

Answer 1

我会从像这样的结构调用我的处理程序：

type Handlers struct {
    mailer *Mailer
}

func(h *Handlers) GetOrCreateUser(w http.ResponseWriter, r *http.Request) {
    user := GetUserFromContext(r.Context())
    if created :=user.GetOrCreate(); created {
        h.mailer.SendEmail(...)
        return
    } else {
        w.Write([]byte("Hello User!"))
    }
}

通过这种方式，您可以使用要使用的Mailer实现来实现结构。

m := mailer.New() // assuming this returns a new mailer
h := Handlers{&m}

r := httprouter.New()
r.GET("/mailer", h.HandlerWithMailer)

并在你的测试中

m := mockMailer.New() // assuming this returns a new mailer
h := Handlers{&m}

r := httprouter.New()
r.GET("/mailer", h.HandlerWithMailer)

go中的接口模拟方法

1 个答案: