/
websocket_app.go
42 lines (30 loc) · 1.07 KB
/
websocket_app.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
package test
import (
"net/http"
"time"
"github.com/nats-io/go-nats"
"github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"code.cloudfoundry.org/gorouter/route"
"code.cloudfoundry.org/gorouter/test/common"
"code.cloudfoundry.org/gorouter/test_util"
)
func NewWebSocketApp(urls []route.Uri, rPort uint16, mbusClient *nats.Conn, delay time.Duration, routeServiceUrl string) *common.TestApp {
app := common.NewTestApp(urls, rPort, mbusClient, nil, routeServiceUrl)
app.AddHandler("/", func(w http.ResponseWriter, r *http.Request) {
defer ginkgo.GinkgoRecover()
Expect(r.Header.Get("Upgrade")).To(Equal("websocket"))
Expect(r.Header.Get("Connection")).To(Equal("upgrade"))
conn, _, err := w.(http.Hijacker).Hijack()
x := test_util.NewHttpConn(conn)
resp := test_util.NewResponse(http.StatusSwitchingProtocols)
resp.Header.Set("Upgrade", "websocket")
resp.Header.Set("Connection", "upgrade")
time.Sleep(delay)
x.WriteResponse(resp)
Expect(err).ToNot(HaveOccurred())
x.CheckLine("hello from client")
x.WriteLine("hello from server")
})
return app
}