golang gorilla/mux 和测试，在哪里存储路由

【问题标题】：golang gorilla/mux and testing, where to store routesgolang gorilla/mux 和测试，在哪里存储路由
【发布时间】：2023-03-30 08:05:02
【问题描述】：

在我看到的大多数文档中，建议像这样使用 gorilla mux...

func main() {
    m := mux.Router()
    m.HandleFunc("/", FuncNameOrDef)
    http.ListenAndServe(":8080", m)
}

这很好，但它给我留下了一个问题，因为为了测试，据我所知，我需要重新声明多路复用器和路由，除非我在这样的函数之外声明多路复用器和路由.. .

var (
    m = mux.Router()
    _ = m.HandleFunc("/", FuncNameOrDef)
)

然后在我的测试中这样做......

func TestSomeView(t *testing.T) {
    ts := httptest.NewServer(m)
    ....testing blah
}

这解决了问题，但它使包变得很丑（所有_ = m.HandleFunc's）有没有更惯用的方法来做到这一点？

【问题讨论】：

标签： go

【解决方案1】：

您可以使用工厂函数为您的应用程序创建一个新的http.Handler。这将允许您以编程方式定义处理程序并在测试中重用它们。

// NewApplicationHandler provides a configured handler for the
// application.
func NewApplicationHandler() http.Handler {
    mux := mux.NewRouter()
    mux.HandleFunc("/", handler)

    return mux
}

func handler(w http.ResponseWriter, r *http.Request) {
    w.Write([]byte("Hello World!"))
}

这个例子可以用在这样的测试中：

func TestApplicationServeMux(t *testing.T) {
    // The handlers are being obtained via a call to the factory function.
    applicationHandler := NewApplicationHandler()

    ts := httptest.NewServer(applicationHandler)
    defer ts.Close()

    res, err := http.Get(ts.URL)

    if err != nil {
        t.Fatalf("%s", err)
    }

    defer res.Body.Close()
    greeting, err := ioutil.ReadAll(res.Body)

    if err != nil {
        log.Fatalf("%s", err)
    }

    want := []byte("Hello World!")

    if !bytes.Equal(want, greeting) {
        t.Errorf("Expected greeting %s; got %s", want, greeting)
    }

}

【讨论】：