mirror of
https://github.com/mattermost/mattermost.git
synced 2025-02-25 18:55:24 -06:00
5dd2e75c10
* extracting i18n functionality to i18n core library * Removing utils.T * Adding documentation and changing one function name for better explanation * Changing other missing utils.T * Adding license string * Renaming corelibs to pkg * Renaming corelibs to pkg (moving directory) * Renaming from pkg to shared * Fixing bodyPage.Html casing * Fixing merges * Fixing merge problem * Fixing tests
101 lines
3.4 KiB
Go
101 lines
3.4 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package app
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/mattermost/mattermost-server/v5/model"
|
|
"github.com/mattermost/mattermost-server/v5/shared/i18n"
|
|
)
|
|
|
|
func TestWebConnShouldSendEvent(t *testing.T) {
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
session, err := th.App.CreateSession(&model.Session{UserId: th.BasicUser.Id, Roles: th.BasicUser.GetRawRoles(), TeamMembers: []*model.TeamMember{
|
|
{
|
|
UserId: th.BasicUser.Id,
|
|
TeamId: th.BasicTeam.Id,
|
|
Roles: model.TEAM_USER_ROLE_ID,
|
|
},
|
|
}})
|
|
require.Nil(t, err)
|
|
|
|
basicUserWc := &WebConn{
|
|
App: th.App,
|
|
UserId: th.BasicUser.Id,
|
|
T: i18n.T,
|
|
}
|
|
|
|
basicUserWc.SetSession(session)
|
|
basicUserWc.SetSessionToken(session.Token)
|
|
basicUserWc.SetSessionExpiresAt(session.ExpiresAt)
|
|
|
|
session2, err := th.App.CreateSession(&model.Session{UserId: th.BasicUser2.Id, Roles: th.BasicUser2.GetRawRoles(), TeamMembers: []*model.TeamMember{
|
|
{
|
|
UserId: th.BasicUser2.Id,
|
|
TeamId: th.BasicTeam.Id,
|
|
Roles: model.TEAM_ADMIN_ROLE_ID,
|
|
},
|
|
}})
|
|
require.Nil(t, err)
|
|
|
|
basicUser2Wc := &WebConn{
|
|
App: th.App,
|
|
UserId: th.BasicUser2.Id,
|
|
T: i18n.T,
|
|
}
|
|
|
|
basicUser2Wc.SetSession(session2)
|
|
basicUser2Wc.SetSessionToken(session2.Token)
|
|
basicUser2Wc.SetSessionExpiresAt(session2.ExpiresAt)
|
|
|
|
session3, err := th.App.CreateSession(&model.Session{UserId: th.SystemAdminUser.Id, Roles: th.SystemAdminUser.GetRawRoles()})
|
|
require.Nil(t, err)
|
|
|
|
adminUserWc := &WebConn{
|
|
App: th.App,
|
|
UserId: th.SystemAdminUser.Id,
|
|
T: i18n.T,
|
|
}
|
|
|
|
adminUserWc.SetSession(session3)
|
|
adminUserWc.SetSessionToken(session3.Token)
|
|
adminUserWc.SetSessionExpiresAt(session3.ExpiresAt)
|
|
|
|
cases := []struct {
|
|
Description string
|
|
Broadcast *model.WebsocketBroadcast
|
|
User1Expected bool
|
|
User2Expected bool
|
|
AdminExpected bool
|
|
}{
|
|
{"should send to all", &model.WebsocketBroadcast{}, true, true, true},
|
|
{"should only send to basic user", &model.WebsocketBroadcast{UserId: th.BasicUser.Id}, true, false, false},
|
|
{"should omit basic user 2", &model.WebsocketBroadcast{OmitUsers: map[string]bool{th.BasicUser2.Id: true}}, true, false, true},
|
|
{"should only send to admin", &model.WebsocketBroadcast{ContainsSensitiveData: true}, false, false, true},
|
|
{"should only send to non-admins", &model.WebsocketBroadcast{ContainsSanitizedData: true}, true, true, false},
|
|
{"should send to nobody", &model.WebsocketBroadcast{ContainsSensitiveData: true, ContainsSanitizedData: true}, false, false, false},
|
|
// needs more cases to get full coverage
|
|
}
|
|
|
|
event := model.NewWebSocketEvent("some_event", "", "", "", nil)
|
|
for _, c := range cases {
|
|
event = event.SetBroadcast(c.Broadcast)
|
|
assert.Equal(t, c.User1Expected, basicUserWc.shouldSendEvent(event), c.Description)
|
|
assert.Equal(t, c.User2Expected, basicUser2Wc.shouldSendEvent(event), c.Description)
|
|
assert.Equal(t, c.AdminExpected, adminUserWc.shouldSendEvent(event), c.Description)
|
|
}
|
|
|
|
event2 := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_UPDATE_TEAM, th.BasicTeam.Id, "", "", nil)
|
|
assert.True(t, basicUserWc.shouldSendEvent(event2))
|
|
assert.True(t, basicUser2Wc.shouldSendEvent(event2))
|
|
|
|
event3 := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_UPDATE_TEAM, "wrongId", "", "", nil)
|
|
assert.False(t, basicUserWc.shouldSendEvent(event3))
|
|
}
|