diff options
author | =Corey Hulen <corey@hulen.com> | 2015-06-14 23:53:32 -0800 |
---|---|---|
committer | =Corey Hulen <corey@hulen.com> | 2015-06-14 23:53:32 -0800 |
commit | cf7a05f80f68b5b1c8bcc0089679dd497cec2506 (patch) | |
tree | 70007378570a6962d7c175ca96af732b71aeb6da /api/web_socket_test.go | |
download | chat-cf7a05f80f68b5b1c8bcc0089679dd497cec2506.tar.gz chat-cf7a05f80f68b5b1c8bcc0089679dd497cec2506.tar.bz2 chat-cf7a05f80f68b5b1c8bcc0089679dd497cec2506.zip |
first commit
Diffstat (limited to 'api/web_socket_test.go')
-rw-r--r-- | api/web_socket_test.go | 129 |
1 files changed, 129 insertions, 0 deletions
diff --git a/api/web_socket_test.go b/api/web_socket_test.go new file mode 100644 index 000000000..c7b612cde --- /dev/null +++ b/api/web_socket_test.go @@ -0,0 +1,129 @@ +// Copyright (c) 2015 Spinpunch, Inc. All Rights Reserved. +// See License.txt for license information. + +package api + +import ( + "github.com/gorilla/websocket" + "github.com/mattermost/platform/model" + "github.com/mattermost/platform/utils" + "net/http" + "testing" + "time" +) + +func TestSocket(t *testing.T) { + Setup() + + url := "ws://localhost:" + utils.Cfg.ServiceSettings.Port + "/api/v1/websocket" + team := &model.Team{Name: "Name", Domain: "z-z-" + model.NewId() + "a", Email: "test@nowhere.com", Type: model.TEAM_OPEN} + team = Client.Must(Client.CreateTeam(team)).Data.(*model.Team) + + user1 := &model.User{TeamId: team.Id, Email: model.NewId() + "corey@test.com", FullName: "Corey Hulen", Password: "pwd"} + user1 = Client.Must(Client.CreateUser(user1, "")).Data.(*model.User) + Srv.Store.User().VerifyEmail(user1.Id) + Client.LoginByEmail(team.Domain, user1.Email, "pwd") + + channel1 := &model.Channel{DisplayName: "Test Web Scoket 1", Name: "a" + model.NewId() + "a", Type: model.CHANNEL_OPEN, TeamId: team.Id} + channel1 = Client.Must(Client.CreateChannel(channel1)).Data.(*model.Channel) + + channel2 := &model.Channel{DisplayName: "Test Web Scoket 2", Name: "a" + model.NewId() + "a", Type: model.CHANNEL_OPEN, TeamId: team.Id} + channel2 = Client.Must(Client.CreateChannel(channel2)).Data.(*model.Channel) + + header1 := http.Header{} + header1.Set(model.HEADER_AUTH, "BEARER "+Client.AuthToken) + + c1, _, err := websocket.DefaultDialer.Dial(url, header1) + if err != nil { + t.Fatal(err) + } + + user2 := &model.User{TeamId: team.Id, Email: model.NewId() + "corey@test.com", FullName: "Corey Hulen", Password: "pwd"} + user2 = Client.Must(Client.CreateUser(user2, "")).Data.(*model.User) + Srv.Store.User().VerifyEmail(user2.Id) + Client.LoginByEmail(team.Domain, user2.Email, "pwd") + + header2 := http.Header{} + header2.Set(model.HEADER_AUTH, "BEARER "+Client.AuthToken) + + c2, _, err := websocket.DefaultDialer.Dial(url, header2) + if err != nil { + t.Fatal(err) + } + + time.Sleep(300 * time.Millisecond) + Client.Must(Client.JoinChannel(channel1.Id)) + + // Read the join channel message that gets generated + var rmsg model.Message + if err := c2.ReadJSON(&rmsg); err != nil { + t.Fatal(err) + } + + // Test sending message without a channelId + m := model.NewMessage("", "", "", model.ACTION_TYPING) + m.Add("RootId", model.NewId()) + m.Add("ParentId", model.NewId()) + + c1.WriteJSON(m) + + if err := c2.ReadJSON(&rmsg); err != nil { + t.Fatal(err) + } + + if team.Id != rmsg.TeamId { + t.Fatal("Ids do not match") + } + + if m.Props["RootId"] != rmsg.Props["RootId"] { + t.Fatal("Ids do not match") + } + + // Test sending messsage to Channel you have access to + m = model.NewMessage("", channel1.Id, "", model.ACTION_TYPING) + m.Add("RootId", model.NewId()) + m.Add("ParentId", model.NewId()) + + c1.WriteJSON(m) + + if err := c2.ReadJSON(&rmsg); err != nil { + t.Fatal(err) + } + + if team.Id != rmsg.TeamId { + t.Fatal("Ids do not match") + } + + if m.Props["RootId"] != rmsg.Props["RootId"] { + t.Fatal("Ids do not match") + } + + // Test sending message to Channel you *do not* have access too + m = model.NewMessage("", channel2.Id, "", model.ACTION_TYPING) + m.Add("RootId", model.NewId()) + m.Add("ParentId", model.NewId()) + + c1.WriteJSON(m) + + go func() { + if err := c2.ReadJSON(&rmsg); err != nil { + t.Fatal(err) + } + + t.Fatal(err) + }() + + time.Sleep(2 * time.Second) + + hub.Stop(team.Id) + +} + +func TestZZWebScoketTearDown(t *testing.T) { + // *IMPORTANT* - Kind of hacky + // This should be the last function in any test file + // that calls Setup() + // Should be in the last file too sorted by name + time.Sleep(2 * time.Second) + TearDown() +} |