Added ChannelExtra.MemberCount field to reflect the actual member count when we truncate the list of members to 20

1 files changed, 4 insertions, 4 deletions

diff --git a/store/sql_channel_store_test.go b/store/sql_channel_store_test.go
index f6a0fb713..8662fcbd3 100644
--- a/store/sql_channel_store_test.go
+++ b/store/sql_channel_store_test.go
@@ -339,15 +339,15 @@ func TestChannelMemberStore(t *testing.T) {
 		t.Fatal("Member update time incorrect")
 	}
 
-	members := (<-store.Channel().GetMembers(o1.ChannelId)).Data.([]model.ChannelMember)
-	if len(members) != 2 {
+	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))
 
-	members = (<-store.Channel().GetMembers(o1.ChannelId)).Data.([]model.ChannelMember)
-	if len(members) != 1 {
+	count = (<-store.Channel().GetMemberCount(o1.ChannelId)).Data.(int64)
+	if count != 1 {
 		t.Fatal("should have removed 1 member")
 	}