8
8
gc "gopkg.in/check.v1"
10
10
"github.com/juju/juju/state"
11
"github.com/juju/juju/status"
13
type statusHistoryFunc func(int) ([]state.StatusInfo, error)
14
type statusHistoryFunc func(int) ([]status.StatusInfo, error)
15
func checkInitialWorkloadStatus(c *gc.C, statusInfo state.StatusInfo) {
16
c.Check(statusInfo.Status, gc.Equals, state.StatusUnknown)
16
func checkInitialWorkloadStatus(c *gc.C, statusInfo status.StatusInfo) {
17
c.Check(statusInfo.Status, gc.Equals, status.StatusUnknown)
17
18
c.Check(statusInfo.Message, gc.Equals, "Waiting for agent initialization to finish")
18
19
c.Check(statusInfo.Data, gc.HasLen, 0)
19
20
c.Check(statusInfo.Since, gc.NotNil)
22
23
func primeUnitStatusHistory(c *gc.C, unit *state.Unit, count int) {
23
24
for i := 0; i < count; i++ {
24
err := unit.SetStatus(state.StatusActive, "", map[string]interface{}{"$foo": i})
25
err := unit.SetStatus(status.StatusActive, "", map[string]interface{}{"$foo": i})
25
26
c.Assert(err, gc.IsNil)
29
func checkPrimedUnitStatus(c *gc.C, statusInfo state.StatusInfo, expect int) {
30
c.Check(statusInfo.Status, gc.Equals, state.StatusActive)
30
func checkPrimedUnitStatus(c *gc.C, statusInfo status.StatusInfo, expect int) {
31
c.Check(statusInfo.Status, gc.Equals, status.StatusActive)
31
32
c.Check(statusInfo.Message, gc.Equals, "")
32
33
c.Check(statusInfo.Data, jc.DeepEquals, map[string]interface{}{"$foo": expect})
33
34
c.Check(statusInfo.Since, gc.NotNil)
36
func checkInitialUnitAgentStatus(c *gc.C, statusInfo state.StatusInfo) {
37
c.Check(statusInfo.Status, gc.Equals, state.StatusAllocating)
37
func checkInitialUnitAgentStatus(c *gc.C, statusInfo status.StatusInfo) {
38
c.Check(statusInfo.Status, gc.Equals, status.StatusAllocating)
38
39
c.Check(statusInfo.Message, gc.Equals, "")
39
40
c.Check(statusInfo.Data, gc.HasLen, 0)
40
41
c.Assert(statusInfo.Since, gc.NotNil)
43
44
func primeUnitAgentStatusHistory(c *gc.C, agent *state.UnitAgent, count int) {
44
45
for i := 0; i < count; i++ {
45
err := agent.SetStatus(state.StatusExecuting, "", map[string]interface{}{"$bar": i})
46
err := agent.SetStatus(status.StatusExecuting, "", map[string]interface{}{"$bar": i})
46
47
c.Assert(err, gc.IsNil)
50
func checkPrimedUnitAgentStatus(c *gc.C, statusInfo state.StatusInfo, expect int) {
51
c.Check(statusInfo.Status, gc.Equals, state.StatusExecuting)
51
func checkPrimedUnitAgentStatus(c *gc.C, statusInfo status.StatusInfo, expect int) {
52
c.Check(statusInfo.Status, gc.Equals, status.StatusExecuting)
52
53
c.Check(statusInfo.Message, gc.Equals, "")
53
54
c.Check(statusInfo.Data, jc.DeepEquals, map[string]interface{}{"$bar": expect})
54
55
c.Check(statusInfo.Since, gc.NotNil)