summaryrefslogtreecommitdiffstats
path: root/store/sql_team_store_test.go
diff options
context:
space:
mode:
authorJoram Wilander <jwawilander@gmail.com>2016-11-04 12:27:19 -0400
committerChristopher Speller <crspeller@gmail.com>2016-11-04 12:27:19 -0400
commit00787974d0a87b1a54f15cf75d2dab398546b87e (patch)
tree6f41b56b10183b6543309b790cd13b1fbf7559af /store/sql_team_store_test.go
parent263f29068386cdd3b5393e00ce97f776532c723f (diff)
downloadchat-00787974d0a87b1a54f15cf75d2dab398546b87e.tar.gz
chat-00787974d0a87b1a54f15cf75d2dab398546b87e.tar.bz2
chat-00787974d0a87b1a54f15cf75d2dab398546b87e.zip
PLT-4481 Fix member count for team user lists and channel invite list (#4422)
* Fix member count for team user lists and channel invite list * Fix client unit test
Diffstat (limited to 'store/sql_team_store_test.go')
-rw-r--r--store/sql_team_store_test.go32
1 files changed, 28 insertions, 4 deletions
diff --git a/store/sql_team_store_test.go b/store/sql_team_store_test.go
index 46215d9be..0e472a961 100644
--- a/store/sql_team_store_test.go
+++ b/store/sql_team_store_test.go
@@ -482,11 +482,27 @@ func TestTeamStoreMemberCount(t *testing.T) {
u1.Email = model.NewId()
Must(store.User().Save(u1))
+ u2 := &model.User{}
+ u2.Email = model.NewId()
+ u2.DeleteAt = 1
+ Must(store.User().Save(u2))
+
teamId1 := model.NewId()
m1 := &model.TeamMember{TeamId: teamId1, UserId: u1.Id}
Must(store.Team().SaveMember(m1))
- if result := <-store.Team().GetMemberCount(teamId1); result.Err != nil {
+ m2 := &model.TeamMember{TeamId: teamId1, UserId: u2.Id}
+ Must(store.Team().SaveMember(m2))
+
+ if result := <-store.Team().GetTotalMemberCount(teamId1); result.Err != nil {
+ t.Fatal(result.Err)
+ } else {
+ if result.Data.(int64) != 2 {
+ t.Fatal("wrong count")
+ }
+ }
+
+ if result := <-store.Team().GetActiveMemberCount(teamId1); result.Err != nil {
t.Fatal(result.Err)
} else {
if result.Data.(int64) != 1 {
@@ -494,10 +510,18 @@ func TestTeamStoreMemberCount(t *testing.T) {
}
}
- m2 := &model.TeamMember{TeamId: teamId1, UserId: model.NewId()}
- Must(store.Team().SaveMember(m2))
+ m3 := &model.TeamMember{TeamId: teamId1, UserId: model.NewId()}
+ Must(store.Team().SaveMember(m3))
+
+ if result := <-store.Team().GetTotalMemberCount(teamId1); result.Err != nil {
+ t.Fatal(result.Err)
+ } else {
+ if result.Data.(int64) != 2 {
+ t.Fatal("wrong count")
+ }
+ }
- if result := <-store.Team().GetMemberCount(teamId1); result.Err != nil {
+ if result := <-store.Team().GetActiveMemberCount(teamId1); result.Err != nil {
t.Fatal(result.Err)
} else {
if result.Data.(int64) != 1 {