summaryrefslogtreecommitdiffstats
path: root/store/sql_channel_store_test.go
diff options
context:
space:
mode:
authorChris <ccbrown112@gmail.com>2017-09-25 09:11:25 -0500
committerJoram Wilander <jwawilander@gmail.com>2017-09-25 10:11:25 -0400
commit49fe5fbf3db56fc466b8997b182ee135d7a4365d (patch)
tree1252fea09aa3ce899e2e8edb1fb7b42900f50bca /store/sql_channel_store_test.go
parentb2c5b97601b61f5748b46e4e386134203111ebb0 (diff)
downloadchat-49fe5fbf3db56fc466b8997b182ee135d7a4365d.tar.gz
chat-49fe5fbf3db56fc466b8997b182ee135d7a4365d.tar.bz2
chat-49fe5fbf3db56fc466b8997b182ee135d7a4365d.zip
Move sql store code into store/sqlstore package (#7502)
* move sql store code into store/sqlstore package * move non-sql constants back up to store * fix api test * derp
Diffstat (limited to 'store/sql_channel_store_test.go')
-rw-r--r--store/sql_channel_store_test.go1961
1 files changed, 0 insertions, 1961 deletions
diff --git a/store/sql_channel_store_test.go b/store/sql_channel_store_test.go
deleted file mode 100644
index c09cfd0b4..000000000
--- a/store/sql_channel_store_test.go
+++ /dev/null
@@ -1,1961 +0,0 @@
-// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
-// See License.txt for license information.
-
-package store
-
-import (
- "testing"
- "time"
-
- "github.com/mattermost/mattermost-server/model"
-)
-
-func TestChannelStoreSave(t *testing.T) {
- Setup()
-
- teamId := model.NewId()
-
- o1 := model.Channel{}
- o1.TeamId = teamId
- o1.DisplayName = "Name"
- o1.Name = "zz" + 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 = "zz" + 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")
- }
-}
-
-func TestChannelStoreSaveDirectChannel(t *testing.T) {
- Setup()
-
- teamId := model.NewId()
-
- o1 := model.Channel{}
- o1.TeamId = teamId
- o1.DisplayName = "Name"
- o1.Name = "zz" + 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, 0, 100)).Data.(*model.ChannelMembers)
- 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")
- }
-
- // Attempt to save a direct channel that already exists
- o1a := model.Channel{
- TeamId: o1.TeamId,
- DisplayName: o1.DisplayName,
- Name: o1.Name,
- Type: o1.Type,
- }
-
- if result := <-store.Channel().SaveDirectChannel(&o1a, &m1, &m2); result.Err == nil {
- t.Fatal("should've failed to save a duplicate direct channel")
- } else if result.Err.Id != CHANNEL_EXISTS_ERROR {
- t.Fatal("should've returned CHANNEL_EXISTS_ERROR")
- } else if returned := result.Data.(*model.Channel); returned.Id != o1.Id {
- t.Fatal("should've returned original channel when saving a duplicate direct channel")
- }
-
- // Attempt to save a non-direct channel
- o1.Id = ""
- o1.Name = "zz" + 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 TestChannelStoreCreateDirectChannel(t *testing.T) {
- Setup()
-
- 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}))
-
- res := <-store.Channel().CreateDirectChannel(u1.Id, u2.Id)
- if res.Err != nil {
- t.Fatal("couldn't create direct channel", res.Err)
- }
-
- c1 := res.Data.(*model.Channel)
-
- members := (<-store.Channel().GetMembers(c1.Id, 0, 100)).Data.(*model.ChannelMembers)
- if len(*members) != 2 {
- t.Fatal("should have saved 2 members")
- }
-}
-
-func TestChannelStoreUpdate(t *testing.T) {
- Setup()
-
- o1 := model.Channel{}
- o1.TeamId = model.NewId()
- o1.DisplayName = "Name"
- o1.Name = "zz" + model.NewId() + "b"
- o1.Type = model.CHANNEL_OPEN
- Must(store.Channel().Save(&o1))
-
- o2 := model.Channel{}
- o2.TeamId = o1.TeamId
- o2.DisplayName = "Name"
- o2.Name = "zz" + model.NewId() + "b"
- o2.Type = model.CHANNEL_OPEN
- Must(store.Channel().Save(&o2))
-
- 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")
- }
-
- o2.Name = o1.Name
- if err := (<-store.Channel().Update(&o2)).Err; err == nil {
- t.Fatal("Update should have failed because of existing name")
- }
-}
-
-func TestGetChannelUnread(t *testing.T) {
- Setup()
-
- teamId1 := model.NewId()
- teamId2 := model.NewId()
-
- uid := model.NewId()
- m1 := &model.TeamMember{TeamId: teamId1, UserId: uid}
- m2 := &model.TeamMember{TeamId: teamId2, UserId: uid}
- Must(store.Team().SaveMember(m1))
- Must(store.Team().SaveMember(m2))
- notifyPropsModel := model.GetDefaultChannelNotifyProps()
-
- // Setup Channel 1
- c1 := &model.Channel{TeamId: m1.TeamId, Name: model.NewId(), DisplayName: "Downtown", Type: model.CHANNEL_OPEN, TotalMsgCount: 100}
- Must(store.Channel().Save(c1))
- cm1 := &model.ChannelMember{ChannelId: c1.Id, UserId: m1.UserId, NotifyProps: notifyPropsModel, MsgCount: 90}
- Must(store.Channel().SaveMember(cm1))
-
- // Setup Channel 2
- c2 := &model.Channel{TeamId: m2.TeamId, Name: model.NewId(), DisplayName: "Cultural", Type: model.CHANNEL_OPEN, TotalMsgCount: 100}
- Must(store.Channel().Save(c2))
- cm2 := &model.ChannelMember{ChannelId: c2.Id, UserId: m2.UserId, NotifyProps: notifyPropsModel, MsgCount: 90, MentionCount: 5}
- Must(store.Channel().SaveMember(cm2))
-
- // Check for Channel 1
- if resp := <-store.Channel().GetChannelUnread(c1.Id, uid); resp.Err != nil {
- t.Fatal(resp.Err)
- } else {
- ch := resp.Data.(*model.ChannelUnread)
- if c1.Id != ch.ChannelId {
- t.Fatal("wrong channel id")
- }
-
- if teamId1 != ch.TeamId {
- t.Fatal("wrong team id for channel 1")
- }
-
- if ch.NotifyProps == nil {
- t.Fatal("wrong props for channel 1")
- }
-
- if ch.MentionCount != 0 {
- t.Fatal("wrong MentionCount for channel 1")
- }
-
- if ch.MsgCount != 10 {
- t.Fatal("wrong MsgCount for channel 1")
- }
- }
-
- // Check for Channel 2
- if resp2 := <-store.Channel().GetChannelUnread(c2.Id, uid); resp2.Err != nil {
- t.Fatal(resp2.Err)
- } else {
- ch2 := resp2.Data.(*model.ChannelUnread)
- if c2.Id != ch2.ChannelId {
- t.Fatal("wrong channel id")
- }
-
- if teamId2 != ch2.TeamId {
- t.Fatal("wrong team id")
- }
-
- if ch2.MentionCount != 5 {
- t.Fatal("wrong MentionCount for channel 2")
- }
-
- if ch2.MsgCount != 10 {
- t.Fatal("wrong MsgCount for channel 2")
- }
- }
-}
-
-func TestChannelStoreGet(t *testing.T) {
- Setup()
-
- o1 := model.Channel{}
- o1.TeamId = model.NewId()
- o1.DisplayName = "Name"
- o1.Name = "zz" + model.NewId() + "b"
- o1.Type = model.CHANNEL_OPEN
- Must(store.Channel().Save(&o1))
-
- if r1 := <-store.Channel().Get(o1.Id, false); 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("", false)).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 = "zz" + 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, false); r2.Err != nil {
- t.Fatal(r2.Err)
- } else {
- if r2.Data.(*model.Channel).ToJson() != o2.ToJson() {
- t.Fatal("invalid returned channel")
- }
- }
-
- if r4 := <-store.Channel().Get(o2.Id, true); r4.Err != nil {
- t.Fatal(r4.Err)
- } else {
- if r4.Data.(*model.Channel).ToJson() != o2.ToJson() {
- t.Fatal("invalid returned channel")
- }
- }
-
- if r3 := <-store.Channel().GetAll(o1.TeamId); r3.Err != nil {
- t.Fatal(r3.Err)
- } else {
- channels := r3.Data.([]*model.Channel)
- if len(channels) == 0 {
- t.Fatal("too little")
- }
- }
-
- if r3 := <-store.Channel().GetTeamChannels(o1.TeamId); r3.Err != nil {
- t.Fatal(r3.Err)
- } else {
- channels := r3.Data.(*model.ChannelList)
- if len(*channels) == 0 {
- t.Fatal("too little")
- }
- }
-}
-
-func TestChannelStoreGetForPost(t *testing.T) {
- Setup()
-
- o1 := Must(store.Channel().Save(&model.Channel{
- TeamId: model.NewId(),
- DisplayName: "Name",
- Name: "zz" + model.NewId() + "b",
- Type: model.CHANNEL_OPEN,
- })).(*model.Channel)
-
- p1 := Must(store.Post().Save(&model.Post{
- UserId: model.NewId(),
- ChannelId: o1.Id,
- Message: "test",
- })).(*model.Post)
-
- if r1 := <-store.Channel().GetForPost(p1.Id); r1.Err != nil {
- t.Fatal(r1.Err)
- } else if r1.Data.(*model.Channel).Id != o1.Id {
- t.Fatal("incorrect channel returned")
- }
-}
-
-func TestSqlChannelStoreRestore(t *testing.T) {
- Setup()
-
- o1 := model.Channel{}
- o1.TeamId = model.NewId()
- o1.DisplayName = "Channel1"
- o1.Name = "zz" + model.NewId() + "b"
- o1.Type = model.CHANNEL_OPEN
- Must(store.Channel().Save(&o1))
-
- if r := <-store.Channel().Delete(o1.Id, model.GetMillis()); r.Err != nil {
- t.Fatal(r.Err)
- }
-
- if r := <-store.Channel().Get(o1.Id, false); r.Data.(*model.Channel).DeleteAt == 0 {
- t.Fatal("should have been deleted")
- }
-
- if r := <-store.Channel().Restore(o1.Id, model.GetMillis()); r.Err != nil {
- t.Fatal(r.Err)
- }
-
- if r := <-store.Channel().Get(o1.Id, false); r.Data.(*model.Channel).DeleteAt != 0 {
- t.Fatal("should have been restored")
- }
-
-}
-
-func TestChannelStoreDelete(t *testing.T) {
- Setup()
-
- o1 := model.Channel{}
- o1.TeamId = model.NewId()
- o1.DisplayName = "Channel1"
- o1.Name = "zz" + 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 = "zz" + 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 = "zz" + 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 = "zz" + 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, false); 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) != 1 {
- t.Fatal("invalid number of channels")
- }
-
- cresult = <-store.Channel().GetMoreChannels(o1.TeamId, m1.UserId, 0, 100)
- list = cresult.Data.(*model.ChannelList)
-
- if len(*list) != 1 {
- t.Fatal("invalid number of channels")
- }
-
- <-store.Channel().PermanentDelete(o2.Id)
-
- cresult = <-store.Channel().GetChannels(o1.TeamId, m1.UserId)
- t.Log(cresult.Err)
- if cresult.Err.Id != "store.sql_channel.get_channels.not_found.app_error" {
- t.Fatal("no channels should be found")
- }
-
- if r := <-store.Channel().PermanentDeleteByTeam(o1.TeamId); r.Err != nil {
- t.Fatal(r.Err)
- }
-}
-
-func TestChannelStoreGetByName(t *testing.T) {
- Setup()
-
- o1 := model.Channel{}
- o1.TeamId = model.NewId()
- o1.DisplayName = "Name"
- o1.Name = "zz" + model.NewId() + "b"
- o1.Type = model.CHANNEL_OPEN
- Must(store.Channel().Save(&o1))
-
- r1 := <-store.Channel().GetByName(o1.TeamId, o1.Name, true)
- if 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, "", true)).Err; err == nil {
- t.Fatal("Missing id should have failed")
- }
-
- if r1 := <-store.Channel().GetByName(o1.TeamId, o1.Name, false); 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, "", false)).Err; err == nil {
- t.Fatal("Missing id should have failed")
- }
-
- Must(store.Channel().Delete(r1.Data.(*model.Channel).Id, model.GetMillis()))
-
- if err := (<-store.Channel().GetByName(o1.TeamId, "", false)).Err; err == nil {
- t.Fatal("Deleted channel should not be returned by GetByName()")
- }
-}
-
-func TestChannelStoreGetDeletedByName(t *testing.T) {
- Setup()
-
- o1 := model.Channel{}
- o1.TeamId = model.NewId()
- o1.DisplayName = "Name"
- o1.Name = "zz" + model.NewId() + "b"
- o1.Type = model.CHANNEL_OPEN
- o1.DeleteAt = model.GetMillis()
- Must(store.Channel().Save(&o1))
-
- if r1 := <-store.Channel().GetDeletedByName(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().GetDeletedByName(o1.TeamId, "")).Err; err == nil {
- t.Fatal("Missing id should have failed")
- }
-}
-
-func TestChannelStoreGetDeleted(t *testing.T) {
- Setup()
-
- o1 := model.Channel{}
- o1.TeamId = model.NewId()
- o1.DisplayName = "Channel1"
- o1.Name = "zz" + model.NewId() + "b"
- o1.Type = model.CHANNEL_OPEN
- o1.DeleteAt = model.GetMillis()
- Must(store.Channel().Save(&o1))
-
- cresult := <-store.Channel().GetDeleted(o1.TeamId, 0, 100)
- if cresult.Err != nil {
- t.Fatal(cresult.Err)
- }
- list := cresult.Data.(*model.ChannelList)
-
- if len(*list) != 1 {
- t.Fatal("wrong list")
- }
-
- if (*list)[0].Name != o1.Name {
- t.Fatal("missing channel")
- }
-
- o2 := model.Channel{}
- o2.TeamId = o1.TeamId
- o2.DisplayName = "Channel2"
- o2.Name = "zz" + model.NewId() + "b"
- o2.Type = model.CHANNEL_OPEN
- Must(store.Channel().Save(&o2))
-
- cresult = <-store.Channel().GetDeleted(o1.TeamId, 0, 100)
- if cresult.Err != nil {
- t.Fatal(cresult.Err)
- }
- list = cresult.Data.(*model.ChannelList)
-
- if len(*list) != 1 {
- t.Fatal("wrong list")
- }
-
- o3 := model.Channel{}
- o3.TeamId = o1.TeamId
- o3.DisplayName = "Channel3"
- o3.Name = "zz" + model.NewId() + "b"
- o3.Type = model.CHANNEL_OPEN
- o3.DeleteAt = model.GetMillis()
- Must(store.Channel().Save(&o3))
-
- cresult = <-store.Channel().GetDeleted(o1.TeamId, 0, 100)
- if cresult.Err != nil {
- t.Fatal(cresult.Err)
- }
- list = cresult.Data.(*model.ChannelList)
-
- if len(*list) != 2 {
- t.Fatal("wrong list length")
- }
-
- cresult = <-store.Channel().GetDeleted(o1.TeamId, 0, 1)
- if cresult.Err != nil {
- t.Fatal(cresult.Err)
- }
- list = cresult.Data.(*model.ChannelList)
-
- if len(*list) != 1 {
- t.Fatal("wrong list length")
- }
-
- cresult = <-store.Channel().GetDeleted(o1.TeamId, 1, 1)
- if cresult.Err != nil {
- t.Fatal(cresult.Err)
- }
- list = cresult.Data.(*model.ChannelList)
-
- if len(*list) != 1 {
- t.Fatal("wrong list length")
- }
-
-}
-
-func TestChannelMemberStore(t *testing.T) {
- Setup()
-
- c1 := model.Channel{}
- c1.TeamId = model.NewId()
- c1.DisplayName = "NameName"
- c1.Name = "zz" + model.NewId() + "b"
- c1.Type = model.CHANNEL_OPEN
- c1 = *Must(store.Channel().Save(&c1)).(*model.Channel)
-
- c1t1 := (<-store.Channel().Get(c1.Id, false)).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, false)).Data.(*model.Channel)
- t2 := c1t2.ExtraUpdateAt
-
- if t2 <= t1 {
- t.Fatal("Member update time incorrect")
- }
-
- count := (<-store.Channel().GetMemberCount(o1.ChannelId, true)).Data.(int64)
- if count != 2 {
- t.Fatal("should have saved 2 members")
- }
-
- count = (<-store.Channel().GetMemberCount(o1.ChannelId, true)).Data.(int64)
- if count != 2 {
- t.Fatal("should have saved 2 members")
- }
-
- if store.Channel().GetMemberCountFromCache(o1.ChannelId) != 2 {
- t.Fatal("should have saved 2 members")
- }
-
- if store.Channel().GetMemberCountFromCache("junk") != 0 {
- t.Fatal("should have saved 0 members")
- }
-
- count = (<-store.Channel().GetMemberCount(o1.ChannelId, false)).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, false)).Data.(int64)
- if count != 1 {
- t.Fatal("should have removed 1 member")
- }
-
- c1t3 := (<-store.Channel().Get(c1.Id, false)).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")
- }
-
- if err := (<-store.Channel().SaveMember(&o1)).Err; err == nil {
- t.Fatal("Should have been a duplicate")
- }
-
- c1t4 := (<-store.Channel().Get(c1.Id, false)).Data.(*model.Channel)
- t4 := c1t4.ExtraUpdateAt
- if t4 != t3 {
- t.Fatal("Should not update time upon failure")
- }
-}
-
-func TestChannelDeleteMemberStore(t *testing.T) {
- Setup()
-
- c1 := model.Channel{}
- c1.TeamId = model.NewId()
- c1.DisplayName = "NameName"
- c1.Name = "zz" + model.NewId() + "b"
- c1.Type = model.CHANNEL_OPEN
- c1 = *Must(store.Channel().Save(&c1)).(*model.Channel)
-
- c1t1 := (<-store.Channel().Get(c1.Id, false)).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, false)).Data.(*model.Channel)
- t2 := c1t2.ExtraUpdateAt
-
- if t2 <= t1 {
- t.Fatal("Member update time incorrect")
- }
-
- count := (<-store.Channel().GetMemberCount(o1.ChannelId, false)).Data.(int64)
- if count != 2 {
- t.Fatal("should have saved 2 members")
- }
-
- Must(store.Channel().PermanentDeleteMembersByUser(o2.UserId))
-
- count = (<-store.Channel().GetMemberCount(o1.ChannelId, false)).Data.(int64)
- if count != 1 {
- t.Fatal("should have removed 1 member")
- }
-
- if r1 := <-store.Channel().PermanentDeleteMembersByChannel(o1.ChannelId); r1.Err != nil {
- t.Fatal(r1.Err)
- }
-
- count = (<-store.Channel().GetMemberCount(o1.ChannelId, false)).Data.(int64)
- if count != 0 {
- t.Fatal("should have removed all members")
- }
-}
-
-func TestChannelStoreGetChannels(t *testing.T) {
- Setup()
-
- o2 := model.Channel{}
- o2.TeamId = model.NewId()
- o2.DisplayName = "Channel2"
- o2.Name = "zz" + 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 = "zz" + 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)[0].Id != o1.Id {
- t.Fatal("missing channel")
- }
-
- acresult := <-store.Channel().GetAllChannelMembersForUser(m1.UserId, false)
- ids := acresult.Data.(map[string]string)
- if _, ok := ids[o1.Id]; !ok {
- t.Fatal("missing channel")
- }
-
- acresult2 := <-store.Channel().GetAllChannelMembersForUser(m1.UserId, true)
- ids2 := acresult2.Data.(map[string]string)
- if _, ok := ids2[o1.Id]; !ok {
- t.Fatal("missing channel")
- }
-
- acresult3 := <-store.Channel().GetAllChannelMembersForUser(m1.UserId, true)
- ids3 := acresult3.Data.(map[string]string)
- if _, ok := ids3[o1.Id]; !ok {
- t.Fatal("missing channel")
- }
-
- if !store.Channel().IsUserInChannelUseCache(m1.UserId, o1.Id) {
- t.Fatal("missing channel")
- }
-
- if store.Channel().IsUserInChannelUseCache(m1.UserId, o2.Id) {
- t.Fatal("missing channel")
- }
-
- if store.Channel().IsUserInChannelUseCache(m1.UserId, "blahblah") {
- t.Fatal("missing channel")
- }
-
- if store.Channel().IsUserInChannelUseCache("blahblah", "blahblah") {
- t.Fatal("missing channel")
- }
-
- store.Channel().InvalidateAllChannelMembersForUser(m1.UserId)
-}
-
-func TestChannelStoreGetMoreChannels(t *testing.T) {
- Setup()
-
- o1 := model.Channel{}
- o1.TeamId = model.NewId()
- o1.DisplayName = "Channel1"
- o1.Name = "zz" + 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 = "zz" + 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 = "zz" + 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 = "zz" + 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 = "zz" + model.NewId() + "b"
- o5.Type = model.CHANNEL_PRIVATE
- Must(store.Channel().Save(&o5))
-
- cresult := <-store.Channel().GetMoreChannels(o1.TeamId, m1.UserId, 0, 100)
- if cresult.Err != nil {
- t.Fatal(cresult.Err)
- }
- list := cresult.Data.(*model.ChannelList)
-
- if len(*list) != 1 {
- t.Fatal("wrong list")
- }
-
- if (*list)[0].Name != o3.Name {
- t.Fatal("missing channel")
- }
-
- o6 := model.Channel{}
- o6.TeamId = o1.TeamId
- o6.DisplayName = "ChannelA"
- o6.Name = "zz" + model.NewId() + "b"
- o6.Type = model.CHANNEL_OPEN
- Must(store.Channel().Save(&o6))
-
- cresult = <-store.Channel().GetMoreChannels(o1.TeamId, m1.UserId, 0, 100)
- list = cresult.Data.(*model.ChannelList)
-
- if len(*list) != 2 {
- t.Fatal("wrong list length")
- }
-
- cresult = <-store.Channel().GetMoreChannels(o1.TeamId, m1.UserId, 0, 1)
- list = cresult.Data.(*model.ChannelList)
-
- if len(*list) != 1 {
- t.Fatal("wrong list length")
- }
-
- cresult = <-store.Channel().GetMoreChannels(o1.TeamId, m1.UserId, 1, 1)
- list = cresult.Data.(*model.ChannelList)
-
- if len(*list) != 1 {
- t.Fatal("wrong list length")
- }
-
- if r1 := <-store.Channel().AnalyticsTypeCount(o1.TeamId, model.CHANNEL_OPEN); r1.Err != nil {
- t.Fatal(r1.Err)
- } else {
- if r1.Data.(int64) != 3 {
- 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 TestChannelStoreGetPublicChannelsForTeam(t *testing.T) {
- Setup()
-
- o1 := model.Channel{}
- o1.TeamId = model.NewId()
- o1.DisplayName = "OpenChannel1Team1"
- o1.Name = "zz" + model.NewId() + "b"
- o1.Type = model.CHANNEL_OPEN
- Must(store.Channel().Save(&o1))
-
- o2 := model.Channel{}
- o2.TeamId = model.NewId()
- o2.DisplayName = "OpenChannel1Team2"
- o2.Name = "zz" + model.NewId() + "b"
- o2.Type = model.CHANNEL_OPEN
- Must(store.Channel().Save(&o2))
-
- o3 := model.Channel{}
- o3.TeamId = o1.TeamId
- o3.DisplayName = "PrivateChannel1Team1"
- o3.Name = "zz" + model.NewId() + "b"
- o3.Type = model.CHANNEL_PRIVATE
- Must(store.Channel().Save(&o3))
-
- cresult := <-store.Channel().GetPublicChannelsForTeam(o1.TeamId, 0, 100)
- if cresult.Err != nil {
- t.Fatal(cresult.Err)
- }
- list := cresult.Data.(*model.ChannelList)
-
- if len(*list) != 1 {
- t.Fatal("wrong list")
- }
-
- if (*list)[0].Name != o1.Name {
- t.Fatal("missing channel")
- }
-
- o4 := model.Channel{}
- o4.TeamId = o1.TeamId
- o4.DisplayName = "OpenChannel2Team1"
- o4.Name = "zz" + model.NewId() + "b"
- o4.Type = model.CHANNEL_OPEN
- Must(store.Channel().Save(&o4))
-
- cresult = <-store.Channel().GetPublicChannelsForTeam(o1.TeamId, 0, 100)
- list = cresult.Data.(*model.ChannelList)
-
- if len(*list) != 2 {
- t.Fatal("wrong list length")
- }
-
- cresult = <-store.Channel().GetPublicChannelsForTeam(o1.TeamId, 0, 1)
- list = cresult.Data.(*model.ChannelList)
-
- if len(*list) != 1 {
- t.Fatal("wrong list length")
- }
-
- cresult = <-store.Channel().GetPublicChannelsForTeam(o1.TeamId, 1, 1)
- list = cresult.Data.(*model.ChannelList)
-
- if len(*list) != 1 {
- t.Fatal("wrong list length")
- }
-
- 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) != 1 {
- t.Log(r1.Data)
- t.Fatal("wrong value")
- }
- }
-}
-
-func TestChannelStoreGetPublicChannelsByIdsForTeam(t *testing.T) {
- Setup()
-
- teamId1 := model.NewId()
-
- oc1 := model.Channel{}
- oc1.TeamId = teamId1
- oc1.DisplayName = "OpenChannel1Team1"
- oc1.Name = "zz" + model.NewId() + "b"
- oc1.Type = model.CHANNEL_OPEN
- Must(store.Channel().Save(&oc1))
-
- oc2 := model.Channel{}
- oc2.TeamId = model.NewId()
- oc2.DisplayName = "OpenChannel2TeamOther"
- oc2.Name = "zz" + model.NewId() + "b"
- oc2.Type = model.CHANNEL_OPEN
- Must(store.Channel().Save(&oc2))
-
- pc3 := model.Channel{}
- pc3.TeamId = teamId1
- pc3.DisplayName = "PrivateChannel3Team1"
- pc3.Name = "zz" + model.NewId() + "b"
- pc3.Type = model.CHANNEL_PRIVATE
- Must(store.Channel().Save(&pc3))
-
- cids := []string{oc1.Id}
- cresult := <-store.Channel().GetPublicChannelsByIdsForTeam(teamId1, cids)
- list := cresult.Data.(*model.ChannelList)
-
- if len(*list) != 1 {
- t.Fatal("should return 1 channel")
- }
-
- if (*list)[0].Id != oc1.Id {
- t.Fatal("missing channel")
- }
-
- cids = append(cids, oc2.Id)
- cids = append(cids, model.NewId())
- cids = append(cids, pc3.Id)
- cresult = <-store.Channel().GetPublicChannelsByIdsForTeam(teamId1, cids)
- list = cresult.Data.(*model.ChannelList)
-
- if len(*list) != 1 {
- t.Fatal("should return 1 channel")
- }
-
- oc4 := model.Channel{}
- oc4.TeamId = teamId1
- oc4.DisplayName = "OpenChannel4Team1"
- oc4.Name = "zz" + model.NewId() + "b"
- oc4.Type = model.CHANNEL_OPEN
- Must(store.Channel().Save(&oc4))
-
- cids = append(cids, oc4.Id)
- cresult = <-store.Channel().GetPublicChannelsByIdsForTeam(teamId1, cids)
- list = cresult.Data.(*model.ChannelList)
-
- if len(*list) != 2 {
- t.Fatal("should return 2 channels")
- }
-
- if (*list)[0].Id != oc1.Id {
- t.Fatal("missing channel")
- }
-
- if (*list)[1].Id != oc4.Id {
- t.Fatal("missing channel")
- }
-
- cids = cids[:0]
- cids = append(cids, model.NewId())
- cresult = <-store.Channel().GetPublicChannelsByIdsForTeam(teamId1, cids)
- list = cresult.Data.(*model.ChannelList)
-
- if len(*list) != 0 {
- t.Fatal("should not return a channel")
- }
-}
-
-func TestChannelStoreGetChannelCounts(t *testing.T) {
- Setup()
-
- o2 := model.Channel{}
- o2.TeamId = model.NewId()
- o2.DisplayName = "Channel2"
- o2.Name = "zz" + 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 = "zz" + 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 TestChannelStoreGetMembersForUser(t *testing.T) {
- Setup()
-
- t1 := model.Team{}
- t1.DisplayName = "Name"
- t1.Name = model.NewId()
- t1.Email = model.NewId() + "@nowhere.com"
- t1.Type = model.TEAM_OPEN
- Must(store.Team().Save(&t1))
-
- o1 := model.Channel{}
- o1.TeamId = t1.Id
- o1.DisplayName = "Channel1"
- o1.Name = "zz" + 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 = "zz" + 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 = o2.Id
- m2.UserId = m1.UserId
- m2.NotifyProps = model.GetDefaultChannelNotifyProps()
- Must(store.Channel().SaveMember(&m2))
-
- cresult := <-store.Channel().GetMembersForUser(o1.TeamId, m1.UserId)
- members := cresult.Data.(*model.ChannelMembers)
-
- // no unread messages
- if len(*members) != 2 {
- t.Fatal("wrong number of members")
- }
-}
-
-func TestChannelStoreUpdateLastViewedAt(t *testing.T) {
- Setup()
-
- o1 := model.Channel{}
- o1.TeamId = model.NewId()
- o1.DisplayName = "Channel1"
- o1.Name = "zz" + 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([]string{m1.ChannelId}, m1.UserId)).Err
- if err != nil {
- t.Fatal("failed to update", err)
- }
-
- err = (<-store.Channel().UpdateLastViewedAt([]string{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 = "zz" + 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 TestUpdateChannelMember(t *testing.T) {
- Setup()
-
- userId := model.NewId()
-
- c1 := &model.Channel{
- TeamId: model.NewId(),
- DisplayName: model.NewId(),
- Name: model.NewId(),
- Type: model.CHANNEL_OPEN,
- }
- Must(store.Channel().Save(c1))
-
- m1 := &model.ChannelMember{
- ChannelId: c1.Id,
- UserId: userId,
- NotifyProps: model.GetDefaultChannelNotifyProps(),
- }
- Must(store.Channel().SaveMember(m1))
-
- m1.NotifyProps["test"] = "sometext"
- if result := <-store.Channel().UpdateMember(m1); result.Err != nil {
- t.Fatal(result.Err)
- }
-
- m1.UserId = ""
- if result := <-store.Channel().UpdateMember(m1); result.Err == nil {
- t.Fatal("bad user id - should fail")
- }
-}
-
-func TestGetMember(t *testing.T) {
- Setup()
-
- userId := model.NewId()
-
- c1 := &model.Channel{
- TeamId: model.NewId(),
- DisplayName: model.NewId(),
- Name: model.NewId(),
- Type: model.CHANNEL_OPEN,
- }
- Must(store.Channel().Save(c1))
-
- c2 := &model.Channel{
- TeamId: c1.TeamId,
- DisplayName: model.NewId(),
- Name: model.NewId(),
- Type: model.CHANNEL_OPEN,
- }
- Must(store.Channel().Save(c2))
-
- m1 := &model.ChannelMember{
- ChannelId: c1.Id,
- UserId: userId,
- NotifyProps: model.GetDefaultChannelNotifyProps(),
- }
- Must(store.Channel().SaveMember(m1))
-
- m2 := &model.ChannelMember{
- ChannelId: c2.Id,
- UserId: userId,
- NotifyProps: model.GetDefaultChannelNotifyProps(),
- }
- Must(store.Channel().SaveMember(m2))
-
- if result := <-store.Channel().GetMember(model.NewId(), userId); result.Err == nil {
- t.Fatal("should've failed to get member for non-existant channel")
- }
-
- if result := <-store.Channel().GetMember(c1.Id, model.NewId()); result.Err == nil {
- t.Fatal("should've failed to get member for non-existant user")
- }
-
- if result := <-store.Channel().GetMember(c1.Id, userId); result.Err != nil {
- t.Fatal("shouldn't have errored when getting member", result.Err)
- } else if member := result.Data.(*model.ChannelMember); member.ChannelId != c1.Id {
- t.Fatal("should've gotten member of channel 1")
- } else if member.UserId != userId {
- t.Fatal("should've gotten member for user")
- }
-
- if result := <-store.Channel().GetMember(c2.Id, userId); result.Err != nil {
- t.Fatal("shouldn't have errored when getting member", result.Err)
- } else if member := result.Data.(*model.ChannelMember); member.ChannelId != c2.Id {
- t.Fatal("should've gotten member of channel 2")
- } else if member.UserId != userId {
- t.Fatal("should've gotten member for user")
- }
-
- if result := <-store.Channel().GetAllChannelMembersNotifyPropsForChannel(c2.Id, false); result.Err != nil {
- t.Fatal(result.Err)
- } else {
- props := result.Data.(map[string]model.StringMap)
- if len(props) == 0 {
- t.Fatal("should not be empty")
- }
- }
-
- if result := <-store.Channel().GetAllChannelMembersNotifyPropsForChannel(c2.Id, true); result.Err != nil {
- t.Fatal(result.Err)
- } else {
- props := result.Data.(map[string]model.StringMap)
- if len(props) == 0 {
- t.Fatal("should not be empty")
- }
- }
-
- store.Channel().InvalidateCacheForChannelMembersNotifyProps(c2.Id)
-}
-
-func TestChannelStoreGetMemberForPost(t *testing.T) {
- Setup()
-
- o1 := Must(store.Channel().Save(&model.Channel{
- TeamId: model.NewId(),
- DisplayName: "Name",
- Name: "zz" + model.NewId() + "b",
- Type: model.CHANNEL_OPEN,
- })).(*model.Channel)
-
- m1 := Must(store.Channel().SaveMember(&model.ChannelMember{
- ChannelId: o1.Id,
- UserId: model.NewId(),
- NotifyProps: model.GetDefaultChannelNotifyProps(),
- })).(*model.ChannelMember)
-
- p1 := Must(store.Post().Save(&model.Post{
- UserId: model.NewId(),
- ChannelId: o1.Id,
- Message: "test",
- })).(*model.Post)
-
- if r1 := <-store.Channel().GetMemberForPost(p1.Id, m1.UserId); r1.Err != nil {
- t.Fatal(r1.Err)
- } else if r1.Data.(*model.ChannelMember).ToJson() != m1.ToJson() {
- t.Fatal("invalid returned channel member")
- }
-
- if r2 := <-store.Channel().GetMemberForPost(p1.Id, model.NewId()); r2.Err == nil {
- t.Fatal("shouldn't have returned a member")
- }
-}
-
-func TestGetMemberCount(t *testing.T) {
- Setup()
-
- teamId := model.NewId()
-
- c1 := model.Channel{
- TeamId: teamId,
- DisplayName: "Channel1",
- Name: "zz" + model.NewId() + "b",
- Type: model.CHANNEL_OPEN,
- }
- Must(store.Channel().Save(&c1))
-
- c2 := model.Channel{
- TeamId: teamId,
- DisplayName: "Channel2",
- Name: "zz" + model.NewId() + "b",
- Type: model.CHANNEL_OPEN,
- }
- Must(store.Channel().Save(&c2))
-
- 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, false); result.Err != nil {
- t.Fatalf("failed to get member count: %v", result.Err)
- } else if result.Data.(int64) != 1 {
- t.Fatalf("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, false); result.Err != nil {
- t.Fatalf("failed to get member count: %v", result.Err)
- } else if result.Data.(int64) != 2 {
- t.Fatalf("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, false); result.Err != nil {
- t.Fatalf("failed to get member count: %v", result.Err)
- } else if result.Data.(int64) != 2 {
- t.Fatalf("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, false); result.Err != nil {
- t.Fatalf("failed to get member count: %v", result.Err)
- } else if result.Data.(int64) != 2 {
- t.Fatalf("got incorrect member count %v", result.Data)
- }
-}
-
-func TestUpdateExtrasByUser(t *testing.T) {
- Setup()
-
- teamId := model.NewId()
-
- c1 := model.Channel{
- TeamId: teamId,
- DisplayName: "Channel1",
- Name: "zz" + model.NewId() + "b",
- Type: model.CHANNEL_OPEN,
- }
- Must(store.Channel().Save(&c1))
-
- c2 := model.Channel{
- TeamId: teamId,
- DisplayName: "Channel2",
- Name: "zz" + model.NewId() + "b",
- Type: model.CHANNEL_OPEN,
- }
- Must(store.Channel().Save(&c2))
-
- 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.Fatalf("failed to update extras by user: %v", result.Err)
- }
-
- u1.DeleteAt = 0
- Must(store.User().Update(u1, true))
-
- if result := <-store.Channel().ExtraUpdateByUser(u1.Id, u1.DeleteAt); result.Err != nil {
- t.Fatalf("failed to update extras by user: %v", result.Err)
- }
-}
-
-func TestChannelStoreSearchMore(t *testing.T) {
- Setup()
-
- o1 := model.Channel{}
- o1.TeamId = model.NewId()
- o1.DisplayName = "ChannelA"
- o1.Name = "zz" + 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 = "zz" + 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 = "zz" + 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 = "zz" + 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 = "zz" + model.NewId() + "b"
- o5.Type = model.CHANNEL_PRIVATE
- Must(store.Channel().Save(&o5))
-
- if result := <-store.Channel().SearchMore(m1.UserId, o1.TeamId, "ChannelA"); result.Err != nil {
- t.Fatal(result.Err)
- } else {
- channels := result.Data.(*model.ChannelList)
- if len(*channels) == 0 {
- t.Fatal("should not be empty")
- }
-
- if (*channels)[0].Name != o3.Name {
- t.Fatal("wrong channel returned")
- }
- }
-
- if result := <-store.Channel().SearchMore(m1.UserId, o1.TeamId, o4.Name); result.Err != nil {
- t.Fatal(result.Err)
- } else {
- channels := result.Data.(*model.ChannelList)
- if len(*channels) != 0 {
- t.Fatal("should be empty")
- }
- }
-
- if result := <-store.Channel().SearchMore(m1.UserId, o1.TeamId, o3.Name); result.Err != nil {
- t.Fatal(result.Err)
- } else {
- channels := result.Data.(*model.ChannelList)
- if len(*channels) == 0 {
- t.Fatal("should not be empty")
- }
-
- if (*channels)[0].Name != o3.Name {
- t.Fatal("wrong channel returned")
- }
- }
-
-}
-
-func TestChannelStoreSearchInTeam(t *testing.T) {
- Setup()
-
- o1 := model.Channel{}
- o1.TeamId = model.NewId()
- o1.DisplayName = "ChannelA"
- o1.Name = "zz" + 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 = "zz" + 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 = "zz" + 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 = "zz" + 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 = "zz" + model.NewId() + "b"
- o5.Type = model.CHANNEL_PRIVATE
- Must(store.Channel().Save(&o5))
-
- if result := <-store.Channel().SearchInTeam(o1.TeamId, "ChannelA"); result.Err != nil {
- t.Fatal(result.Err)
- } else {
- channels := result.Data.(*model.ChannelList)
- if len(*channels) != 2 {
- t.Fatal("wrong length")
- }
- }
-
- if result := <-store.Channel().SearchInTeam(o1.TeamId, ""); result.Err != nil {
- t.Fatal(result.Err)
- } else {
- channels := result.Data.(*model.ChannelList)
- if len(*channels) == 0 {
- t.Fatal("should not be empty")
- }
- }
-
- if result := <-store.Channel().SearchInTeam(o1.TeamId, "blargh"); result.Err != nil {
- t.Fatal(result.Err)
- } else {
- channels := result.Data.(*model.ChannelList)
- if len(*channels) != 0 {
- t.Fatal("should be empty")
- }
- }
-}
-
-func TestChannelStoreGetMembersByIds(t *testing.T) {
- Setup()
-
- o1 := model.Channel{}
- o1.TeamId = model.NewId()
- o1.DisplayName = "ChannelA"
- o1.Name = "zz" + model.NewId() + "b"
- o1.Type = model.CHANNEL_OPEN
- Must(store.Channel().Save(&o1))
-
- m1 := &model.ChannelMember{ChannelId: o1.Id, UserId: model.NewId(), NotifyProps: model.GetDefaultChannelNotifyProps()}
- Must(store.Channel().SaveMember(m1))
-
- if r := <-store.Channel().GetMembersByIds(m1.ChannelId, []string{m1.UserId}); r.Err != nil {
- t.Fatal(r.Err)
- } else {
- rm1 := (*r.Data.(*model.ChannelMembers))[0]
-
- if rm1.ChannelId != m1.ChannelId {
- t.Fatal("bad team id")
- }
-
- if rm1.UserId != m1.UserId {
- t.Fatal("bad user id")
- }
- }
-
- m2 := &model.ChannelMember{ChannelId: o1.Id, UserId: model.NewId(), NotifyProps: model.GetDefaultChannelNotifyProps()}
- Must(store.Channel().SaveMember(m2))
-
- if r := <-store.Channel().GetMembersByIds(m1.ChannelId, []string{m1.UserId, m2.UserId, model.NewId()}); r.Err != nil {
- t.Fatal(r.Err)
- } else {
- rm := (*r.Data.(*model.ChannelMembers))
-
- if len(rm) != 2 {
- t.Fatal("return wrong number of results")
- }
- }
-
- if r := <-store.Channel().GetMembersByIds(m1.ChannelId, []string{}); r.Err == nil {
- t.Fatal("empty user ids - should have failed")
- }
-}
-
-func TestChannelStoreAnalyticsDeletedTypeCount(t *testing.T) {
- Setup()
-
- o1 := model.Channel{}
- o1.TeamId = model.NewId()
- o1.DisplayName = "ChannelA"
- o1.Name = "zz" + 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 = "zz" + model.NewId() + "b"
- o2.Type = model.CHANNEL_OPEN
- Must(store.Channel().Save(&o2))
-
- p3 := model.Channel{}
- p3.TeamId = model.NewId()
- p3.DisplayName = "Channel3"
- p3.Name = "zz" + model.NewId() + "b"
- p3.Type = model.CHANNEL_PRIVATE
- Must(store.Channel().Save(&p3))
-
- u1 := &model.User{}
- u1.Email = model.NewId()
- u1.Nickname = model.NewId()
- Must(store.User().Save(u1))
-
- u2 := &model.User{}
- u2.Email = model.NewId()
- u2.Nickname = model.NewId()
- Must(store.User().Save(u2))
-
- var d4 *model.Channel
- if result := <-store.Channel().CreateDirectChannel(u1.Id, u2.Id); result.Err != nil {
- t.Fatalf(result.Err.Error())
- } else {
- d4 = result.Data.(*model.Channel)
- }
-
- var openStartCount int64
- if result := <-store.Channel().AnalyticsDeletedTypeCount("", "O"); result.Err != nil {
- t.Fatal(result.Err.Error())
- } else {
- openStartCount = result.Data.(int64)
- }
-
- var privateStartCount int64
- if result := <-store.Channel().AnalyticsDeletedTypeCount("", "P"); result.Err != nil {
- t.Fatal(result.Err.Error())
- } else {
- privateStartCount = result.Data.(int64)
- }
-
- var directStartCount int64
- if result := <-store.Channel().AnalyticsDeletedTypeCount("", "D"); result.Err != nil {
- t.Fatal(result.Err.Error())
- } else {
- directStartCount = result.Data.(int64)
- }
-
- Must(store.Channel().Delete(o1.Id, model.GetMillis()))
- Must(store.Channel().Delete(o2.Id, model.GetMillis()))
- Must(store.Channel().Delete(p3.Id, model.GetMillis()))
- Must(store.Channel().Delete(d4.Id, model.GetMillis()))
-
- if result := <-store.Channel().AnalyticsDeletedTypeCount("", "O"); result.Err != nil {
- t.Fatal(result.Err.Error())
- } else {
- if result.Data.(int64) != openStartCount+2 {
- t.Fatalf("Wrong open channel deleted count.")
- }
- }
-
- if result := <-store.Channel().AnalyticsDeletedTypeCount("", "P"); result.Err != nil {
- t.Fatal(result.Err.Error())
- } else {
- if result.Data.(int64) != privateStartCount+1 {
- t.Fatalf("Wrong private channel deleted count.")
- }
- }
-
- if result := <-store.Channel().AnalyticsDeletedTypeCount("", "D"); result.Err != nil {
- t.Fatal(result.Err.Error())
- } else {
- if result.Data.(int64) != directStartCount+1 {
- t.Fatalf("Wrong direct channel deleted count.")
- }
- }
-}
-
-func TestChannelStoreGetPinnedPosts(t *testing.T) {
- Setup()
-
- o1 := Must(store.Channel().Save(&model.Channel{
- TeamId: model.NewId(),
- DisplayName: "Name",
- Name: "zz" + model.NewId() + "b",
- Type: model.CHANNEL_OPEN,
- })).(*model.Channel)
-
- p1 := Must(store.Post().Save(&model.Post{
- UserId: model.NewId(),
- ChannelId: o1.Id,
- Message: "test",
- IsPinned: true,
- })).(*model.Post)
-
- if r1 := <-store.Channel().GetPinnedPosts(o1.Id); r1.Err != nil {
- t.Fatal(r1.Err)
- } else if r1.Data.(*model.PostList).Posts[p1.Id] == nil {
- t.Fatal("didn't return relevant pinned posts")
- }
-
- o2 := Must(store.Channel().Save(&model.Channel{
- TeamId: model.NewId(),
- DisplayName: "Name",
- Name: "zz" + model.NewId() + "b",
- Type: model.CHANNEL_OPEN,
- })).(*model.Channel)
-
- Must(store.Post().Save(&model.Post{
- UserId: model.NewId(),
- ChannelId: o2.Id,
- Message: "test",
- }))
-
- if r2 := <-store.Channel().GetPinnedPosts(o2.Id); r2.Err != nil {
- t.Fatal(r2.Err)
- } else if len(r2.Data.(*model.PostList).Posts) != 0 {
- t.Fatal("wasn't supposed to return posts")
- }
-}