test standard router

This commit is contained in:
Kevin Mulvey 2015-05-29 07:29:33 -04:00
parent b512566acf
commit da5be64c9f
1 changed files with 26 additions and 0 deletions

View File

@ -4,6 +4,7 @@ import (
"bufio" "bufio"
"bytes" "bytes"
"fmt" "fmt"
"io/ioutil"
"net" "net"
"net/http" "net/http"
"testing" "testing"
@ -12,6 +13,31 @@ import (
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
) )
func TestRun(t *testing.T) {
buffer := new(bytes.Buffer)
router := New()
go func() {
router.Use(LoggerWithWriter(buffer))
router.GET("/example", func(c *Context) { c.String(http.StatusOK, "it worked") })
router.Run(":5150")
}()
// have to wait for the goroutine to start and run the server
// otherwise the main thread will complete
time.Sleep(5 * time.Millisecond)
resp, err := http.Get("http://localhost:5150/example")
if err != nil {
t.FailNow()
}
defer resp.Body.Close()
body, ioerr := ioutil.ReadAll(resp.Body)
if ioerr != nil {
t.FailNow()
}
assert.Equal(t, "it worked", body, "resp body should match")
assert.Equal(t, "200 OK", resp.Status, "should get a 200")
}
func TestUnixSocket(t *testing.T) { func TestUnixSocket(t *testing.T) {
buffer := new(bytes.Buffer) buffer := new(bytes.Buffer)
router := New() router := New()