testflight
Installation
go get github.com/drewolson/testflightimport "github.com/drewolson/testflight"Usage
testflight makes it simple to test your http servers in Go. Suppose you're using pat to create a simple http handler, like so:
func Handler() http.Handler {
m := pat.New()
m.Get("/hello/:name", http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
io.WriteString(w, "hello, "+req.URL.Query().Get(":name"))
}))
m.Post("/post/form", http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
req.ParseForm()
name := req.Form.Get("name")
w.WriteHeader(201)
io.WriteString(w, name+" created")
}))
return m
}Let's use testflight to test our handler. Keep in mind that testflight is doing full-stack http tests. We're also using assert for test assertions.
func TestGet(t *testing.T) {
testflight.WithServer(Handler(), func(r *testflight.Requester) {
response := r.Get("/hello/drew")
assert.Equal(t, 200, response.StatusCode)
assert.Equal(t, "hello, drew", response.Body)
})
}
func TestPostWithForm(t *testing.T) {
testflight.WithServer(Handler(), func(r *testflight.Requester) {
response := r.Post("/post/form", testflight.FORM_ENCODED, "name=Drew")
assert.Equal(t, 201, response.StatusCode)
assert.Equal(t, "Drew created", response.Body)
})
}The testflight.Requester struct has the following methods: Get, Post, Put, Delete and Do. Do accepts an *http.Request for times when you need more explicit control of your request. See testflight_test.go for more usage information.
Testing Websockets
Testflight also allows you to perform full-stack testing of websockets. You'll want to import both the testflight and testflight/ws packages.
import (
"github.com/drewolson/testflight"
"github.com/drewolson/testflight/ws"
)Now, let's make a handler with a websocket route.
func Handler() http.Handler {
mux := http.NewServeMux()
mux.Handle("/websocket", websocket.Handler(func(ws *websocket.Conn) {
var name string
websocket.Message.Receive(ws, &name)
websocket.Message.Send(ws, "Hello, "+name)
}))
return mux
}Finally, let's write the test.
func TestWebSocket(t *testing.T) {
testflight.WithServer(Handler(), func(r *testflight.Requester) {
connection := ws.Connect(r, "/websocket")
connection.SendMessage("Drew")
message, _ := connection.ReceiveMessage()
assert.Equal(t, "Hello, Drew", message)
})
}Contributing
Requires go >= 1.11.
Run the tests.
go test -v ./...Now write new tests, fix them and send me a pull request!