mattermost-community-enterp.../channels/app/plugin_health_check_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

89 lines
2.4 KiB
Go

// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
// See LICENSE.txt for license information.
package app
import (
"testing"
"github.com/stretchr/testify/require"
"github.com/mattermost/mattermost/server/public/model"
"github.com/mattermost/mattermost/server/public/plugin"
)
func TestHealthCheckJob(t *testing.T) {
mainHelper.Parallel(t)
th := Setup(t)
defer th.TearDown()
tearDown, _, _ := SetAppEnvironmentWithPlugins(t, []string{
`
package main
import (
"github.com/mattermost/mattermost/server/public/model"
"github.com/mattermost/mattermost/server/public/plugin"
)
type MyPlugin struct {
plugin.MattermostPlugin
}
func (p *MyPlugin) MessageWillBePosted(c *plugin.Context, post *model.Post) (*model.Post, string) {
panic("Uncaught error")
}
func main() {
plugin.ClientMain(&MyPlugin{})
}
`,
}, th.App, th.NewPluginAPI)
defer tearDown()
env := th.App.GetPluginsEnvironment()
job := env.GetPluginHealthCheckJob()
require.NotNil(t, job)
bundles := env.Active()
require.Equal(t, 1, len(bundles))
id := bundles[0].Manifest.Id
// First health check
hooks, err := env.HooksForPlugin(id)
require.NoError(t, err)
hooks.MessageWillBePosted(&plugin.Context{}, &model.Post{})
job.CheckPlugin(id)
bundles = env.Active()
require.Equal(t, 1, len(bundles))
require.Equal(t, id, bundles[0].Manifest.Id)
require.Equal(t, model.PluginStateRunning, env.GetPluginState(id))
// Second health check
hooks, err = env.HooksForPlugin(id)
require.NoError(t, err)
hooks.MessageWillBePosted(&plugin.Context{}, &model.Post{})
job.CheckPlugin(id)
bundles = env.Active()
require.Equal(t, 1, len(bundles))
require.Equal(t, id, bundles[0].Manifest.Id)
require.Equal(t, model.PluginStateRunning, env.GetPluginState(id))
// Third health check, plugin should be deactivated by the job
hooks, err = env.HooksForPlugin(id)
require.NoError(t, err)
hooks.MessageWillBePosted(&plugin.Context{}, &model.Post{})
job.CheckPlugin(id)
bundles = env.Active()
require.Equal(t, 0, len(bundles))
require.Equal(t, model.PluginStateFailedToStayRunning, env.GetPluginState(id))
// Activated manually, plugin should stay active
_, _, err = env.Activate(id)
require.NoError(t, err)
job.CheckPlugin(id)
bundles = env.Active()
require.Equal(t, 1, len(bundles))
require.Equal(t, model.PluginStateRunning, env.GetPluginState(id))
}