summaryrefslogtreecommitdiffstats
path: root/app/web_hub_test.go
diff options
context:
space:
mode:
authorChris <ccbrown112@gmail.com>2017-11-13 13:59:51 -0600
committerChristopher Speller <crspeller@gmail.com>2017-11-13 11:59:51 -0800
commit865f9f83a7cf2f9eb1fe2735b6bd498f56486f50 (patch)
tree2554279f7682599ca69370557194dfa984557db3 /app/web_hub_test.go
parent1329aa51b605cb54ba9aae3a82a0a87b881fb7b3 (diff)
downloadchat-865f9f83a7cf2f9eb1fe2735b6bd498f56486f50.tar.gz
chat-865f9f83a7cf2f9eb1fe2735b6bd498f56486f50.tar.bz2
chat-865f9f83a7cf2f9eb1fe2735b6bd498f56486f50.zip
Improve test coverage reporting / accuracy (#7819)
* improve test coverage reporting / accuracy * handle absolute coverpaths * move tests into multiple files * rename codecov.yml (https://github.com/codecov/support/issues/426)
Diffstat (limited to 'app/web_hub_test.go')
-rw-r--r--app/web_hub_test.go60
1 files changed, 60 insertions, 0 deletions
diff --git a/app/web_hub_test.go b/app/web_hub_test.go
new file mode 100644
index 000000000..62c2552de
--- /dev/null
+++ b/app/web_hub_test.go
@@ -0,0 +1,60 @@
+package app
+
+import (
+ "net"
+ "net/http"
+ "net/http/httptest"
+ "testing"
+
+ "github.com/gorilla/websocket"
+ goi18n "github.com/nicksnyder/go-i18n/i18n"
+ "github.com/stretchr/testify/require"
+
+ "github.com/mattermost/mattermost-server/model"
+)
+
+func dummyWebsocketHandler(t *testing.T) http.HandlerFunc {
+ return func(w http.ResponseWriter, req *http.Request) {
+ upgrader := &websocket.Upgrader{
+ ReadBufferSize: 1024,
+ WriteBufferSize: 1024,
+ }
+ conn, err := upgrader.Upgrade(w, req, nil)
+ for err == nil {
+ _, _, err = conn.ReadMessage()
+ }
+ if _, ok := err.(*websocket.CloseError); !ok {
+ require.NoError(t, err)
+ }
+ }
+}
+
+func registerDummyWebConn(t *testing.T, a *App, addr net.Addr, userId string) *WebConn {
+ session, appErr := a.CreateSession(&model.Session{
+ UserId: userId,
+ })
+ require.Nil(t, appErr)
+
+ d := websocket.Dialer{}
+ c, _, err := d.Dial("ws://"+addr.String()+"/ws", nil)
+ require.NoError(t, err)
+
+ wc := a.NewWebConn(c, *session, goi18n.IdentityTfunc(), "en")
+ a.HubRegister(wc)
+ go wc.Pump()
+ return wc
+}
+
+func TestHubStopWithMultipleConnections(t *testing.T) {
+ th := Setup().InitBasic()
+ defer th.TearDown()
+
+ s := httptest.NewServer(http.HandlerFunc(dummyWebsocketHandler(t)))
+ defer s.Close()
+
+ th.App.HubStart()
+ registerDummyWebConn(t, th.App, s.Listener.Addr(), th.BasicUser.Id)
+ registerDummyWebConn(t, th.App, s.Listener.Addr(), th.BasicUser.Id)
+ registerDummyWebConn(t, th.App, s.Listener.Addr(), th.BasicUser.Id)
+ th.App.HubStop()
+}