← Back to branch summary

~nskaggs/+junk/xenial-test

« back to all changes in this revision

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

  • Committer: Nicholas Skaggs
  • Date: 2016-10-24 20:56:05 UTC
  • Revision ID: nicholas.skaggs@canonical.com-20161024205605-z8lta0uvuhtxwzwl
Initi with beta15

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/github.com/juju/juju/state/status_machine_test.go
 
1
// Copyright 2012-2015 Canonical Ltd.
 
2
// Licensed under the AGPLv3, see LICENCE file for details.
 
3
 
 
4
package state_test
 
5
 
 
6
import (
 
7
        "time"
 
8
 
 
9
        jc "github.com/juju/testing/checkers"
 
10
        gc "gopkg.in/check.v1"
 
11
 
 
12
        "github.com/juju/juju/state"
 
13
        "github.com/juju/juju/status"
 
14
)
 
15
 
 
16
type MachineStatusSuite struct {
 
17
        ConnSuite
 
18
        machine *state.Machine
 
19
}
 
20
 
 
21
var _ = gc.Suite(&MachineStatusSuite{})
 
22
 
 
23
func (s *MachineStatusSuite) SetUpTest(c *gc.C) {
 
24
        s.ConnSuite.SetUpTest(c)
 
25
        s.machine = s.Factory.MakeMachine(c, nil)
 
26
}
 
27
 
 
28
func (s *MachineStatusSuite) TestInitialStatus(c *gc.C) {
 
29
        s.checkInitialStatus(c)
 
30
}
 
31
 
 
32
func (s *MachineStatusSuite) checkInitialStatus(c *gc.C) {
 
33
        statusInfo, err := s.machine.Status()
 
34
        c.Check(err, jc.ErrorIsNil)
 
35
        c.Check(statusInfo.Status, gc.Equals, status.StatusPending)
 
36
        c.Check(statusInfo.Message, gc.Equals, "")
 
37
        c.Check(statusInfo.Data, gc.HasLen, 0)
 
38
        c.Check(statusInfo.Since, gc.NotNil)
 
39
}
 
40
 
 
41
func (s *MachineStatusSuite) TestSetErrorStatusWithoutInfo(c *gc.C) {
 
42
        now := time.Now()
 
43
        sInfo := status.StatusInfo{
 
44
                Status:  status.StatusError,
 
45
                Message: "",
 
46
                Since:   &now,
 
47
        }
 
48
        err := s.machine.SetStatus(sInfo)
 
49
        c.Check(err, gc.ErrorMatches, `cannot set status "error" without info`)
 
50
 
 
51
        s.checkInitialStatus(c)
 
52
}
 
53
 
 
54
func (s *MachineStatusSuite) TestSetDownStatus(c *gc.C) {
 
55
        now := time.Now()
 
56
        sInfo := status.StatusInfo{
 
57
                Status:  status.StatusDown,
 
58
                Message: "",
 
59
                Since:   &now,
 
60
        }
 
61
        err := s.machine.SetStatus(sInfo)
 
62
        c.Check(err, gc.ErrorMatches, `cannot set status "down"`)
 
63
 
 
64
        s.checkInitialStatus(c)
 
65
}
 
66
 
 
67
func (s *MachineStatusSuite) TestSetUnknownStatus(c *gc.C) {
 
68
        now := time.Now()
 
69
        sInfo := status.StatusInfo{
 
70
                Status:  status.Status("vliegkat"),
 
71
                Message: "orville",
 
72
                Since:   &now,
 
73
        }
 
74
        err := s.machine.SetStatus(sInfo)
 
75
        c.Assert(err, gc.ErrorMatches, `cannot set invalid status "vliegkat"`)
 
76
 
 
77
        s.checkInitialStatus(c)
 
78
}
 
79
 
 
80
func (s *MachineStatusSuite) TestSetOverwritesData(c *gc.C) {
 
81
        now := time.Now()
 
82
        sInfo := status.StatusInfo{
 
83
                Status:  status.StatusStarted,
 
84
                Message: "blah",
 
85
                Data: map[string]interface{}{
 
86
                        "pew.pew": "zap",
 
87
                },
 
88
                Since: &now,
 
89
        }
 
90
        err := s.machine.SetStatus(sInfo)
 
91
        c.Check(err, jc.ErrorIsNil)
 
92
 
 
93
        s.checkGetSetStatus(c)
 
94
}
 
