func (s *WireFormatSuite) TestSetStatus(c *gc.C) {
resp := wireformat.EnvironmentResponses{}
c.Assert(resp, gc.HasLen, 0)
modelUUID := "model-uuid"
modelUUID2 := "model-uuid2"
unitName := "some-unit/0"
unitName2 := "some-unit/1"
resp.SetStatus(modelUUID, unitName, "GREEN", "")
c.Assert(resp, gc.HasLen, 1)
c.Assert(resp[modelUUID].UnitStatuses[unitName].Status, gc.Equals, "GREEN")
c.Assert(resp[modelUUID].UnitStatuses[unitName].Info, gc.Equals, "")
resp.SetStatus(modelUUID, unitName2, "RED", "Unit unresponsive.")
c.Assert(resp, gc.HasLen, 1)
c.Assert(resp[modelUUID].UnitStatuses[unitName].Status, gc.Equals, "GREEN")
c.Assert(resp[modelUUID].UnitStatuses[unitName].Info, gc.Equals, "")
c.Assert(resp[modelUUID].UnitStatuses[unitName2].Status, gc.Equals, "RED")
c.Assert(resp[modelUUID].UnitStatuses[unitName2].Info, gc.Equals, "Unit unresponsive.")
resp.SetStatus(modelUUID2, unitName, "UNKNOWN", "")
c.Assert(resp, gc.HasLen, 2)
c.Assert(resp[modelUUID2].UnitStatuses[unitName].Status, gc.Equals, "UNKNOWN")
c.Assert(resp[modelUUID2].UnitStatuses[unitName].Info, gc.Equals, "")
resp.SetStatus(modelUUID, unitName, "RED", "Invalid data received.")
c.Assert(resp, gc.HasLen, 2)
c.Assert(resp[modelUUID].UnitStatuses[unitName].Status, gc.Equals, "RED")
c.Assert(resp[modelUUID].UnitStatuses[unitName].Info, gc.Equals, "Invalid data received.")
}
