mirror of
https://github.com/mattermost/mattermost.git
synced 2026-04-14 05:28:31 -04:00
131 lines
4.3 KiB
Go
131 lines
4.3 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package model
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestChannelMemberIsValid(t *testing.T) {
|
|
o := ChannelMember{}
|
|
|
|
require.NotNil(t, o.IsValid(), "should be invalid")
|
|
|
|
o.ChannelId = NewId()
|
|
require.NotNil(t, o.IsValid(), "should be invalid")
|
|
|
|
o.UserId = NewId()
|
|
require.NotNil(t, o.IsValid(), "should be invalid because of missing notify props")
|
|
|
|
o.NotifyProps = GetDefaultChannelNotifyProps()
|
|
require.Nil(t, o.IsValid(), "should be valid")
|
|
|
|
o.NotifyProps["desktop"] = "junk"
|
|
require.NotNil(t, o.IsValid(), "should be invalid")
|
|
|
|
o.NotifyProps["desktop"] = "123456789012345678901"
|
|
require.NotNil(t, o.IsValid(), "should be invalid")
|
|
|
|
o.NotifyProps["desktop"] = ChannelNotifyAll
|
|
require.Nil(t, o.IsValid(), "should be valid")
|
|
|
|
o.NotifyProps["mark_unread"] = "123456789012345678901"
|
|
require.NotNil(t, o.IsValid(), "should be invalid")
|
|
|
|
o.NotifyProps["mark_unread"] = ChannelMarkUnreadAll
|
|
require.Nil(t, o.IsValid(), "should be valid")
|
|
|
|
o.Roles = ""
|
|
require.Nil(t, o.IsValid(), "should be invalid")
|
|
|
|
o.NotifyProps["property"] = strings.Repeat("Z", ChannelMemberNotifyPropsMaxRunes)
|
|
require.NotNil(t, o.IsValid(), "should be invalid")
|
|
}
|
|
|
|
func TestIsChannelMemberNotifyPropsValid(t *testing.T) {
|
|
t.Run("should require certain fields unless allowMissingFields is true", func(t *testing.T) {
|
|
notifyProps := map[string]string{}
|
|
|
|
err := IsChannelMemberNotifyPropsValid(notifyProps, false)
|
|
assert.NotNil(t, err)
|
|
|
|
err = IsChannelMemberNotifyPropsValid(notifyProps, true)
|
|
assert.Nil(t, err)
|
|
})
|
|
}
|
|
|
|
func TestChannelMemberSanitizeForCurrentUser(t *testing.T) {
|
|
currentUserId := NewId()
|
|
otherUserId := NewId()
|
|
channelId := NewId()
|
|
|
|
t.Run("should not sanitize current user's own membership", func(t *testing.T) {
|
|
member := &ChannelMember{
|
|
ChannelId: channelId,
|
|
UserId: currentUserId,
|
|
LastViewedAt: 1234567890000,
|
|
LastUpdateAt: 1234567890000,
|
|
NotifyProps: GetDefaultChannelNotifyProps(),
|
|
}
|
|
|
|
originalLastViewedAt := member.LastViewedAt
|
|
originalLastUpdateAt := member.LastUpdateAt
|
|
|
|
member.SanitizeForCurrentUser(currentUserId)
|
|
|
|
assert.Equal(t, originalLastViewedAt, member.LastViewedAt, "LastViewedAt should not be sanitized for current user")
|
|
assert.Equal(t, originalLastUpdateAt, member.LastUpdateAt, "LastUpdateAt should not be sanitized for current user")
|
|
})
|
|
|
|
t.Run("should sanitize other users' membership data", func(t *testing.T) {
|
|
member := &ChannelMember{
|
|
ChannelId: channelId,
|
|
UserId: otherUserId,
|
|
LastViewedAt: 1234567890000,
|
|
LastUpdateAt: 1234567890000,
|
|
NotifyProps: GetDefaultChannelNotifyProps(),
|
|
}
|
|
|
|
member.SanitizeForCurrentUser(currentUserId)
|
|
|
|
assert.Equal(t, int64(-1), member.LastViewedAt, "LastViewedAt should be sanitized for other users")
|
|
assert.Equal(t, int64(-1), member.LastUpdateAt, "LastUpdateAt should be sanitized for other users")
|
|
})
|
|
|
|
t.Run("should preserve other fields when sanitizing", func(t *testing.T) {
|
|
member := &ChannelMember{
|
|
ChannelId: channelId,
|
|
UserId: otherUserId,
|
|
Roles: "channel_user",
|
|
LastViewedAt: 1234567890000,
|
|
LastUpdateAt: 1234567890000,
|
|
MsgCount: 100,
|
|
MentionCount: 5,
|
|
NotifyProps: GetDefaultChannelNotifyProps(),
|
|
SchemeUser: true,
|
|
SchemeAdmin: false,
|
|
ExplicitRoles: "",
|
|
}
|
|
|
|
originalRoles := member.Roles
|
|
originalMsgCount := member.MsgCount
|
|
originalMentionCount := member.MentionCount
|
|
originalSchemeUser := member.SchemeUser
|
|
originalSchemeAdmin := member.SchemeAdmin
|
|
|
|
member.SanitizeForCurrentUser(currentUserId)
|
|
|
|
assert.Equal(t, int64(-1), member.LastViewedAt, "LastViewedAt should be sanitized")
|
|
assert.Equal(t, int64(-1), member.LastUpdateAt, "LastUpdateAt should be sanitized")
|
|
assert.Equal(t, originalRoles, member.Roles, "Roles should be preserved")
|
|
assert.Equal(t, originalMsgCount, member.MsgCount, "MsgCount should be preserved")
|
|
assert.Equal(t, originalMentionCount, member.MentionCount, "MentionCount should be preserved")
|
|
assert.Equal(t, originalSchemeUser, member.SchemeUser, "SchemeUser should be preserved")
|
|
assert.Equal(t, originalSchemeAdmin, member.SchemeAdmin, "SchemeAdmin should be preserved")
|
|
})
|
|
}
|