mirror of
https://github.com/mattermost/mattermost.git
synced 2025-02-25 18:55:24 -06:00
899ab31fff
* Implement experimental REST API endpoints for plugins * Updates per feedback and rebase * Update tests * Further updates * Update extraction of plugins * Use OS temp dir for plugins instead of search path * Fail extraction on paths that attempt to traverse upward * Update pluginenv ActivePlugins()
118 lines
2.6 KiB
Go
118 lines
2.6 KiB
Go
package rpcplugin
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/mattermost/platform/model"
|
|
)
|
|
|
|
func TestSupervisor(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "")
|
|
require.NoError(t, err)
|
|
defer os.RemoveAll(dir)
|
|
|
|
backend := filepath.Join(dir, "backend.exe")
|
|
compileGo(t, `
|
|
package main
|
|
|
|
import (
|
|
"github.com/mattermost/platform/plugin/rpcplugin"
|
|
)
|
|
|
|
type MyPlugin struct {}
|
|
|
|
func main() {
|
|
rpcplugin.Main(&MyPlugin{})
|
|
}
|
|
`, backend)
|
|
|
|
ioutil.WriteFile(filepath.Join(dir, "plugin.json"), []byte(`{"id": "foo", "backend": {"executable": "backend.exe"}}`), 0600)
|
|
|
|
bundle := model.BundleInfoForPath(dir)
|
|
supervisor, err := SupervisorProvider(bundle)
|
|
require.NoError(t, err)
|
|
require.NoError(t, supervisor.Start())
|
|
require.NoError(t, supervisor.Hooks().OnActivate(nil))
|
|
require.NoError(t, supervisor.Stop())
|
|
}
|
|
|
|
// If plugin development goes really wrong, let's make sure plugin activation won't block forever.
|
|
func TestSupervisor_StartTimeout(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "")
|
|
require.NoError(t, err)
|
|
defer os.RemoveAll(dir)
|
|
|
|
backend := filepath.Join(dir, "backend.exe")
|
|
compileGo(t, `
|
|
package main
|
|
|
|
func main() {
|
|
for {
|
|
}
|
|
}
|
|
`, backend)
|
|
|
|
ioutil.WriteFile(filepath.Join(dir, "plugin.json"), []byte(`{"id": "foo", "backend": {"executable": "backend.exe"}}`), 0600)
|
|
|
|
bundle := model.BundleInfoForPath(dir)
|
|
supervisor, err := SupervisorProvider(bundle)
|
|
require.NoError(t, err)
|
|
require.Error(t, supervisor.Start())
|
|
}
|
|
|
|
// Crashed plugins should be relaunched.
|
|
func TestSupervisor_PluginCrash(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "")
|
|
require.NoError(t, err)
|
|
defer os.RemoveAll(dir)
|
|
|
|
backend := filepath.Join(dir, "backend.exe")
|
|
compileGo(t, `
|
|
package main
|
|
|
|
import (
|
|
"os"
|
|
|
|
"github.com/mattermost/platform/plugin"
|
|
"github.com/mattermost/platform/plugin/rpcplugin"
|
|
)
|
|
|
|
type MyPlugin struct {}
|
|
|
|
func (p *MyPlugin) OnActivate(api plugin.API) error {
|
|
os.Exit(1)
|
|
return nil
|
|
}
|
|
|
|
func main() {
|
|
rpcplugin.Main(&MyPlugin{})
|
|
}
|
|
`, backend)
|
|
|
|
ioutil.WriteFile(filepath.Join(dir, "plugin.json"), []byte(`{"id": "foo", "backend": {"executable": "backend.exe"}}`), 0600)
|
|
|
|
bundle := model.BundleInfoForPath(dir)
|
|
supervisor, err := SupervisorProvider(bundle)
|
|
require.NoError(t, err)
|
|
require.NoError(t, supervisor.Start())
|
|
require.Error(t, supervisor.Hooks().OnActivate(nil))
|
|
|
|
recovered := false
|
|
for i := 0; i < 30; i++ {
|
|
if supervisor.Hooks().OnDeactivate() == nil {
|
|
recovered = true
|
|
break
|
|
}
|
|
time.Sleep(time.Millisecond * 100)
|
|
}
|
|
assert.True(t, recovered)
|
|
require.NoError(t, supervisor.Stop())
|
|
}
|