// NewDeployerAPI creates a new client-side DeployerAPI facade.
func NewDeployerAPI(
st *state.State,
resources *common.Resources,
authorizer common.Authorizer,
) (*DeployerAPI, error) {
if !authorizer.AuthMachineAgent() {
return nil, common.ErrPerm
}
getAuthFunc := func() (common.AuthFunc, error) {
// Get all units of the machine and cache them.
thisMachineTag := authorizer.GetAuthTag()
units, err := getAllUnits(st, thisMachineTag)
if err != nil {
return nil, err
}
// Then we just check if the unit is already known.
return func(tag string) bool {
for _, unit := range units {
if names.UnitTag(unit) == tag {
return true
}
}
return false
}, nil
}
return &DeployerAPI{
Remover: common.NewRemover(st, getAuthFunc),
PasswordChanger: common.NewPasswordChanger(st, getAuthFunc),
LifeGetter: common.NewLifeGetter(st, getAuthFunc),
st: st,
resources: resources,
authorizer: authorizer,
}, nil
}
// NewDeployerAPI creates a new client-side DeployerAPI facade.
func NewDeployerAPI(
st *state.State,
resources *common.Resources,
authorizer common.Authorizer,
) (*DeployerAPI, error) {
if !authorizer.AuthMachineAgent() {
return nil, common.ErrPerm
}
getAuthFunc := func() (common.AuthFunc, error) {
// Get all units of the machine and cache them.
knownUnits := set.NewStrings()
thisMachineTag := authorizer.GetAuthTag()
if units, err := getAllUnits(st, thisMachineTag); err != nil {
return nil, err
} else {
for _, unit := range units {
knownUnits.Add(unit)
}
}
// Then we just check if the unit is already known.
return func(tag string) bool {
unitName := state.UnitNameFromTag(tag)
return knownUnits.Contains(unitName)
}, nil
}
return &DeployerAPI{
Remover: common.NewRemover(st, getAuthFunc),
PasswordChanger: common.NewPasswordChanger(st, getAuthFunc),
LifeGetter: common.NewLifeGetter(st, getAuthFunc),
st: st,
resources: resources,
authorizer: authorizer,
}, nil
}
func (*removeSuite) TestRemoveError(c *gc.C) {
getCanModify := func() (common.AuthFunc, error) {
return nil, fmt.Errorf("pow")
}
r := common.NewRemover(&fakeState{}, getCanModify)
_, err := r.Remove(params.Entities{[]params.Entity{{"x0"}}})
c.Assert(err, gc.ErrorMatches, "pow")
}
func (*removeSuite) TestRemoveNoArgsNoError(c *gc.C) {
getCanModify := func() (common.AuthFunc, error) {
return nil, fmt.Errorf("pow")
}
r := common.NewRemover(&fakeState{}, getCanModify)
result, err := r.Remove(params.Entities{})
c.Assert(err, gc.IsNil)
c.Assert(result.Results, gc.HasLen, 0)
}
func (*removeSuite) TestRemove(c *gc.C) {
st := &fakeState{
entities: map[string]entityWithError{
"x0": &fakeRemover{life: state.Dying, errEnsureDead: fmt.Errorf("x0 EnsureDead fails")},
"x1": &fakeRemover{life: state.Dying, errRemove: fmt.Errorf("x1 Remove fails")},
"x2": &fakeRemover{life: state.Alive},
"x3": &fakeRemover{life: state.Dying},
"x4": &fakeRemover{life: state.Dead},
"x5": &fakeRemover{fetchError: "x5 error"},
},
}
getCanModify := func() (common.AuthFunc, error) {
return func(tag string) bool {
switch tag {
case "x0", "x1", "x2", "x3", "x5":
return true
}
return false
}, nil
}
r := common.NewRemover(st, getCanModify)
entities := params.Entities{[]params.Entity{
{"x0"}, {"x1"}, {"x2"}, {"x3"}, {"x4"}, {"x5"}, {"x6"},
}}
result, err := r.Remove(entities)
c.Assert(err, gc.IsNil)
c.Assert(result, gc.DeepEquals, params.ErrorResults{
Results: []params.ErrorResult{
{¶ms.Error{Message: "x0 EnsureDead fails"}},
{¶ms.Error{Message: "x1 Remove fails"}},
{¶ms.Error{Message: `cannot remove entity "x2": still alive`}},
{nil},
{apiservertesting.ErrUnauthorized},
{¶ms.Error{Message: "x5 error"}},
{apiservertesting.ErrUnauthorized},
},
})
}