95
 
 
96
func (s *MachineStatusSuite) TestGetSetStatusAlive(c *gc.C) {
 
97
        s.checkGetSetStatus(c)
 
98
}
 
99
 
 
100
func (s *MachineStatusSuite) checkGetSetStatus(c *gc.C) {
 
101
        now := time.Now()
 
102
        sInfo := status.StatusInfo{
 
103
                Status:  status.StatusStarted,
 
104
                Message: "blah",
 
105
                Data: map[string]interface{}{
 
106
                        "$foo.bar.baz": map[string]interface{}{
 
107
                                "pew.pew": "zap",
 
108
                        },
 
109
                },
 
110
                Since: &now,
 
111
        }
 
112
        err := s.machine.SetStatus(sInfo)
 
113
        c.Check(err, jc.ErrorIsNil)
 
114
 
 
115
        machine, err := s.State.Machine(s.machine.Id())
 
116
        c.Assert(err, jc.ErrorIsNil)
 
117
 
 
118
        statusInfo, err := machine.Status()
 
119
        c.Check(err, jc.ErrorIsNil)
 
120
        c.Check(statusInfo.Status, gc.Equals, status.StatusStarted)
 
121
        c.Check(statusInfo.Message, gc.Equals, "blah")
 
122
        c.Check(statusInfo.Data, jc.DeepEquals, map[string]interface{}{
 
123
                "$foo.bar.baz": map[string]interface{}{
 
124
                        "pew.pew": "zap",
 
125
                },
 
126
        })
 
127
        c.Check(statusInfo.Since, gc.NotNil)
 
128
}
 
129
 
 
130
func (s *MachineStatusSuite) TestGetSetStatusDying(c *gc.C) {
 
131
        err := s.machine.Destroy()
 
132
        c.Assert(err, jc.ErrorIsNil)
 
133
 
 
134
        s.checkGetSetStatus(c)
 
135
}
 
136
 
 
137
func (s *MachineStatusSuite) TestGetSetStatusDead(c *gc.C) {
 
138
        err := s.machine.EnsureDead()
 
139
        c.Assert(err, jc.ErrorIsNil)
 
140
 
 
141
        // NOTE: it would be more technically correct to reject status updates
 
142
        // while Dead, but it's easier and clearer, not to mention more efficient,
 
143
        // to just depend on status doc existence.
 
144
        s.checkGetSetStatus(c)
 
145
}
 
146
 
 
147
func (s *MachineStatusSuite) TestGetSetStatusGone(c *gc.C) {
 
148
        err := s.machine.EnsureDead()
 
149
        c.Assert(err, jc.ErrorIsNil)
 
150
        err = s.machine.Remove()
 
151
        c.Assert(err, jc.ErrorIsNil)
 
152
 
 
153
        now := time.Now()
 
154
        sInfo := status.StatusInfo{
 
155
                Status:  status.StatusStarted,
 
156
                Message: "not really",
 
157
                Since:   &now,
 
158
        }
 
159
        err = s.machine.SetStatus(sInfo)
 
160
        c.Check(err, gc.ErrorMatches, `cannot set status: machine not found`)
 
161
 
 
162
        statusInfo, err := s.machine.Status()
 
163
        c.Check(err, gc.ErrorMatches, `cannot get status: machine not found`)
 
164
        c.Check(statusInfo, gc.DeepEquals, status.StatusInfo{})
 
165
}
 
166
 
 
167
func (s *MachineStatusSuite) TestSetStatusPendingProvisioned(c *gc.C) {
 
168
        now := time.Now()
 
169
        sInfo := status.StatusInfo{
 
170
                Status:  status.StatusPending,
 
171
                Message: "",
 
172
                Since:   &now,
 
173
        }
 
174
        err := s.machine.SetStatus(sInfo)
 
175
        c.Check(err, gc.ErrorMatches, `cannot set status "pending"`)
 
176
}
 
177
 
 
178
func (s *MachineStatusSuite) TestSetStatusPendingUnprovisioned(c *gc.C) {
 
179
        machine, err := s.State.AddMachine("quantal", state.JobHostUnits)
 
180
        c.Assert(err, jc.ErrorIsNil)
 
181
        now := time.Now()
 
182
        sInfo := status.StatusInfo{
 
183
                Status:  status.StatusPending,
 
184
                Message: "",
 
185
                Since:   &now,
 
186
        }
 
187
        err = machine.SetStatus(sInfo)
 
188
        c.Check(err, jc.ErrorIsNil)
 
189
}