diff --git a/gin_integration_test.go b/gin_integration_test.go new file mode 100644 index 00000000..cb121641 --- /dev/null +++ b/gin_integration_test.go @@ -0,0 +1,66 @@ +package gin + +import ( + "bufio" + "bytes" + "fmt" + "io/ioutil" + "net" + "net/http" + "testing" + "time" + + "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", string(body[:]), "resp body should match") + assert.Equal(t, "200 OK", resp.Status, "should get a 200") +} + +func TestUnixSocket(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.RunUnix("/tmp/unix_unit_test") + }() + // have to wait for the goroutine to start and run the server + // otherwise the main thread will complete + time.Sleep(5 * time.Millisecond) + + c, err := net.Dial("unix", "/tmp/unix_unit_test") + if err != nil { + println(err) + t.FailNow() + } + fmt.Fprintf(c, "GET /example HTTP/1.0\r\n\r\n") + scanner := bufio.NewScanner(c) + var response string + for scanner.Scan() { + response += scanner.Text() + } + assert.Contains(t, response, "HTTP/1.0 200", "should get a 200") + assert.Contains(t, response, "it worked", "resp body should match") +}