func (s *contextSuite) TestContextComponentMissing(c *gc.C) {
hctx, _ := s.NewHookContext()
_, err := context.ContextComponent(hctx)
c.Check(err, gc.ErrorMatches, fmt.Sprintf("component %q not registered", payload.ComponentName))
s.Stub.CheckCallNames(c, "Component")
}
func (s *contextSuite) TestContextComponentDisabled(c *gc.C) {
hctx, info := s.NewHookContext()
info.SetComponent(payload.ComponentName, nil)
_, err := context.ContextComponent(hctx)
c.Check(err, gc.ErrorMatches, fmt.Sprintf("component %q disabled", payload.ComponentName))
s.Stub.CheckCallNames(c, "Component")
}
func (s *contextSuite) TestContextComponentWrong(c *gc.C) {
hctx, info := s.NewHookContext()
compCtx := &jujuctesting.ContextComponent{}
info.SetComponent(payload.ComponentName, compCtx)
_, err := context.ContextComponent(hctx)
c.Check(err, gc.ErrorMatches, "wrong component context type registered: .*")
s.Stub.CheckCallNames(c, "Component")
}
func (s *contextSuite) TestContextComponentOkay(c *gc.C) {
hctx, info := s.NewHookContext()
expected := context.NewContext(s.apiClient, s.dataDir)
info.SetComponent(payload.ComponentName, expected)
compCtx, err := context.ContextComponent(hctx)
c.Assert(err, jc.ErrorIsNil)
c.Check(compCtx, gc.Equals, expected)
s.Stub.CheckCallNames(c, "Component")
s.Stub.CheckCall(c, 0, "Component", payload.ComponentName)
}
