diff options
author | Chris <ccbrown112@gmail.com> | 2017-11-13 13:59:51 -0600 |
---|---|---|
committer | Christopher Speller <crspeller@gmail.com> | 2017-11-13 11:59:51 -0800 |
commit | 865f9f83a7cf2f9eb1fe2735b6bd498f56486f50 (patch) | |
tree | 2554279f7682599ca69370557194dfa984557db3 /app/web_hub_test.go | |
parent | 1329aa51b605cb54ba9aae3a82a0a87b881fb7b3 (diff) | |
download | chat-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.go | 60 |
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() +} |