mattermost-community-enterp.../channels/app/usage_test.go
Claude ec1f89217a Merge: Complete Mattermost Server with Community Enterprise
Full Mattermost server source with integrated Community Enterprise features.
Includes vendor directory for offline/air-gapped builds.

Structure:
- enterprise-impl/: Enterprise feature implementations
- enterprise-community/: Init files that register implementations
- enterprise/: Bridge imports (community_imports.go)
- vendor/: All dependencies for offline builds

Build (online):
  go build ./cmd/mattermost

Build (offline/air-gapped):
  go build -mod=vendor ./cmd/mattermost

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
2025-12-17 23:59:07 +09:00

51 lines
1.4 KiB
Go

// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
// See LICENSE.txt for license information.
package app
import (
"errors"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/mock"
"github.com/mattermost/mattermost/server/v8/channels/store/storetest/mocks"
)
func TestGetPostsUsage(t *testing.T) {
mainHelper.Parallel(t)
t.Run("returns error when AnalyticsPostCount fails", func(t *testing.T) {
th := SetupWithStoreMock(t)
defer th.TearDown()
errMsg := "Test posts count error"
mockStore := th.App.Srv().Store().(*mocks.Store)
mockPostStore := mocks.PostStore{}
mockPostStore.On("AnalyticsPostCount", mock.Anything).Return(int64(0), errors.New(errMsg))
mockStore.On("Post").Return(&mockPostStore)
usage, appErr := th.App.GetPostsUsage()
assert.Zero(t, usage)
assert.ErrorContains(t, appErr, errMsg)
})
t.Run("returns rounded off count when AnalyticsPostCount returns valid count", func(t *testing.T) {
th := SetupWithStoreMock(t)
defer th.TearDown()
var mockCount int64 = 4321
var expected int64 = 4000
mockStore := th.App.Srv().Store().(*mocks.Store)
mockPostStore := mocks.PostStore{}
mockPostStore.On("AnalyticsPostCount", mock.Anything).Return(mockCount, nil)
mockStore.On("Post").Return(&mockPostStore)
count, appErr := th.App.GetPostsUsage()
assert.Nil(t, appErr)
assert.Equal(t, expected, count)
})
}