upstream-mattermost/server/channels/app/teams/teams_test.go

// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
// See LICENSE.txt for license information.

package teams

import (
	"strings"
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/mattermost/mattermost/server/public/model"
)

func TestCreateTeam(t *testing.T) {
	th := Setup(t)

	id := model.NewId()
	team := &model.Team{
		DisplayName: "dn_" + id,
		Name:        "name" + id,
		Email:       "success+" + id + "@simulator.amazonses.com",
		Type:        model.TeamOpen,
	}

	_, err := th.service.CreateTeam(th.Context, team)
	require.NoError(t, err, "Should create a new team")

	_, err = th.service.CreateTeam(th.Context, team)
	require.Error(t, err, "Should not create a new team - team already exist")
}

func TestCreateTeamWithExperimentalDefaultChannels(t *testing.T) {
	th := Setup(t)
	th.UpdateConfig(func(cfg *model.Config) {
		cfg.TeamSettings.ExperimentalDefaultChannels = []string{"channel-1", "channel-2"}
	})

	id := model.NewId()
	team := &model.Team{
		DisplayName: "dn_" + id,
		Name:        "name" + id,
		Email:       "success+" + id + "@simulator.amazonses.com",
		Type:        model.TeamOpen,
	}

	_, err := th.service.CreateTeam(th.Context, team)
	require.NoError(t, err, "Should create a new team")

	createdTeam, err := th.service.GetTeam(team.Id)
	require.NoError(t, err)
	require.Equal(t, createdTeam.Name, "name"+id)

	channels, err := th.service.channelStore.GetAll(team.Id)
	require.NoError(t, err)
	require.Len(t, channels, 3)

	ch, err := th.service.channelStore.GetByName(team.Id, "town-square", false)
	require.NoError(t, err)
	require.NotNil(t, ch)

	ch, err = th.service.channelStore.GetByName(team.Id, "channel-1", false)
	require.NoError(t, err)
	require.NotNil(t, ch)
	require.Equal(t, ch.DisplayName, "channel-1")

	ch, err = th.service.channelStore.GetByName(team.Id, "channel-2", false)
	require.NoError(t, err)
	require.NotNil(t, ch)
	require.Equal(t, ch.DisplayName, "channel-2")
}

