func (s *unitPayloadsSuite) TestSetStatusMissing(c *gc.C) {
id := "f47ac10b-58cc-4372-a567-0e02b2c3d479"
ps := state.UnitPayloads{
Persist: s.persist,
Unit: "a-service/0",
}
err := ps.SetStatus(id, payload.StateRunning)
c.Check(err, jc.Satisfies, errors.IsNotFound)
}
func (s *unitPayloadsSuite) TestSetStatusFailed(c *gc.C) {
id := "f47ac10b-58cc-4372-a567-0e02b2c3d479"
failure := errors.Errorf("<failed!>")
s.stub.SetErrors(failure)
pl := s.newPayload("docker", "payloadA/payloadA-xyz")
s.persist.setPayload(id, &pl)
ps := state.UnitPayloads{
Persist: s.persist,
Unit: "a-service/0",
}
err := ps.SetStatus(id, payload.StateRunning)
c.Check(errors.Cause(err), gc.Equals, failure)
}
func (s *unitPayloadsSuite) TestSetStatusOkay(c *gc.C) {
id := "f47ac10b-58cc-4372-a567-0e02b2c3d479"
pl := s.newPayload("docker", "payloadA/payloadA-xyz")
s.persist.setPayload(id, &pl)
ps := state.UnitPayloads{
Persist: s.persist,
Unit: "a-service/0",
}
err := ps.SetStatus(id, payload.StateRunning)
c.Assert(err, jc.ErrorIsNil)
s.stub.CheckCallNames(c, "SetStatus")
current := s.persist.payloads[id]
c.Check(current.Status, jc.DeepEquals, payload.StateRunning)
}