Go语言中的单元测试：处理程序

Question

这是我编写的处理程序，用于从mongodb检索文档。
如果找到文档，我们将根据需要加载和呈现模板。 如果失败，则会重定向到404页。

func EventNextHandler(w http.ResponseWriter, r *http.Request) {

    search := bson.M{"data.start_time": bson.M{"$gte": time.Now()}}
    sort := "data.start_time"
    var result *Event
    err := db.Find(&Event{}, search).Sort(sort).One(&result)
    if err != nil && err != mgo.ErrNotFound {
        panic(err)
    }
    if err == mgo.ErrNotFound {
        fmt.Println("No such object in db. Redirect")
        http.Redirect(w, r, "/404/", http.StatusFound)
        return
    }

    // TODO: 
    // This is the absolute path parsing of template files so tests will pass
    // Code can be better organized
    var eventnext = template.Must(template.ParseFiles(
        path.Join(conf.Config.ProjectRoot, "templates/_base.html"),
        path.Join(conf.Config.ProjectRoot, "templates/event_next.html"),
    ))

    type templateData struct {
        Context *conf.Context
        E *Event
    }

    data := templateData{conf.DefaultContext(conf.Config), result}

    eventnext.Execute(w, data)

}

手动尝试这个，一切都很好。

然而，我似乎无法通过单元测试。在相应的测试文件中，以下是我的测试代码，尝试加载我的 EventNextHandler。

func TestEventNextHandler(t *testing.T) {

    // integration test on http requests to EventNextHandler

    request, _ := http.NewRequest("GET", "/events/next/", nil)
    response := httptest.NewRecorder()

    EventNextHandler(response, request)

    if response.Code != http.StatusOK {
        t.Fatalf("Non-expected status code%v:\n\tbody: %v", "200", response.Code)
    }

}

测试在代码行 EventNextHandler(response, request) 处失败。

而在我的错误信息中，它指的是处理程序代码中的 err := db.Find(&Event{}, search).Sort(sort).One(&result) 行。

完整的错误消息如下：

=== RUN TestEventNextHandler
--- FAIL: TestEventNextHandler (0.00 seconds)
panic: runtime error: invalid memory address or nil pointer dereference [recovered]
    panic: runtime error: invalid memory address or nil pointer dereference
[signal 0xb code=0x1 addr=0x8 pc=0x113eb8]

goroutine 4 [running]:
testing.func·004()
    /usr/local/go/src/pkg/testing/testing.go:348 +0xcd
hp/db.Cursor(0xc2000c3cf0, 0xc2000fb1c0, 0x252d00)
    /Users/calvin/gopath/src/hp/db/db.go:57 +0x98
hp/db.Find(0xc2000c3cf0, 0xc2000fb1c0, 0x252d00, 0xc2000e55c0, 0x252d00, ...)
    /Users/calvin/gopath/src/hp/db/db.go:61 +0x2f
hp/event.EventNextHandler(0xc2000e5580, 0xc2000a16a0, 0xc2000c5680)
    /Users/calvin/gopath/src/hp/event/controllers.go:106 +0x1da
hp/event.TestEventNextHandler(0xc2000d5240)
    /Users/calvin/gopath/src/hp/event/controllers_test.go:16 +0x107
testing.tRunner(0xc2000d5240, 0x526fe0)
    /usr/local/go/src/pkg/testing/testing.go:353 +0x8a
created by testing.RunTests
    /usr/local/go/src/pkg/testing/testing.go:433 +0x86b

goroutine 1 [chan receive]:
testing.RunTests(0x3bdff0, 0x526fe0, 0x1, 0x1, 0x1, ...)
    /usr/local/go/src/pkg/testing/testing.go:434 +0x88e
testing.Main(0x3bdff0, 0x526fe0, 0x1, 0x1, 0x532580, ...)
    /usr/local/go/src/pkg/testing/testing.go:365 +0x8a
main.main()
    hp/event/_test/_testmain.go:43 +0x9a

如何正确编写测试？要考虑从mongodb中未检索到任何内容的情况，并应用断言来验证，从而编写一个经过验证的测试。

- Calvin Cheng

---

db 是如何定义和初始化的？如果在测试期间没有进行初始化，那么这就是你的问题所在。如果你想要更简单的方法来为测试参数化处理程序，则可以将其实现为 Handler 类型而不是处理程序函数，并将数据库连接作为该类型的成员。 - James Henstridge

是的，那就是问题所在。我想通了。谢谢！:-p - Calvin Cheng

顺便提一下，你可以考虑使用GoConvey来让 Go 的测试更容易。该设置只需在任何测试用例中完成一次即可。 - Matt

Answer 1

看起来我没有在我的测试中初始化db。这与其他语言的单元测试框架类似，需要进行“设置”步骤。

我必须确保导入db包，并实现两行代码连接到数据库并注册索引。

func TestEventNextHandler(t *testing.T) {

    // set up test database

    db.Connect("127.0.0.1", "test_db")
    db.RegisterAllIndexes()

    // integration test on http requests to EventNextHandler

    request, _ := http.NewRequest("GET", "/events/next/", nil)
    response := httptest.NewRecorder()

    EventNextHandler(response, request)

    if response.Code != 302 {
        t.Fatalf("Non-expected status code %v:\n\tbody: %v", "200", response.Code)
    }

}

- Calvin Cheng

---

为什么在处理程序上进行单元测试需要在数据库级别上建立连接？难道不应该模拟吗？ - TasteMyBiceps