func TestJoinUserToTeam(t *testing.T) {
	th := Setup(t)

	id := model.NewId()
	team := &model.Team{
		DisplayName: "dn_" + id,
		Name:        "name" + id,
		Email:       "success+" + id + "@simulator.amazonses.com",
		Type:        model.TeamOpen,
	}

	_, err := th.service.CreateTeam(th.Context, team)
	require.NoError(t, err, "Should create a new team")

	maxUsersPerTeam := th.service.config().TeamSettings.MaxUsersPerTeam
	defer func() {
		th.UpdateConfig(func(cfg *model.Config) { cfg.TeamSettings.MaxUsersPerTeam = maxUsersPerTeam })
		th.DeleteTeam(team)
	}()
	one := 1
	th.UpdateConfig(func(cfg *model.Config) { cfg.TeamSettings.MaxUsersPerTeam = &one })

	t.Run("new join", func(t *testing.T) {
		user := model.User{Email: strings.ToLower(model.NewId()) + "success+test@example.com", Nickname: "Darth Vader", Username: "vader" + model.NewId(), Password: model.NewTestPassword(), AuthService: ""}
		ruser := th.CreateUser(&user)
		defer th.DeleteUser(&user)

		_, alreadyAdded, err := th.service.JoinUserToTeam(th.Context, team, ruser)
		require.False(t, alreadyAdded, "Should return already added equal to false")
		require.NoError(t, err)
	})

	t.Run("join when you are a member", func(t *testing.T) {
		user := model.User{Email: strings.ToLower(model.NewId()) + "success+test@example.com", Nickname: "Darth Vader", Username: "vader" + model.NewId(), Password: model.NewTestPassword(), AuthService: ""}
		ruser := th.CreateUser(&user)
		defer th.DeleteUser(&user)

		_, _, err := th.service.JoinUserToTeam(th.Context, team, ruser)
		require.NoError(t, err)

		_, alreadyAdded, err := th.service.JoinUserToTeam(th.Context, team, ruser)
		require.True(t, alreadyAdded, "Should return already added")
		require.NoError(t, err)
	})

	t.Run("re-join after leaving", func(t *testing.T) {
		user := model.User{Email: strings.ToLower(model.NewId()) + "success+test@example.com", Nickname: "Darth Vader", Username: "vader" + model.NewId(), Password: model.NewTestPassword(), AuthService: ""}
		ruser := th.CreateUser(&user)
		defer th.DeleteUser(&user)

		member, _, err := th.service.JoinUserToTeam(th.Context, team, ruser)
		require.NoError(t, err)
		err = th.service.RemoveTeamMember(th.Context, member)
		require.NoError(t, err)

		_, alreadyAdded, err := th.service.JoinUserToTeam(th.Context, team, ruser)
		require.False(t, alreadyAdded, "Should return already added equal to false")
		require.NoError(t, err)
	})

	t.Run("guest with the team email does not become team admin", func(t *testing.T) {
		user := model.User{Email: team.Email, Nickname: "Guest Vader", Username: "vader" + model.NewId(), Password: model.NewTestPassword(), Roles: model.SystemGuestRoleId, AuthService: ""}
		ruser := th.CreateUser(&user)
		defer th.DeleteUser(&user)

		member, _, err := th.service.JoinUserToTeam(th.Context, team, ruser)
		require.NoError(t, err)
		require.True(t, member.SchemeGuest)
		require.False(t, member.SchemeUser)
		require.False(t, member.SchemeAdmin)

		err = th.service.RemoveTeamMember(th.Context, member)
		require.NoError(t, err)

		member, _, err = th.service.JoinUserToTeam(th.Context, team, ruser)
		require.NoError(t, err)
		require.True(t, member.SchemeGuest)
		require.False(t, member.SchemeUser)
		require.False(t, member.SchemeAdmin)
	})

	t.Run("regular user with the team email becomes team admin", func(t *testing.T) {
		user := model.User{Email: team.Email, Nickname: "Darth Vader", Username: "vader" + model.NewId(), Password: model.NewTestPassword(), AuthService: ""}
		ruser := th.CreateUser(&user)
		defer th.DeleteUser(&user)

		member, _, err := th.service.JoinUserToTeam(th.Context, team, ruser)
		require.NoError(t, err)
		require.False(t, member.SchemeGuest)
		require.True(t, member.SchemeUser)
		require.True(t, member.SchemeAdmin)
	})

	t.Run("new join with limit problem", func(t *testing.T) {
		user1 := model.User{Email: strings.ToLower(model.NewId()) + "success+test@example.com", Nickname: "Darth Vader", Username: "vader" + model.NewId(), Password: model.NewTestPassword(), AuthService: ""}
		ruser1 := th.CreateUser(&user1)
		user2 := model.User{Email: strings.ToLower(model.NewId()) + "success+test@example.com", Nickname: "Darth Vader", Username: "vader" + model.NewId(), Password: model.NewTestPassword(), AuthService: ""}
		ruser2 := th.CreateUser(&user2)

		defer th.DeleteUser(&user1)
		defer th.DeleteUser(&user2)

		_, _, err := th.service.JoinUserToTeam(th.Context, team, ruser1)
		require.NoError(t, err)

		_, _, err = th.service.JoinUserToTeam(th.Context, team, ruser2)
		require.Error(t, err, "Should fail")
	})

	t.Run("re-join after leaving with limit problem", func(t *testing.T) {
		user1 := model.User{Email: strings.ToLower(model.NewId()) + "success+test@example.com", Nickname: "Darth Vader", Username: "vader" + model.NewId(), Password: model.NewTestPassword(), AuthService: ""}
		ruser1 := th.CreateUser(&user1)

		user2 := model.User{Email: strings.ToLower(model.NewId()) + "success+test@example.com", Nickname: "Darth Vader", Username: "vader" + model.NewId(), Password: model.NewTestPassword(), AuthService: ""}
		ruser2 := th.CreateUser(&user2)

		defer th.DeleteUser(&user1)
		defer th.DeleteUser(&user2)

		member, _, err := th.service.JoinUserToTeam(th.Context, team, ruser1)
		require.NoError(t, err)
		err = th.service.RemoveTeamMember(th.Context, member)
		require.NoError(t, err)
		_, _, err = th.service.JoinUserToTeam(th.Context, team, ruser2)
		require.NoError(t, err)

		_, _, err = th.service.JoinUserToTeam(th.Context, team, ruser1)
		require.Error(t, err, "Should fail")
	})
}