// getBlockAPI returns a block api for listing blocks.
func getBlockAPI(c *envcmd.EnvCommandBase) (block.BlockListAPI, error) {
root, err := c.NewAPIRoot()
if err != nil {
return nil, err
}
return apiblock.NewClient(root), nil
}
// getBlockAPI returns a block api for block manipulation.
func getBlockAPI(c newAPIRoot) (*apiblock.Client, error) {
root, err := c.NewAPIRoot()
if err != nil {
return nil, err
}
return apiblock.NewClient(root), nil
}
func (s *blockMockSuite) TestSwitchBlockOff(c *gc.C) {
called := false
blockType := state.DestroyBlock.String()
apiCaller := basetesting.APICallerFunc(
func(objType string,
version int,
id, request string,
a, response interface{},
) error {
called = true
c.Check(objType, gc.Equals, "Block")
c.Check(id, gc.Equals, "")
c.Check(request, gc.Equals, "SwitchBlockOff")
args, ok := a.(params.BlockSwitchParams)
c.Assert(ok, jc.IsTrue)
// message is never sent, so this argument should
// always be empty string.
c.Assert(args.Message, gc.DeepEquals, "")
c.Assert(args.Type, gc.DeepEquals, blockType)
_, ok = response.(*params.ErrorResult)
c.Assert(ok, jc.IsTrue)
return nil
})
blockClient := block.NewClient(apiCaller)
err := blockClient.SwitchBlockOff(blockType)
c.Assert(called, jc.IsTrue)
c.Assert(err, gc.IsNil)
}
// getBlockAPI returns a block api for listing blocks.
func getBlockAPI(c *modelcmd.ModelCommandBase) (listBlocksAPI, error) {
root, err := c.NewAPIRoot()
if err != nil {
return nil, errors.Trace(err)
}
return block.NewClient(root), nil
}
func (s *blockMockSuite) TestSwitchBlockOn(c *gc.C) {
called := false
blockType := state.DestroyBlock.String()
msg := "for test switch block on"
apiCaller := basetesting.APICallerFunc(
func(objType string,
version int,
id, request string,
a, response interface{},
) error {
called = true
c.Check(objType, gc.Equals, "Block")
c.Check(id, gc.Equals, "")
c.Check(request, gc.Equals, "SwitchBlockOn")
args, ok := a.(params.BlockSwitchParams)
c.Assert(ok, jc.IsTrue)
c.Assert(args.Message, gc.DeepEquals, msg)
c.Assert(args.Type, gc.DeepEquals, blockType)
_, ok = response.(*params.ErrorResult)
c.Assert(ok, jc.IsTrue)
return nil
})
blockClient := block.NewClient(apiCaller)
err := blockClient.SwitchBlockOn(blockType, msg)
c.Assert(called, jc.IsTrue)
c.Assert(err, gc.IsNil)
}
func (s *blockSuite) SetUpTest(c *gc.C) {
s.JujuConnSuite.SetUpTest(c)
s.blockClient = block.NewClient(s.APIState)
c.Assert(s.blockClient, gc.NotNil)
s.AddCleanup(func(*gc.C) {
s.blockClient.ClientFacade.Close()
})
}
func (s *blockSuite) SetUpTest(c *gc.C) {
s.JujuConnSuite.SetUpTest(c)
conn, err := juju.NewAPIState(s.AdminUserTag(c), s.Environ, api.DialOpts{})
c.Assert(err, jc.ErrorIsNil)
s.AddCleanup(func(*gc.C) { conn.Close() })
s.ctx = &mockContext{
agentConfig: &mockAgentConfig{dataDir: s.DataDir()},
apiState: conn,
state: s.State,
}
s.blockClient = block.NewClient(conn)
}
func (s *blockMockSuite) TestSwitchBlockOnError(c *gc.C) {
called := false
errmsg := "test error"
apiCaller := basetesting.APICallerFunc(
func(objType string,
version int,
id, request string,
a, response interface{},
) error {
called = true
result, ok := response.(*params.ErrorResult)
c.Assert(ok, jc.IsTrue)
result.Error = common.ServerError(errors.New(errmsg))
return nil
})
blockClient := block.NewClient(apiCaller)
err := blockClient.SwitchBlockOn("", "")
c.Assert(called, jc.IsTrue)
c.Assert(errors.Cause(err), gc.ErrorMatches, errmsg)
}
func (s *blockMockSuite) TestList(c *gc.C) {
var called bool
one := params.BlockResult{
Result: params.Block{
Id: "-42",
Type: state.DestroyBlock.String(),
Message: "for test switch on",
Tag: "some valid tag, right?",
},
}
errmsg := "another test error"
two := params.BlockResult{
Error: common.ServerError(errors.New(errmsg)),
}
apiCaller := basetesting.APICallerFunc(
func(
objType string,
version int,
id, request string,
a, response interface{}) error {
called = true
c.Check(objType, gc.Equals, "Block")
c.Check(id, gc.Equals, "")
c.Check(request, gc.Equals, "List")
c.Assert(a, gc.IsNil)
result := response.(*params.BlockResults)
result.Results = []params.BlockResult{one, two}
return nil
})
blockClient := block.NewClient(apiCaller)
found, err := blockClient.List()
c.Assert(called, jc.IsTrue)
c.Assert(errors.Cause(err), gc.ErrorMatches, errmsg)
c.Assert(found, gc.HasLen, 1)
}
// NewBlockHelper creates a block switch used in testing
// to manage desired juju blocks.
func NewBlockHelper(st api.Connection) BlockHelper {
return BlockHelper{
blockClient: block.NewClient(st),
}
}
// NewBlockHelper creates a block switch used in testing
// to manage desired juju blocks.
func NewBlockHelper(st *api.State) BlockHelper {
return BlockHelper{
blockClient: block.NewClient(st),
}
}
// NewBlockHelper creates a block switch used in testing
// to manage desired juju blocks.
func NewBlockHelper(st api.Connection) BlockHelper {
return BlockHelper{
ApiState: st,
client: block.NewClient(st),
}
}
func (s *blockSuite) SetUpTest(c *gc.C) {
s.JujuConnSuite.SetUpTest(c)
s.blockClient = block.NewClient(s.APIState)
c.Assert(s.blockClient, gc.NotNil)
}
// NewBlockHelper creates a block switch used in testing
// to manage desired juju blocks.
func NewBlockHelper(st *api.State) BlockHelper {
return BlockHelper{
ApiState: st,
client: block.NewClient(st),
}
}
// NewCmdBlockHelper creates a block switch used in testing
// to manage desired juju blocks.
func NewCmdBlockHelper(api base.APICallCloser) CmdBlockHelper {
return CmdBlockHelper{
blockClient: block.NewClient(api),
}
}