From 139cb52c99ac525f44a280803447bbbd88369f23 Mon Sep 17 00:00:00 2001 From: enahum Date: Wed, 21 Dec 2016 16:20:11 -0300 Subject: Fix teams unread count (#4848) * Fix teams unread * Moving logic out of the store * fix unit test --- store/sql_team_store_test.go | 23 ++++++++++++++++++----- 1 file changed, 18 insertions(+), 5 deletions(-) (limited to 'store/sql_team_store_test.go') diff --git a/store/sql_team_store_test.go b/store/sql_team_store_test.go index 5e6abd102..8a302a9e2 100644 --- a/store/sql_team_store_test.go +++ b/store/sql_team_store_test.go @@ -555,9 +555,15 @@ func TestMyTeamMembersUnread(t *testing.T) { if r1 := <-store.Team().GetTeamsUnreadForUser("", uid); r1.Err != nil { t.Fatal(r1.Err) } else { - ms := r1.Data.([]*model.TeamUnread) - - if len(ms) != 2 { + ms := r1.Data.([]*model.ChannelUnread) + membersMap := make(map[string]bool) + for i := range ms { + id := ms[i].TeamId + if _, ok := membersMap[id]; !ok { + membersMap[id] = true + } + } + if len(membersMap) != 2 { t.Fatal("Should be the unreads for all the teams") } } @@ -565,9 +571,16 @@ func TestMyTeamMembersUnread(t *testing.T) { if r2 := <-store.Team().GetTeamsUnreadForUser(teamId1, uid); r2.Err != nil { t.Fatal(r2.Err) } else { - ms := r2.Data.([]*model.TeamUnread) + ms := r2.Data.([]*model.ChannelUnread) + membersMap := make(map[string]bool) + for i := range ms { + id := ms[i].TeamId + if _, ok := membersMap[id]; !ok { + membersMap[id] = true + } + } - if len(ms) != 1 { + if len(membersMap) != 1 { t.Fatal("Should be the unreads for just one team") } } -- cgit v1.2.3-1-g7c22