func (payloads) registerHookContextCommands() {
if !markRegistered(payload.ComponentName, "hook-context-commands") {
return
}
jujuc.RegisterCommand(context.RegisterCmdName, func(ctx jujuc.Context) (cmd.Command, error) {
compCtx := payloadsHookContext{ctx}
cmd, err := context.NewRegisterCmd(compCtx)
if err != nil {
return nil, errors.Trace(err)
}
return cmd, nil
})
jujuc.RegisterCommand(context.StatusSetCmdName, func(ctx jujuc.Context) (cmd.Command, error) {
compCtx := payloadsHookContext{ctx}
cmd, err := context.NewStatusSetCmd(compCtx)
if err != nil {
return nil, errors.Trace(err)
}
return cmd, nil
})
jujuc.RegisterCommand(context.UnregisterCmdName, func(ctx jujuc.Context) (cmd.Command, error) {
compCtx := payloadsHookContext{ctx}
cmd, err := context.NewUnregisterCmd(compCtx)
if err != nil {
return nil, errors.Trace(err)
}
return cmd, nil
})
}
func (s *statusSetSuite) SetUpTest(c *gc.C) {
s.IsolationSuite.SetUpTest(c)
s.stub = &testing.Stub{}
s.compCtx = &stubSetStatusContext{stub: s.stub}
s.ctx = coretesting.Context(c)
cmd, err := context.NewStatusSetCmd(s)
c.Assert(err, jc.ErrorIsNil)
s.cmd = cmd
}