func (s *contextSuite) TestNewContextAPICalls(c *gc.C) {
s.apiClient.setNew("A/xyz123")
_, err := context.NewContextAPI(s.apiClient, s.dataDir)
c.Assert(err, jc.ErrorIsNil)
s.Stub.CheckCallNames(c, "List")
}
func (s *contextSuite) TestNewContextAPIEmpty(c *gc.C) {
ctx, err := context.NewContextAPI(s.apiClient, s.dataDir)
c.Assert(err, jc.ErrorIsNil)
payloads, err := ctx.Payloads()
c.Assert(err, jc.ErrorIsNil)
c.Check(payloads, gc.HasLen, 0)
}
func (s *contextSuite) TestNewContextAPIError(c *gc.C) {
expected := errors.Errorf("<failed>")
s.Stub.SetErrors(expected)
_, err := context.NewContextAPI(s.apiClient, s.dataDir)
c.Check(errors.Cause(err), gc.Equals, expected)
s.Stub.CheckCallNames(c, "List")
}
func (s *contextSuite) TestNewContextAPIOkay(c *gc.C) {
expected := s.apiClient.setNew("A/xyx123")
ctx, err := context.NewContextAPI(s.apiClient, s.dataDir)
c.Assert(err, jc.ErrorIsNil)
payloads, err := ctx.Payloads()
c.Assert(err, jc.ErrorIsNil)
c.Check(payloads, jc.DeepEquals, expected)
}
func (c payloads) registerHookContext() {
if !markRegistered(payload.ComponentName, "hook-context") {
return
}
unitercontext.RegisterComponentFunc(payload.ComponentName,
func(config unitercontext.ComponentConfig) (jujuc.ContextComponent, error) {
hctxClient := c.newUnitFacadeClient(config.APICaller)
// TODO(ericsnow) Pass the unit's tag through to the component?
component, err := context.NewContextAPI(hctxClient, config.DataDir)
if err != nil {
return nil, errors.Trace(err)
}
return component, nil
},
)
c.registerHookContextCommands()
c.registerHookContextFacade()
}