mattermost-community-enterp.../public/pluginapi/logrus_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

86 lines
2.2 KiB
Go

package pluginapi_test
import (
"testing"
"github.com/mattermost/mattermost/server/public/plugin/plugintest"
"github.com/sirupsen/logrus"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/mock"
"github.com/mattermost/mattermost/server/public/pluginapi"
)
func TestLogrus(t *testing.T) {
testCases := []struct {
Level logrus.Level
APICall string
}{
{logrus.PanicLevel, "LogError"},
{logrus.FatalLevel, "LogError"},
{logrus.ErrorLevel, "LogError"},
{logrus.WarnLevel, "LogWarn"},
{logrus.InfoLevel, "LogInfo"},
{logrus.DebugLevel, "LogDebug"},
{logrus.TraceLevel, "LogDebug"},
}
for _, testCase := range testCases {
t.Run(testCase.Level.String(), func(t *testing.T) {
logger := logrus.New()
logger.SetLevel(logrus.TraceLevel) // not testing logrus filtering
logger.ReportCaller = true
api := &plugintest.API{}
defer api.AssertExpectations(t)
client := pluginapi.NewClient(api, &plugintest.Driver{})
pluginapi.ConfigureLogrus(logger, client)
// Parameter order of map is non-deterministic, so expect either.
api.On(testCase.APICall, mock.Anything, mock.Anything, mock.Anything, mock.Anything, mock.Anything, mock.Anything, mock.Anything)
entry := logger.WithFields(logrus.Fields{
"a": "a",
"b": 1,
})
if testCase.Level == logrus.PanicLevel {
done := make(chan bool)
go func() {
defer func() {
r := recover()
assert.NotNil(t, r, "expected panic")
close(done)
}()
entry.Panic("message")
}()
<-done
} else {
entry.Log(testCase.Level, "message")
}
// Assert the required API call was executed at most once.
if api.AssertNumberOfCalls(t, testCase.APICall, 1) {
call := api.Calls[0]
for i := 1; i < len(call.Arguments)-1; i += 2 {
argument := call.Arguments[i]
value := call.Arguments[i+1]
switch argument {
case "a":
assert.Equal(t, "a", value, "unexpected value for a")
case "b":
assert.Equal(t, "1", value, "unexpected value for b")
case "plugin_caller":
assert.IsType(t, "string", value)
default:
assert.Fail(t, "unexpected argument and value", "%v: %v", argument, value)
}
}
}
})
}
}