cffe921e62
* Prevent cloud limited installations from restoring teams when at or above the teams limit * Code clean up * fix i18n * Actually fix i18n * updates for govet * Update model/client4.go Co-authored-by: Vishal <vish9812@gmail.com> * [MM-44397] Restrict team creation based on subscription limits (#20282) * restrict team creation based on limits * Update api4/team.go Co-authored-by: Maria A Nunez <maria.nunez@mattermost.com> * Fix tests Co-authored-by: Mattermod <mattermod@users.noreply.github.com> Co-authored-by: Maria A Nunez <maria.nunez@mattermost.com> * Add additional field to teamsusage * Fix * remove useless test * Fix error for team creation * Remove apostrophe Co-authored-by: Mattermod <mattermod@users.noreply.github.com> Co-authored-by: Vishal <vish9812@gmail.com> Co-authored-by: Maria A Nunez <maria.nunez@mattermost.com>
105 строки
2.7 KiB
Go
105 строки
2.7 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package api4
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/mattermost/mattermost-server/v6/model"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestGetPostsUsage(t *testing.T) {
|
|
t.Run("unauthenticated users can not access", func(t *testing.T) {
|
|
th := Setup(t)
|
|
defer th.TearDown()
|
|
|
|
th.Client.Logout()
|
|
|
|
usage, r, err := th.Client.GetPostsUsage()
|
|
assert.Error(t, err)
|
|
assert.Nil(t, usage)
|
|
assert.Equal(t, http.StatusUnauthorized, r.StatusCode)
|
|
})
|
|
|
|
t.Run("good request returns response", func(t *testing.T) {
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
|
|
for i := 0; i < 14; i++ {
|
|
th.CreatePost()
|
|
}
|
|
|
|
total, err := th.Server.Store.Post().AnalyticsPostCount(&model.PostCountOptions{ExcludeDeleted: true})
|
|
require.NoError(t, err)
|
|
usersOnly, err := th.Server.Store.Post().AnalyticsPostCount(&model.PostCountOptions{ExcludeDeleted: true, UsersPostsOnly: true})
|
|
require.NoError(t, err)
|
|
|
|
require.GreaterOrEqual(t, usersOnly, int64(14))
|
|
require.LessOrEqual(t, usersOnly, int64(20))
|
|
require.GreaterOrEqual(t, total, usersOnly)
|
|
|
|
usage, r, err := th.Client.GetPostsUsage()
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, http.StatusOK, r.StatusCode)
|
|
assert.NotNil(t, usage)
|
|
assert.Equal(t, int64(10), usage.Count)
|
|
})
|
|
}
|
|
|
|
func TestGetTeamsUsage(t *testing.T) {
|
|
t.Run("unauthenticated users can not access", func(t *testing.T) {
|
|
th := Setup(t)
|
|
defer th.TearDown()
|
|
|
|
th.Client.Logout()
|
|
|
|
usage, r, err := th.Client.GetTeamsUsage()
|
|
assert.Error(t, err)
|
|
assert.Nil(t, usage)
|
|
assert.Equal(t, http.StatusUnauthorized, r.StatusCode)
|
|
})
|
|
|
|
t.Run("good request returns response", func(t *testing.T) {
|
|
// Following calls create a total of 3 teams
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
th.CreateTeam()
|
|
th.CreateTeam()
|
|
|
|
usage, r, err := th.Client.GetTeamsUsage()
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, http.StatusOK, r.StatusCode)
|
|
assert.NotNil(t, usage)
|
|
assert.Equal(t, int64(3), usage.Active)
|
|
})
|
|
}
|
|
|
|
func TestGetIntegrationsUsage(t *testing.T) {
|
|
t.Run("unauthenticated users can not access", func(t *testing.T) {
|
|
th := Setup(t)
|
|
defer th.TearDown()
|
|
|
|
th.Client.Logout()
|
|
|
|
usage, r, err := th.Client.GetIntegrationsUsage()
|
|
assert.Error(t, err)
|
|
assert.Nil(t, usage)
|
|
assert.Equal(t, http.StatusUnauthorized, r.StatusCode)
|
|
})
|
|
|
|
t.Run("good request returns response", func(t *testing.T) {
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
|
|
usage, r, err := th.Client.GetIntegrationsUsage()
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, http.StatusOK, r.StatusCode)
|
|
assert.NotNil(t, usage)
|
|
assert.Equal(t, 0, usage.Enabled)
|
|
})
|
|
}
|