diff options
author | Chris <ccbrown112@gmail.com> | 2017-12-05 08:19:32 -0600 |
---|---|---|
committer | Joram Wilander <jwawilander@gmail.com> | 2017-12-05 09:19:32 -0500 |
commit | 3c7b40063dbba777d27efd2fa4c7093d5507dca6 (patch) | |
tree | 81b77acde7ae6eb2909e2d99782f51f4989dd5cf /plugin/pluginenv | |
parent | 7a1f81cd52c4b58a058ae11e361a80ee3b24d141 (diff) | |
download | chat-3c7b40063dbba777d27efd2fa4c7093d5507dca6.tar.gz chat-3c7b40063dbba777d27efd2fa4c7093d5507dca6.tar.bz2 chat-3c7b40063dbba777d27efd2fa4c7093d5507dca6.zip |
call OnActivate after plugin crash, update example (#7940)
Diffstat (limited to 'plugin/pluginenv')
-rw-r--r-- | plugin/pluginenv/environment.go | 6 | ||||
-rw-r--r-- | plugin/pluginenv/environment_test.go | 26 |
2 files changed, 7 insertions, 25 deletions
diff --git a/plugin/pluginenv/environment.go b/plugin/pluginenv/environment.go index 62eb20e9d..26511c651 100644 --- a/plugin/pluginenv/environment.go +++ b/plugin/pluginenv/environment.go @@ -148,13 +148,9 @@ func (env *Environment) ActivatePlugin(id string) error { if err != nil { return errors.Wrapf(err, "unable to get api for plugin: %v", id) } - if err := supervisor.Start(); err != nil { + if err := supervisor.Start(api); err != nil { return errors.Wrapf(err, "unable to start plugin: %v", id) } - if err := supervisor.Hooks().OnActivate(api); err != nil { - supervisor.Stop() - return errors.Wrapf(err, "unable to activate plugin: %v", id) - } activePlugin.Supervisor = supervisor } diff --git a/plugin/pluginenv/environment_test.go b/plugin/pluginenv/environment_test.go index e4f63cf70..988e5b08f 100644 --- a/plugin/pluginenv/environment_test.go +++ b/plugin/pluginenv/environment_test.go @@ -44,8 +44,8 @@ type MockSupervisor struct { mock.Mock } -func (m *MockSupervisor) Start() error { - return m.Called().Error(0) +func (m *MockSupervisor) Start(api plugin.API) error { + return m.Called(api).Error(0) } func (m *MockSupervisor) Stop() error { @@ -141,12 +141,10 @@ func TestEnvironment(t *testing.T) { provider.On("API").Return(&api, nil) provider.On("Supervisor").Return(&supervisor, nil) - supervisor.On("Start").Return(nil) + supervisor.On("Start", &api).Return(nil) supervisor.On("Stop").Return(nil) supervisor.On("Hooks").Return(&hooks) - hooks.On("OnActivate", &api).Return(nil) - assert.NoError(t, env.ActivatePlugin("foo")) assert.Equal(t, env.ActivePluginIds(), []string{"foo"}) activePlugins = env.ActivePlugins() @@ -238,17 +236,7 @@ func TestEnvironment_ActivatePluginErrors(t *testing.T) { provider.On("API").Return(&api, nil) provider.On("Supervisor").Return(&supervisor, nil) - supervisor.On("Start").Return(fmt.Errorf("test error")) - }, - "HooksError": func() { - provider.On("API").Return(&api, nil) - provider.On("Supervisor").Return(&supervisor, nil) - - supervisor.On("Start").Return(nil) - supervisor.On("Stop").Return(nil) - supervisor.On("Hooks").Return(&hooks) - - hooks.On("OnActivate", &api).Return(fmt.Errorf("test error")) + supervisor.On("Start", &api).Return(fmt.Errorf("test error")) }, } { t.Run(name, func(t *testing.T) { @@ -291,11 +279,10 @@ func TestEnvironment_ShutdownError(t *testing.T) { provider.On("API").Return(&api, nil) provider.On("Supervisor").Return(&supervisor, nil) - supervisor.On("Start").Return(nil) + supervisor.On("Start", &api).Return(nil) supervisor.On("Stop").Return(fmt.Errorf("test error")) supervisor.On("Hooks").Return(&hooks) - hooks.On("OnActivate", &api).Return(nil) hooks.On("OnDeactivate").Return(fmt.Errorf("test error")) assert.NoError(t, env.ActivatePlugin("foo")) @@ -329,13 +316,12 @@ func TestEnvironment_ConcurrentHookInvocations(t *testing.T) { provider.On("API").Return(&api, nil) provider.On("Supervisor").Return(&supervisor, nil) - supervisor.On("Start").Return(nil) + supervisor.On("Start", &api).Return(nil) supervisor.On("Stop").Return(nil) supervisor.On("Hooks").Return(&hooks) ch := make(chan bool) - hooks.On("OnActivate", &api).Return(nil) hooks.On("OnDeactivate").Return(nil) hooks.On("ServeHTTP", mock.AnythingOfType("*httptest.ResponseRecorder"), mock.AnythingOfType("*http.Request")).Run(func(args mock.Arguments) { r := args.Get(1).(*http.Request) |