func (s *UnitAgentSuite) TestGetUnitAgentStatusHistory(c *gc.C) {
err := state.EraseUnitHistory(s.unit)
c.Assert(err, jc.ErrorIsNil)
agent := s.unit.Agent().(*state.UnitAgent)
globalKey := state.UnitAgentGlobalKey(agent)
history := func(i int) ([]state.StatusInfo, error) {
return agent.StatusHistory(i)
}
testGetUnitStatusHistory(c, history, s.State, globalKey)
}
func (s *UnitAgentSuite) TestSetUnitAgentStatusHistory(c *gc.C) {
err := state.EraseUnitHistory(s.unit)
c.Assert(err, jc.ErrorIsNil)
agent := s.unit.Agent().(*state.UnitAgent)
globalKey := state.UnitAgentGlobalKey(agent)
err = agent.SetStatus(state.StatusIdle, "to push something to history", nil)
c.Assert(err, jc.ErrorIsNil)
statusInfo, err := agent.Status()
c.Assert(err, jc.ErrorIsNil)
c.Assert(statusInfo.Status, gc.Equals, state.StatusIdle)
h, err := state.StatusHistory(10, globalKey, s.State)
c.Assert(err, jc.ErrorIsNil)
c.Assert(h, gc.HasLen, 1)
c.Assert(h[0].Status, gc.Equals, state.StatusAllocating)
c.Assert(h[0].Message, gc.Equals, "")
err = agent.SetStatus(state.StatusExecuting, "executing first thing", nil)
c.Assert(err, jc.ErrorIsNil)
statusInfo, err = agent.Status()
c.Assert(err, jc.ErrorIsNil)
c.Assert(statusInfo.Status, gc.Equals, state.StatusExecuting)
err = agent.SetStatus(state.StatusExecuting, "wow executing again", nil)
c.Assert(err, jc.ErrorIsNil)
statusInfo, err = agent.Status()
c.Assert(err, jc.ErrorIsNil)
c.Assert(statusInfo.Status, gc.Equals, state.StatusExecuting)
h, err = state.StatusHistory(10, globalKey, s.State)
c.Assert(err, jc.ErrorIsNil)
c.Assert(h, gc.HasLen, 3)
var message string
for i := 0; i < 3; i++ {
c.Logf("checking status %q", h[i].Status)
switch h[i].Status {
case state.StatusAllocating:
message = ""
case state.StatusExecuting:
message = "executing first thing"
case state.StatusIdle:
message = "to push something to history"
}
c.Assert(h[i].Message, gc.Equals, message)
}
}