mirror of
https://github.com/mattermost/mattermost.git
synced 2025-02-25 18:55:24 -06:00
2e5617c29b
* Adding TeamMember to system * Fixing all unit tests on the backend * Fixing merge conflicts * Fixing merge conflict * Adding javascript unit tests * Adding TeamMember to system * Fixing all unit tests on the backend * Fixing merge conflicts * Fixing merge conflict * Adding javascript unit tests * Adding client side unit test * Cleaning up the clint side tests * Fixing msg * Adding more client side unit tests * Adding more using tests * Adding last bit of client side unit tests and adding make cmd * Fixing bad merge * Fixing libraries * Updating to new client side API * Fixing borken unit test * Fixing unit tests * ugg...trying to beat gofmt * ugg...trying to beat gofmt * Cleaning up remainder of the server side routes * Adding inital load api * Increased coverage of webhook unit tests (#2660) * Adding loading ... to root html * Fixing bad merge * Removing explicit content type so superagent will guess corectly (#2685) * Fixing merge and unit tests * Adding create team UI * Fixing signup flows * Adding LDAP unit tests and enterprise unit test helper (#2702) * Add the ability to reset MFA from the commandline (#2706) * Fixing compliance unit tests * Fixing client side tests * Adding open server to system console * Moving websocket connection * Fixing unit test * Fixing unit tests * Fixing unit tests * Adding nickname and more LDAP unit tests (#2717) * Adding join open teams * Cleaning up all TODOs in the code * Fixing web sockets * Removing unused webockets file * PLT-2533 Add the ability to reset a user's MFA from the system console (#2715) * Add the ability to reset a user's MFA from the system console * Add client side unit test for adminResetMfa * Reorganizing authentication to fix LDAP error message (#2723) * Fixing failing unit test * Initial upgrade db code * Adding upgrade script * Fixing upgrade script after running on core * Update OAuth and Claim routes to work with user model changes (#2739) * Fixing perminant deletion. Adding ability to delete all user and the entire database (#2740) * Fixing team invite ldap login call (#2741) * Fixing bluebar and some img stuff * Fix all the different file upload web utils (#2743) * Fixing invalid session redirect (#2744) * Redirect on bad channel name (#2746) * Fixing a bunch of issue and removing dead code * Patch to fix error message on leave channel (#2747) * Setting EnableOpenServer to false by default * Fixing config * Fixing upgrade * Fixing reported bugs * Bug fixes for PLT-2057 * PLT-2563 Redo password recovery to use a database table (#2745) * Redo password recovery to use a database table * Update reset password audits * Split out admin and user reset password APIs to be separate * Delete password recovery when user is permanently deleted * Consolidate password resetting into a single function * Removed private channels as an option for outgoing webhooks (#2752) * PLT-2577/PLT-2552 Fixes for backstage (#2753) * Added URL to incoming webhook list * Fixed client functions for adding/removing integrations * Disallowed slash commands without trigger words * Fixed clientside handling of errors on AddCommand page * Minor auth cleanup (#2758) * Changed EditPostModal to just close if you save without making any changes (#2759) * Renamed client -> Client in async_client.jsx and fixed eslint warnings (#2756) * Fixed url in channel info modal (#2755) * Fixing reported issues * Moving to version 3 of the apis * Fixing command unit tests (#2760) * Adding team admins * Fixing DM issue * Fixing eslint error * Properly set EditPostModal's originalText state in all cases (#2762) * Update client config check to assume features is defined if server is licensed (#2772) * Fixing url link * Fixing issue with websocket crashing when sending messages to different teams
950 lines
24 KiB
Go
950 lines
24 KiB
Go
// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved.
|
|
// See License.txt for license information.
|
|
|
|
package store
|
|
|
|
import (
|
|
"github.com/mattermost/platform/model"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestChannelStoreSave(t *testing.T) {
|
|
Setup()
|
|
|
|
teamId := model.NewId()
|
|
|
|
o1 := model.Channel{}
|
|
o1.TeamId = teamId
|
|
o1.DisplayName = "Name"
|
|
o1.Name = "a" + model.NewId() + "b"
|
|
o1.Type = model.CHANNEL_OPEN
|
|
|
|
if err := (<-store.Channel().Save(&o1)).Err; err != nil {
|
|
t.Fatal("couldn't save item", err)
|
|
}
|
|
|
|
if err := (<-store.Channel().Save(&o1)).Err; err == nil {
|
|
t.Fatal("shouldn't be able to update from save")
|
|
}
|
|
|
|
o1.Id = ""
|
|
if err := (<-store.Channel().Save(&o1)).Err; err == nil {
|
|
t.Fatal("should be unique name")
|
|
}
|
|
|
|
o1.Id = ""
|
|
o1.Name = "a" + model.NewId() + "b"
|
|
o1.Type = model.CHANNEL_DIRECT
|
|
if err := (<-store.Channel().Save(&o1)).Err; err == nil {
|
|
t.Fatal("Should not be able to save direct channel")
|
|
}
|
|
|
|
o1.Type = model.CHANNEL_OPEN
|
|
for i := 0; i < 1000; i++ {
|
|
o1.Id = ""
|
|
o1.Name = "a" + model.NewId() + "b"
|
|
if err := (<-store.Channel().Save(&o1)).Err; err != nil {
|
|
t.Fatal("couldn't save item", err)
|
|
}
|
|
}
|
|
|
|
o1.Id = ""
|
|
o1.Name = "a" + model.NewId() + "b"
|
|
if err := (<-store.Channel().Save(&o1)).Err; err == nil {
|
|
t.Fatal("should be the limit")
|
|
}
|
|
}
|
|
|
|
func TestChannelStoreSaveDirectChannel(t *testing.T) {
|
|
Setup()
|
|
|
|
teamId := model.NewId()
|
|
|
|
o1 := model.Channel{}
|
|
o1.TeamId = teamId
|
|
o1.DisplayName = "Name"
|
|
o1.Name = "a" + model.NewId() + "b"
|
|
o1.Type = model.CHANNEL_DIRECT
|
|
|
|
u1 := &model.User{}
|
|
u1.Email = model.NewId()
|
|
u1.Nickname = model.NewId()
|
|
Must(store.User().Save(u1))
|
|
Must(store.Team().SaveMember(&model.TeamMember{TeamId: model.NewId(), UserId: u1.Id}))
|
|
|
|
u2 := &model.User{}
|
|
u2.Email = model.NewId()
|
|
u2.Nickname = model.NewId()
|
|
Must(store.User().Save(u2))
|
|
Must(store.Team().SaveMember(&model.TeamMember{TeamId: model.NewId(), UserId: u2.Id}))
|
|
|
|
m1 := model.ChannelMember{}
|
|
m1.ChannelId = o1.Id
|
|
m1.UserId = u1.Id
|
|
m1.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
|
|
m2 := model.ChannelMember{}
|
|
m2.ChannelId = o1.Id
|
|
m2.UserId = u2.Id
|
|
m2.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
|
|
if err := (<-store.Channel().SaveDirectChannel(&o1, &m1, &m2)).Err; err != nil {
|
|
t.Fatal("couldn't save direct channel", err)
|
|
}
|
|
|
|
members := (<-store.Channel().GetMembers(o1.Id)).Data.([]model.ChannelMember)
|
|
if len(members) != 2 {
|
|
t.Fatal("should have saved 2 members")
|
|
}
|
|
|
|
if err := (<-store.Channel().SaveDirectChannel(&o1, &m1, &m2)).Err; err == nil {
|
|
t.Fatal("shouldn't be able to update from save")
|
|
}
|
|
|
|
o1.Id = ""
|
|
o1.Name = "a" + model.NewId() + "b"
|
|
o1.Type = model.CHANNEL_OPEN
|
|
if err := (<-store.Channel().SaveDirectChannel(&o1, &m1, &m2)).Err; err == nil {
|
|
t.Fatal("Should not be able to save non-direct channel")
|
|
}
|
|
|
|
}
|
|
|
|
func TestChannelStoreUpdate(t *testing.T) {
|
|
Setup()
|
|
|
|
o1 := model.Channel{}
|
|
o1.TeamId = model.NewId()
|
|
o1.DisplayName = "Name"
|
|
o1.Name = "a" + model.NewId() + "b"
|
|
o1.Type = model.CHANNEL_OPEN
|
|
|
|
if err := (<-store.Channel().Save(&o1)).Err; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
time.Sleep(100 * time.Millisecond)
|
|
|
|
if err := (<-store.Channel().Update(&o1)).Err; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
o1.Id = "missing"
|
|
if err := (<-store.Channel().Update(&o1)).Err; err == nil {
|
|
t.Fatal("Update should have failed because of missing key")
|
|
}
|
|
|
|
o1.Id = model.NewId()
|
|
if err := (<-store.Channel().Update(&o1)).Err; err == nil {
|
|
t.Fatal("Update should have faile because id change")
|
|
}
|
|
}
|
|
|
|
func TestChannelStoreGet(t *testing.T) {
|
|
Setup()
|
|
|
|
o1 := model.Channel{}
|
|
o1.TeamId = model.NewId()
|
|
o1.DisplayName = "Name"
|
|
o1.Name = "a" + model.NewId() + "b"
|
|
o1.Type = model.CHANNEL_OPEN
|
|
Must(store.Channel().Save(&o1))
|
|
|
|
if r1 := <-store.Channel().Get(o1.Id); r1.Err != nil {
|
|
t.Fatal(r1.Err)
|
|
} else {
|
|
if r1.Data.(*model.Channel).ToJson() != o1.ToJson() {
|
|
t.Fatal("invalid returned channel")
|
|
}
|
|
}
|
|
|
|
if err := (<-store.Channel().Get("")).Err; err == nil {
|
|
t.Fatal("Missing id should have failed")
|
|
}
|
|
|
|
u1 := &model.User{}
|
|
u1.Email = model.NewId()
|
|
u1.Nickname = model.NewId()
|
|
Must(store.User().Save(u1))
|
|
Must(store.Team().SaveMember(&model.TeamMember{TeamId: model.NewId(), UserId: u1.Id}))
|
|
|
|
u2 := model.User{}
|
|
u2.Email = model.NewId()
|
|
u2.Nickname = model.NewId()
|
|
Must(store.User().Save(&u2))
|
|
Must(store.Team().SaveMember(&model.TeamMember{TeamId: model.NewId(), UserId: u2.Id}))
|
|
|
|
o2 := model.Channel{}
|
|
o2.TeamId = model.NewId()
|
|
o2.DisplayName = "Direct Name"
|
|
o2.Name = "a" + model.NewId() + "b"
|
|
o2.Type = model.CHANNEL_DIRECT
|
|
|
|
m1 := model.ChannelMember{}
|
|
m1.ChannelId = o2.Id
|
|
m1.UserId = u1.Id
|
|
m1.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
|
|
m2 := model.ChannelMember{}
|
|
m2.ChannelId = o2.Id
|
|
m2.UserId = u2.Id
|
|
m2.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
|
|
Must(store.Channel().SaveDirectChannel(&o2, &m1, &m2))
|
|
|
|
if r2 := <-store.Channel().Get(o2.Id); r2.Err != nil {
|
|
t.Fatal(r2.Err)
|
|
} else {
|
|
if r2.Data.(*model.Channel).ToJson() != o2.ToJson() {
|
|
t.Fatal("invalid returned channel")
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestChannelStoreDelete(t *testing.T) {
|
|
Setup()
|
|
|
|
o1 := model.Channel{}
|
|
o1.TeamId = model.NewId()
|
|
o1.DisplayName = "Channel1"
|
|
o1.Name = "a" + model.NewId() + "b"
|
|
o1.Type = model.CHANNEL_OPEN
|
|
Must(store.Channel().Save(&o1))
|
|
|
|
o2 := model.Channel{}
|
|
o2.TeamId = o1.TeamId
|
|
o2.DisplayName = "Channel2"
|
|
o2.Name = "a" + model.NewId() + "b"
|
|
o2.Type = model.CHANNEL_OPEN
|
|
Must(store.Channel().Save(&o2))
|
|
|
|
o3 := model.Channel{}
|
|
o3.TeamId = o1.TeamId
|
|
o3.DisplayName = "Channel3"
|
|
o3.Name = "a" + model.NewId() + "b"
|
|
o3.Type = model.CHANNEL_OPEN
|
|
Must(store.Channel().Save(&o3))
|
|
|
|
o4 := model.Channel{}
|
|
o4.TeamId = o1.TeamId
|
|
o4.DisplayName = "Channel4"
|
|
o4.Name = "a" + model.NewId() + "b"
|
|
o4.Type = model.CHANNEL_OPEN
|
|
Must(store.Channel().Save(&o4))
|
|
|
|
m1 := model.ChannelMember{}
|
|
m1.ChannelId = o1.Id
|
|
m1.UserId = model.NewId()
|
|
m1.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
Must(store.Channel().SaveMember(&m1))
|
|
|
|
m2 := model.ChannelMember{}
|
|
m2.ChannelId = o2.Id
|
|
m2.UserId = m1.UserId
|
|
m2.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
Must(store.Channel().SaveMember(&m2))
|
|
|
|
if r := <-store.Channel().Delete(o1.Id, model.GetMillis()); r.Err != nil {
|
|
t.Fatal(r.Err)
|
|
}
|
|
|
|
if r := <-store.Channel().Get(o1.Id); r.Data.(*model.Channel).DeleteAt == 0 {
|
|
t.Fatal("should have been deleted")
|
|
}
|
|
|
|
if r := <-store.Channel().Delete(o3.Id, model.GetMillis()); r.Err != nil {
|
|
t.Fatal(r.Err)
|
|
}
|
|
|
|
cresult := <-store.Channel().GetChannels(o1.TeamId, m1.UserId)
|
|
list := cresult.Data.(*model.ChannelList)
|
|
|
|
if len(list.Channels) != 1 {
|
|
t.Fatal("invalid number of channels")
|
|
}
|
|
|
|
cresult = <-store.Channel().GetMoreChannels(o1.TeamId, m1.UserId)
|
|
list = cresult.Data.(*model.ChannelList)
|
|
|
|
if len(list.Channels) != 1 {
|
|
t.Fatal("invalid number of channels")
|
|
}
|
|
}
|
|
|
|
func TestChannelStoreGetByName(t *testing.T) {
|
|
Setup()
|
|
|
|
o1 := model.Channel{}
|
|
o1.TeamId = model.NewId()
|
|
o1.DisplayName = "Name"
|
|
o1.Name = "a" + model.NewId() + "b"
|
|
o1.Type = model.CHANNEL_OPEN
|
|
Must(store.Channel().Save(&o1))
|
|
|
|
if r1 := <-store.Channel().GetByName(o1.TeamId, o1.Name); r1.Err != nil {
|
|
t.Fatal(r1.Err)
|
|
} else {
|
|
if r1.Data.(*model.Channel).ToJson() != o1.ToJson() {
|
|
t.Fatal("invalid returned channel")
|
|
}
|
|
}
|
|
|
|
if err := (<-store.Channel().GetByName(o1.TeamId, "")).Err; err == nil {
|
|
t.Fatal("Missing id should have failed")
|
|
}
|
|
}
|
|
|
|
func TestChannelMemberStore(t *testing.T) {
|
|
Setup()
|
|
|
|
c1 := model.Channel{}
|
|
c1.TeamId = model.NewId()
|
|
c1.DisplayName = "NameName"
|
|
c1.Name = "a" + model.NewId() + "b"
|
|
c1.Type = model.CHANNEL_OPEN
|
|
c1 = *Must(store.Channel().Save(&c1)).(*model.Channel)
|
|
|
|
c1t1 := (<-store.Channel().Get(c1.Id)).Data.(*model.Channel)
|
|
t1 := c1t1.ExtraUpdateAt
|
|
|
|
u1 := model.User{}
|
|
u1.Email = model.NewId()
|
|
u1.Nickname = model.NewId()
|
|
Must(store.User().Save(&u1))
|
|
Must(store.Team().SaveMember(&model.TeamMember{TeamId: model.NewId(), UserId: u1.Id}))
|
|
|
|
u2 := model.User{}
|
|
u2.Email = model.NewId()
|
|
u2.Nickname = model.NewId()
|
|
Must(store.User().Save(&u2))
|
|
Must(store.Team().SaveMember(&model.TeamMember{TeamId: model.NewId(), UserId: u2.Id}))
|
|
|
|
o1 := model.ChannelMember{}
|
|
o1.ChannelId = c1.Id
|
|
o1.UserId = u1.Id
|
|
o1.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
Must(store.Channel().SaveMember(&o1))
|
|
|
|
o2 := model.ChannelMember{}
|
|
o2.ChannelId = c1.Id
|
|
o2.UserId = u2.Id
|
|
o2.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
Must(store.Channel().SaveMember(&o2))
|
|
|
|
c1t2 := (<-store.Channel().Get(c1.Id)).Data.(*model.Channel)
|
|
t2 := c1t2.ExtraUpdateAt
|
|
|
|
if t2 <= t1 {
|
|
t.Fatal("Member update time incorrect")
|
|
}
|
|
|
|
count := (<-store.Channel().GetMemberCount(o1.ChannelId)).Data.(int64)
|
|
if count != 2 {
|
|
t.Fatal("should have saved 2 members")
|
|
}
|
|
|
|
Must(store.Channel().RemoveMember(o2.ChannelId, o2.UserId))
|
|
|
|
count = (<-store.Channel().GetMemberCount(o1.ChannelId)).Data.(int64)
|
|
if count != 1 {
|
|
t.Fatal("should have removed 1 member")
|
|
}
|
|
|
|
c1t3 := (<-store.Channel().Get(c1.Id)).Data.(*model.Channel)
|
|
t3 := c1t3.ExtraUpdateAt
|
|
|
|
if t3 <= t2 || t3 <= t1 {
|
|
t.Fatal("Member update time incorrect on delete")
|
|
}
|
|
|
|
member := (<-store.Channel().GetMember(o1.ChannelId, o1.UserId)).Data.(model.ChannelMember)
|
|
if member.ChannelId != o1.ChannelId {
|
|
t.Fatal("should have go member")
|
|
}
|
|
|
|
extraMembers := (<-store.Channel().GetExtraMembers(o1.ChannelId, 20)).Data.([]model.ExtraMember)
|
|
if len(extraMembers) != 1 {
|
|
t.Fatal("should have 1 extra members")
|
|
}
|
|
|
|
if err := (<-store.Channel().SaveMember(&o1)).Err; err == nil {
|
|
t.Fatal("Should have been a duplicate")
|
|
}
|
|
|
|
c1t4 := (<-store.Channel().Get(c1.Id)).Data.(*model.Channel)
|
|
t4 := c1t4.ExtraUpdateAt
|
|
if t4 != t3 {
|
|
t.Fatal("Should not update time upon failure")
|
|
}
|
|
|
|
// rejoin the channel and make sure that an inactive user isn't returned by GetExtraMambers
|
|
Must(store.Channel().SaveMember(&o2))
|
|
|
|
u2.DeleteAt = 1000
|
|
Must(store.User().Update(&u2, true))
|
|
|
|
if result := <-store.Channel().GetExtraMembers(o1.ChannelId, 20); result.Err != nil {
|
|
t.Fatal(result.Err)
|
|
} else if extraMembers := result.Data.([]model.ExtraMember); len(extraMembers) != 1 {
|
|
t.Fatal("should have 1 extra members")
|
|
}
|
|
}
|
|
|
|
func TestChannelDeleteMemberStore(t *testing.T) {
|
|
Setup()
|
|
|
|
c1 := model.Channel{}
|
|
c1.TeamId = model.NewId()
|
|
c1.DisplayName = "NameName"
|
|
c1.Name = "a" + model.NewId() + "b"
|
|
c1.Type = model.CHANNEL_OPEN
|
|
c1 = *Must(store.Channel().Save(&c1)).(*model.Channel)
|
|
|
|
c1t1 := (<-store.Channel().Get(c1.Id)).Data.(*model.Channel)
|
|
t1 := c1t1.ExtraUpdateAt
|
|
|
|
u1 := model.User{}
|
|
u1.Email = model.NewId()
|
|
u1.Nickname = model.NewId()
|
|
Must(store.User().Save(&u1))
|
|
Must(store.Team().SaveMember(&model.TeamMember{TeamId: model.NewId(), UserId: u1.Id}))
|
|
|
|
u2 := model.User{}
|
|
u2.Email = model.NewId()
|
|
u2.Nickname = model.NewId()
|
|
Must(store.User().Save(&u2))
|
|
Must(store.Team().SaveMember(&model.TeamMember{TeamId: model.NewId(), UserId: u2.Id}))
|
|
|
|
o1 := model.ChannelMember{}
|
|
o1.ChannelId = c1.Id
|
|
o1.UserId = u1.Id
|
|
o1.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
Must(store.Channel().SaveMember(&o1))
|
|
|
|
o2 := model.ChannelMember{}
|
|
o2.ChannelId = c1.Id
|
|
o2.UserId = u2.Id
|
|
o2.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
Must(store.Channel().SaveMember(&o2))
|
|
|
|
c1t2 := (<-store.Channel().Get(c1.Id)).Data.(*model.Channel)
|
|
t2 := c1t2.ExtraUpdateAt
|
|
|
|
if t2 <= t1 {
|
|
t.Fatal("Member update time incorrect")
|
|
}
|
|
|
|
count := (<-store.Channel().GetMemberCount(o1.ChannelId)).Data.(int64)
|
|
if count != 2 {
|
|
t.Fatal("should have saved 2 members")
|
|
}
|
|
|
|
Must(store.Channel().PermanentDeleteMembersByUser(o2.UserId))
|
|
|
|
count = (<-store.Channel().GetMemberCount(o1.ChannelId)).Data.(int64)
|
|
if count != 1 {
|
|
t.Fatal("should have removed 1 member")
|
|
}
|
|
}
|
|
|
|
func TestChannelStorePermissionsTo(t *testing.T) {
|
|
Setup()
|
|
|
|
o1 := model.Channel{}
|
|
o1.TeamId = model.NewId()
|
|
o1.DisplayName = "Channel1"
|
|
o1.Name = "a" + model.NewId() + "b"
|
|
o1.Type = model.CHANNEL_OPEN
|
|
Must(store.Channel().Save(&o1))
|
|
|
|
m1 := model.ChannelMember{}
|
|
m1.ChannelId = o1.Id
|
|
m1.UserId = model.NewId()
|
|
m1.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
Must(store.Channel().SaveMember(&m1))
|
|
|
|
count := (<-store.Channel().CheckPermissionsTo(o1.TeamId, o1.Id, m1.UserId)).Data.(int64)
|
|
if count != 1 {
|
|
t.Fatal("should have permissions")
|
|
}
|
|
|
|
count = (<-store.Channel().CheckPermissionsToNoTeam(o1.Id, m1.UserId)).Data.(int64)
|
|
if count != 1 {
|
|
t.Fatal("should have permissions")
|
|
}
|
|
|
|
count = (<-store.Channel().CheckPermissionsTo("junk", o1.Id, m1.UserId)).Data.(int64)
|
|
if count != 0 {
|
|
t.Fatal("shouldn't have permissions")
|
|
}
|
|
|
|
count = (<-store.Channel().CheckPermissionsTo(o1.TeamId, "junk", m1.UserId)).Data.(int64)
|
|
if count != 0 {
|
|
t.Fatal("shouldn't have permissions")
|
|
}
|
|
|
|
count = (<-store.Channel().CheckPermissionsToNoTeam("junk", m1.UserId)).Data.(int64)
|
|
if count != 0 {
|
|
t.Fatal("shouldn't have permissions")
|
|
}
|
|
|
|
count = (<-store.Channel().CheckPermissionsTo(o1.TeamId, o1.Id, "junk")).Data.(int64)
|
|
if count != 0 {
|
|
t.Fatal("shouldn't have permissions")
|
|
}
|
|
|
|
count = (<-store.Channel().CheckPermissionsToNoTeam(o1.Id, "junk")).Data.(int64)
|
|
if count != 0 {
|
|
t.Fatal("shouldn't have permissions")
|
|
}
|
|
|
|
channelId := (<-store.Channel().CheckPermissionsToByName(o1.TeamId, o1.Name, m1.UserId)).Data.(string)
|
|
if channelId != o1.Id {
|
|
t.Fatal("should have permissions")
|
|
}
|
|
|
|
channelId = (<-store.Channel().CheckPermissionsToByName(o1.TeamId, "missing", m1.UserId)).Data.(string)
|
|
if channelId != "" {
|
|
t.Fatal("should not have permissions")
|
|
}
|
|
}
|
|
|
|
func TestChannelStoreOpenChannelPermissionsTo(t *testing.T) {
|
|
Setup()
|
|
|
|
o1 := model.Channel{}
|
|
o1.TeamId = model.NewId()
|
|
o1.DisplayName = "Channel1"
|
|
o1.Name = "a" + model.NewId() + "b"
|
|
o1.Type = model.CHANNEL_OPEN
|
|
Must(store.Channel().Save(&o1))
|
|
|
|
count := (<-store.Channel().CheckOpenChannelPermissions(o1.TeamId, o1.Id)).Data.(int64)
|
|
if count != 1 {
|
|
t.Fatal("should have permissions")
|
|
}
|
|
|
|
count = (<-store.Channel().CheckOpenChannelPermissions("junk", o1.Id)).Data.(int64)
|
|
if count != 0 {
|
|
t.Fatal("shouldn't have permissions")
|
|
}
|
|
|
|
count = (<-store.Channel().CheckOpenChannelPermissions(o1.TeamId, "junk")).Data.(int64)
|
|
if count != 0 {
|
|
t.Fatal("shouldn't have permissions")
|
|
}
|
|
}
|
|
|
|
func TestChannelStoreGetChannels(t *testing.T) {
|
|
Setup()
|
|
|
|
o2 := model.Channel{}
|
|
o2.TeamId = model.NewId()
|
|
o2.DisplayName = "Channel2"
|
|
o2.Name = "a" + model.NewId() + "b"
|
|
o2.Type = model.CHANNEL_OPEN
|
|
Must(store.Channel().Save(&o2))
|
|
|
|
o1 := model.Channel{}
|
|
o1.TeamId = model.NewId()
|
|
o1.DisplayName = "Channel1"
|
|
o1.Name = "a" + model.NewId() + "b"
|
|
o1.Type = model.CHANNEL_OPEN
|
|
Must(store.Channel().Save(&o1))
|
|
|
|
m1 := model.ChannelMember{}
|
|
m1.ChannelId = o1.Id
|
|
m1.UserId = model.NewId()
|
|
m1.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
Must(store.Channel().SaveMember(&m1))
|
|
|
|
m2 := model.ChannelMember{}
|
|
m2.ChannelId = o1.Id
|
|
m2.UserId = model.NewId()
|
|
m2.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
Must(store.Channel().SaveMember(&m2))
|
|
|
|
m3 := model.ChannelMember{}
|
|
m3.ChannelId = o2.Id
|
|
m3.UserId = model.NewId()
|
|
m3.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
Must(store.Channel().SaveMember(&m3))
|
|
|
|
cresult := <-store.Channel().GetChannels(o1.TeamId, m1.UserId)
|
|
list := cresult.Data.(*model.ChannelList)
|
|
|
|
if list.Channels[0].Id != o1.Id {
|
|
t.Fatal("missing channel")
|
|
}
|
|
}
|
|
|
|
func TestChannelStoreGetMoreChannels(t *testing.T) {
|
|
Setup()
|
|
|
|
o1 := model.Channel{}
|
|
o1.TeamId = model.NewId()
|
|
o1.DisplayName = "Channel1"
|
|
o1.Name = "a" + model.NewId() + "b"
|
|
o1.Type = model.CHANNEL_OPEN
|
|
Must(store.Channel().Save(&o1))
|
|
|
|
o2 := model.Channel{}
|
|
o2.TeamId = model.NewId()
|
|
o2.DisplayName = "Channel2"
|
|
o2.Name = "a" + model.NewId() + "b"
|
|
o2.Type = model.CHANNEL_OPEN
|
|
Must(store.Channel().Save(&o2))
|
|
|
|
m1 := model.ChannelMember{}
|
|
m1.ChannelId = o1.Id
|
|
m1.UserId = model.NewId()
|
|
m1.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
Must(store.Channel().SaveMember(&m1))
|
|
|
|
m2 := model.ChannelMember{}
|
|
m2.ChannelId = o1.Id
|
|
m2.UserId = model.NewId()
|
|
m2.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
Must(store.Channel().SaveMember(&m2))
|
|
|
|
m3 := model.ChannelMember{}
|
|
m3.ChannelId = o2.Id
|
|
m3.UserId = model.NewId()
|
|
m3.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
Must(store.Channel().SaveMember(&m3))
|
|
|
|
o3 := model.Channel{}
|
|
o3.TeamId = o1.TeamId
|
|
o3.DisplayName = "ChannelA"
|
|
o3.Name = "a" + model.NewId() + "b"
|
|
o3.Type = model.CHANNEL_OPEN
|
|
Must(store.Channel().Save(&o3))
|
|
|
|
o4 := model.Channel{}
|
|
o4.TeamId = o1.TeamId
|
|
o4.DisplayName = "ChannelB"
|
|
o4.Name = "a" + model.NewId() + "b"
|
|
o4.Type = model.CHANNEL_PRIVATE
|
|
Must(store.Channel().Save(&o4))
|
|
|
|
o5 := model.Channel{}
|
|
o5.TeamId = o1.TeamId
|
|
o5.DisplayName = "ChannelC"
|
|
o5.Name = "a" + model.NewId() + "b"
|
|
o5.Type = model.CHANNEL_PRIVATE
|
|
Must(store.Channel().Save(&o5))
|
|
|
|
cresult := <-store.Channel().GetMoreChannels(o1.TeamId, m1.UserId)
|
|
list := cresult.Data.(*model.ChannelList)
|
|
|
|
if len(list.Channels) != 1 {
|
|
t.Fatal("wrong list")
|
|
}
|
|
|
|
if list.Channels[0].Name != o3.Name {
|
|
t.Fatal("missing channel")
|
|
}
|
|
|
|
if r1 := <-store.Channel().AnalyticsTypeCount(o1.TeamId, model.CHANNEL_OPEN); r1.Err != nil {
|
|
t.Fatal(r1.Err)
|
|
} else {
|
|
if r1.Data.(int64) != 2 {
|
|
t.Log(r1.Data)
|
|
t.Fatal("wrong value")
|
|
}
|
|
}
|
|
|
|
if r1 := <-store.Channel().AnalyticsTypeCount(o1.TeamId, model.CHANNEL_PRIVATE); r1.Err != nil {
|
|
t.Fatal(r1.Err)
|
|
} else {
|
|
if r1.Data.(int64) != 2 {
|
|
t.Log(r1.Data)
|
|
t.Fatal("wrong value")
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestChannelStoreGetChannelCounts(t *testing.T) {
|
|
Setup()
|
|
|
|
o2 := model.Channel{}
|
|
o2.TeamId = model.NewId()
|
|
o2.DisplayName = "Channel2"
|
|
o2.Name = "a" + model.NewId() + "b"
|
|
o2.Type = model.CHANNEL_OPEN
|
|
Must(store.Channel().Save(&o2))
|
|
|
|
o1 := model.Channel{}
|
|
o1.TeamId = model.NewId()
|
|
o1.DisplayName = "Channel1"
|
|
o1.Name = "a" + model.NewId() + "b"
|
|
o1.Type = model.CHANNEL_OPEN
|
|
Must(store.Channel().Save(&o1))
|
|
|
|
m1 := model.ChannelMember{}
|
|
m1.ChannelId = o1.Id
|
|
m1.UserId = model.NewId()
|
|
m1.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
Must(store.Channel().SaveMember(&m1))
|
|
|
|
m2 := model.ChannelMember{}
|
|
m2.ChannelId = o1.Id
|
|
m2.UserId = model.NewId()
|
|
m2.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
Must(store.Channel().SaveMember(&m2))
|
|
|
|
m3 := model.ChannelMember{}
|
|
m3.ChannelId = o2.Id
|
|
m3.UserId = model.NewId()
|
|
m3.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
Must(store.Channel().SaveMember(&m3))
|
|
|
|
cresult := <-store.Channel().GetChannelCounts(o1.TeamId, m1.UserId)
|
|
counts := cresult.Data.(*model.ChannelCounts)
|
|
|
|
if len(counts.Counts) != 1 {
|
|
t.Fatal("wrong number of counts")
|
|
}
|
|
|
|
if len(counts.UpdateTimes) != 1 {
|
|
t.Fatal("wrong number of update times")
|
|
}
|
|
}
|
|
|
|
func TestChannelStoreUpdateLastViewedAt(t *testing.T) {
|
|
Setup()
|
|
|
|
o1 := model.Channel{}
|
|
o1.TeamId = model.NewId()
|
|
o1.DisplayName = "Channel1"
|
|
o1.Name = "a" + model.NewId() + "b"
|
|
o1.Type = model.CHANNEL_OPEN
|
|
o1.TotalMsgCount = 25
|
|
Must(store.Channel().Save(&o1))
|
|
|
|
m1 := model.ChannelMember{}
|
|
m1.ChannelId = o1.Id
|
|
m1.UserId = model.NewId()
|
|
m1.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
Must(store.Channel().SaveMember(&m1))
|
|
|
|
err := (<-store.Channel().UpdateLastViewedAt(m1.ChannelId, m1.UserId)).Err
|
|
if err != nil {
|
|
t.Fatal("failed to update", err)
|
|
}
|
|
|
|
err = (<-store.Channel().UpdateLastViewedAt(m1.ChannelId, "missing id")).Err
|
|
if err != nil {
|
|
t.Fatal("failed to update")
|
|
}
|
|
}
|
|
|
|
func TestChannelStoreIncrementMentionCount(t *testing.T) {
|
|
Setup()
|
|
|
|
o1 := model.Channel{}
|
|
o1.TeamId = model.NewId()
|
|
o1.DisplayName = "Channel1"
|
|
o1.Name = "a" + model.NewId() + "b"
|
|
o1.Type = model.CHANNEL_OPEN
|
|
o1.TotalMsgCount = 25
|
|
Must(store.Channel().Save(&o1))
|
|
|
|
m1 := model.ChannelMember{}
|
|
m1.ChannelId = o1.Id
|
|
m1.UserId = model.NewId()
|
|
m1.NotifyProps = model.GetDefaultChannelNotifyProps()
|
|
Must(store.Channel().SaveMember(&m1))
|
|
|
|
err := (<-store.Channel().IncrementMentionCount(m1.ChannelId, m1.UserId)).Err
|
|
if err != nil {
|
|
t.Fatal("failed to update")
|
|
}
|
|
|
|
err = (<-store.Channel().IncrementMentionCount(m1.ChannelId, "missing id")).Err
|
|
if err != nil {
|
|
t.Fatal("failed to update")
|
|
}
|
|
|
|
err = (<-store.Channel().IncrementMentionCount("missing id", m1.UserId)).Err
|
|
if err != nil {
|
|
t.Fatal("failed to update")
|
|
}
|
|
|
|
err = (<-store.Channel().IncrementMentionCount("missing id", "missing id")).Err
|
|
if err != nil {
|
|
t.Fatal("failed to update")
|
|
}
|
|
}
|
|
|
|
func TestGetMemberCount(t *testing.T) {
|
|
Setup()
|
|
|
|
teamId := model.NewId()
|
|
|
|
c1 := model.Channel{
|
|
TeamId: teamId,
|
|
DisplayName: "Channel1",
|
|
Name: "a" + model.NewId() + "b",
|
|
Type: model.CHANNEL_OPEN,
|
|
}
|
|
Must(store.Channel().Save(&c1))
|
|
|
|
c2 := model.Channel{
|
|
TeamId: teamId,
|
|
DisplayName: "Channel2",
|
|
Name: "a" + model.NewId() + "b",
|
|
Type: model.CHANNEL_OPEN,
|
|
}
|
|
Must(store.Channel().Save(&c2))
|
|
|
|
t.Logf("c1.Id = %v", c1.Id)
|
|
|
|
u1 := &model.User{
|
|
Email: model.NewId(),
|
|
DeleteAt: 0,
|
|
}
|
|
Must(store.User().Save(u1))
|
|
Must(store.Team().SaveMember(&model.TeamMember{TeamId: teamId, UserId: u1.Id}))
|
|
|
|
m1 := model.ChannelMember{
|
|
ChannelId: c1.Id,
|
|
UserId: u1.Id,
|
|
NotifyProps: model.GetDefaultChannelNotifyProps(),
|
|
}
|
|
Must(store.Channel().SaveMember(&m1))
|
|
|
|
if result := <-store.Channel().GetMemberCount(c1.Id); result.Err != nil {
|
|
t.Fatal("failed to get member count: %v", result.Err)
|
|
} else if result.Data.(int64) != 1 {
|
|
t.Fatal("got incorrect member count %v", result.Data)
|
|
}
|
|
|
|
u2 := model.User{
|
|
Email: model.NewId(),
|
|
DeleteAt: 0,
|
|
}
|
|
Must(store.User().Save(&u2))
|
|
Must(store.Team().SaveMember(&model.TeamMember{TeamId: teamId, UserId: u2.Id}))
|
|
|
|
m2 := model.ChannelMember{
|
|
ChannelId: c1.Id,
|
|
UserId: u2.Id,
|
|
NotifyProps: model.GetDefaultChannelNotifyProps(),
|
|
}
|
|
Must(store.Channel().SaveMember(&m2))
|
|
|
|
if result := <-store.Channel().GetMemberCount(c1.Id); result.Err != nil {
|
|
t.Fatal("failed to get member count: %v", result.Err)
|
|
} else if result.Data.(int64) != 2 {
|
|
t.Fatal("got incorrect member count %v", result.Data)
|
|
}
|
|
|
|
// make sure members of other channels aren't counted
|
|
u3 := model.User{
|
|
Email: model.NewId(),
|
|
DeleteAt: 0,
|
|
}
|
|
Must(store.User().Save(&u3))
|
|
Must(store.Team().SaveMember(&model.TeamMember{TeamId: teamId, UserId: u3.Id}))
|
|
|
|
m3 := model.ChannelMember{
|
|
ChannelId: c2.Id,
|
|
UserId: u3.Id,
|
|
NotifyProps: model.GetDefaultChannelNotifyProps(),
|
|
}
|
|
Must(store.Channel().SaveMember(&m3))
|
|
|
|
if result := <-store.Channel().GetMemberCount(c1.Id); result.Err != nil {
|
|
t.Fatal("failed to get member count: %v", result.Err)
|
|
} else if result.Data.(int64) != 2 {
|
|
t.Fatal("got incorrect member count %v", result.Data)
|
|
}
|
|
|
|
// make sure inactive users aren't counted
|
|
u4 := &model.User{
|
|
Email: model.NewId(),
|
|
DeleteAt: 10000,
|
|
}
|
|
Must(store.User().Save(u4))
|
|
Must(store.Team().SaveMember(&model.TeamMember{TeamId: teamId, UserId: u4.Id}))
|
|
|
|
m4 := model.ChannelMember{
|
|
ChannelId: c1.Id,
|
|
UserId: u4.Id,
|
|
NotifyProps: model.GetDefaultChannelNotifyProps(),
|
|
}
|
|
Must(store.Channel().SaveMember(&m4))
|
|
|
|
if result := <-store.Channel().GetMemberCount(c1.Id); result.Err != nil {
|
|
t.Fatal("failed to get member count: %v", result.Err)
|
|
} else if result.Data.(int64) != 2 {
|
|
t.Fatal("got incorrect member count %v", result.Data)
|
|
}
|
|
}
|
|
|
|
func TestUpdateExtrasByUser(t *testing.T) {
|
|
Setup()
|
|
|
|
teamId := model.NewId()
|
|
|
|
c1 := model.Channel{
|
|
TeamId: teamId,
|
|
DisplayName: "Channel1",
|
|
Name: "a" + model.NewId() + "b",
|
|
Type: model.CHANNEL_OPEN,
|
|
}
|
|
Must(store.Channel().Save(&c1))
|
|
|
|
c2 := model.Channel{
|
|
TeamId: teamId,
|
|
DisplayName: "Channel2",
|
|
Name: "a" + model.NewId() + "b",
|
|
Type: model.CHANNEL_OPEN,
|
|
}
|
|
Must(store.Channel().Save(&c2))
|
|
|
|
t.Logf("c1.Id = %v", c1.Id)
|
|
|
|
u1 := &model.User{
|
|
Email: model.NewId(),
|
|
DeleteAt: 0,
|
|
}
|
|
Must(store.User().Save(u1))
|
|
Must(store.Team().SaveMember(&model.TeamMember{TeamId: teamId, UserId: u1.Id}))
|
|
|
|
m1 := model.ChannelMember{
|
|
ChannelId: c1.Id,
|
|
UserId: u1.Id,
|
|
NotifyProps: model.GetDefaultChannelNotifyProps(),
|
|
}
|
|
Must(store.Channel().SaveMember(&m1))
|
|
|
|
u1.DeleteAt = model.GetMillis()
|
|
Must(store.User().Update(u1, true))
|
|
|
|
if result := <-store.Channel().ExtraUpdateByUser(u1.Id, u1.DeleteAt); result.Err != nil {
|
|
t.Fatal("failed to update extras by user: %v", result.Err)
|
|
}
|
|
|
|
if result := <-store.Channel().GetExtraMembers(c1.Id, -1); result.Err != nil {
|
|
t.Fatal("failed to get extras: %v", result.Err)
|
|
} else if len(result.Data.([]model.ExtraMember)) != 0 {
|
|
t.Fatal("got incorrect member count %v", len(result.Data.([]model.ExtraMember)))
|
|
}
|
|
|
|
u1.DeleteAt = 0
|
|
Must(store.User().Update(u1, true))
|
|
|
|
if result := <-store.Channel().ExtraUpdateByUser(u1.Id, u1.DeleteAt); result.Err != nil {
|
|
t.Fatal("failed to update extras by user: %v", result.Err)
|
|
}
|
|
|
|
if result := <-store.Channel().GetExtraMembers(c1.Id, -1); result.Err != nil {
|
|
t.Fatal("failed to get extras: %v", result.Err)
|
|
} else if len(result.Data.([]model.ExtraMember)) != 1 {
|
|
t.Fatal("got incorrect member count %v", len(result.Data.([]model.ExtraMember)))
|
|
}
|
|
}
|