diff options
Diffstat (limited to 'plugin/plugintest/hooks.go')
| -rw-r--r-- | plugin/plugintest/hooks.go | 17 |
1 files changed, 16 insertions, 1 deletions
diff --git a/plugin/plugintest/hooks.go b/plugin/plugintest/hooks.go index 56d048d6a..9ea11a9fb 100644 --- a/plugin/plugintest/hooks.go +++ b/plugin/plugintest/hooks.go @@ -8,6 +8,7 @@ import ( "github.com/stretchr/testify/mock" + "github.com/mattermost/mattermost-server/model" "github.com/mattermost/mattermost-server/plugin" ) @@ -18,7 +19,11 @@ type Hooks struct { var _ plugin.Hooks = (*Hooks)(nil) func (m *Hooks) OnActivate(api plugin.API) error { - return m.Called(api).Error(0) + ret := m.Called(api) + if f, ok := ret.Get(0).(func(plugin.API) error); ok { + return f(api) + } + return ret.Error(0) } func (m *Hooks) OnDeactivate() error { @@ -32,3 +37,13 @@ func (m *Hooks) OnConfigurationChange() error { func (m *Hooks) ServeHTTP(w http.ResponseWriter, r *http.Request) { m.Called(w, r) } + +func (m *Hooks) ExecuteCommand(args *model.CommandArgs) (*model.CommandResponse, *model.AppError) { + ret := m.Called(args) + if f, ok := ret.Get(0).(func(*model.CommandArgs) (*model.CommandResponse, *model.AppError)); ok { + return f(args) + } + resp, _ := ret.Get(0).(*model.CommandResponse) + err, _ := ret.Get(1).(*model.AppError) + return resp, err +} |