mirror of
https://github.com/mattermost/mattermost.git
synced 2025-02-25 18:55:24 -06:00
a4507327a7
Refactor mlog - simplify mlog by removing redundant code - remove Zap dependency - update unit test helpers - update logging config - update auditing
85 lines
2.5 KiB
Go
85 lines
2.5 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package plugin
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/mattermost/mattermost-server/v6/model"
|
|
"github.com/mattermost/mattermost-server/v6/shared/mlog"
|
|
"github.com/mattermost/mattermost-server/v6/utils"
|
|
)
|
|
|
|
func TestSupervisor(t *testing.T) {
|
|
for name, f := range map[string]func(*testing.T){
|
|
"Supervisor_InvalidExecutablePath": testSupervisorInvalidExecutablePath,
|
|
"Supervisor_NonExistentExecutablePath": testSupervisorNonExistentExecutablePath,
|
|
"Supervisor_StartTimeout": testSupervisorStartTimeout,
|
|
} {
|
|
t.Run(name, f)
|
|
}
|
|
}
|
|
|
|
func testSupervisorInvalidExecutablePath(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "")
|
|
require.NoError(t, err)
|
|
defer os.RemoveAll(dir)
|
|
|
|
ioutil.WriteFile(filepath.Join(dir, "plugin.json"), []byte(`{"id": "foo", "server": {"executable": "/foo/../../backend.exe"}}`), 0600)
|
|
|
|
bundle := model.BundleInfoForPath(dir)
|
|
log := mlog.CreateConsoleTestLogger(true, mlog.LvlError)
|
|
defer log.Shutdown()
|
|
supervisor, err := newSupervisor(bundle, nil, nil, log, nil)
|
|
assert.Nil(t, supervisor)
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func testSupervisorNonExistentExecutablePath(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "")
|
|
require.NoError(t, err)
|
|
defer os.RemoveAll(dir)
|
|
|
|
ioutil.WriteFile(filepath.Join(dir, "plugin.json"), []byte(`{"id": "foo", "server": {"executable": "thisfileshouldnotexist"}}`), 0600)
|
|
|
|
bundle := model.BundleInfoForPath(dir)
|
|
log := mlog.CreateConsoleTestLogger(true, mlog.LvlError)
|
|
defer log.Shutdown()
|
|
supervisor, err := newSupervisor(bundle, nil, nil, log, nil)
|
|
require.Error(t, err)
|
|
require.Nil(t, supervisor)
|
|
}
|
|
|
|
// If plugin development goes really wrong, let's make sure plugin activation won't block forever.
|
|
func testSupervisorStartTimeout(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "")
|
|
require.NoError(t, err)
|
|
defer os.RemoveAll(dir)
|
|
|
|
backend := filepath.Join(dir, "backend.exe")
|
|
utils.CompileGo(t, `
|
|
package main
|
|
|
|
func main() {
|
|
for {
|
|
}
|
|
}
|
|
`, backend)
|
|
|
|
ioutil.WriteFile(filepath.Join(dir, "plugin.json"), []byte(`{"id": "foo", "server": {"executable": "backend.exe"}}`), 0600)
|
|
|
|
bundle := model.BundleInfoForPath(dir)
|
|
log := mlog.CreateConsoleTestLogger(true, mlog.LvlError)
|
|
defer log.Shutdown()
|
|
supervisor, err := newSupervisor(bundle, nil, nil, log, nil)
|
|
require.Error(t, err)
|
|
require.Nil(t, supervisor)
|
|
}
|