~juju-qa/ubuntu/xenial/juju/xenial-2.0-beta3

« back to all changes in this revision

Viewing changes to src/github.com/juju/juju/state/status_util_test.go

  • Committer: Martin Packman
  • Date: 2016-03-30 19:31:08 UTC
  • mfrom: (1.1.41)
  • Revision ID: martin.packman@canonical.com-20160330193108-h9iz3ak334uk0z5r
Merge new upstream source 2.0~beta3

Show diffs side-by-side

added added

removed removed

Lines of Context:
8
8
        gc "gopkg.in/check.v1"
9
9
 
10
10
        "github.com/juju/juju/state"
 
11
        "github.com/juju/juju/status"
11
12
)
12
13
 
13
 
type statusHistoryFunc func(int) ([]state.StatusInfo, error)
 
14
type statusHistoryFunc func(int) ([]status.StatusInfo, error)
14
15
 
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)
21
22
 
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)
26
27
        }
27
28
}
28
29
 
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)
34
35
}
35
36
 
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)
42
43
 
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)
47
48
        }
48
49
}
49
50
 
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)