← Back to branch summary

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

« back to all changes in this revision

Viewing changes to src/github.com/juju/juju/apiserver/uniter/uniter_test.go

  • Committer: Nicholas Skaggs
  • Date: 2016-09-30 14:39:30 UTC
  • mfrom: (1.8.1)
  • Revision ID: nicholas.skaggs@canonical.com-20160930143930-vwwhrefh6ftckccy
import upstream

Show diffs side-by-side

added added

removed removed

Lines of Context:
src/github.com/juju/juju/apiserver/uniter/uniter_test.go
134
134
func (s *uniterSuite) TestSetStatus(c *gc.C) {
135
135
        now := time.Now()
136
136
        sInfo := status.StatusInfo{
137
 
                Status:  status.StatusExecuting,
 
137
                Status:  status.Executing,
138
138
                Message: "blah",
139
139
                Since:   &now,
140
140
        }
141
141
        err := s.wordpressUnit.SetAgentStatus(sInfo)
142
142
        c.Assert(err, jc.ErrorIsNil)
143
143
        sInfo = status.StatusInfo{
144
 
                Status:  status.StatusExecuting,
 
144
                Status:  status.Executing,
145
145
                Message: "foo",
146
146
                Since:   &now,
147
147
        }
150
150
 
151
151
        args := params.SetStatus{
152
152
                Entities: []params.EntityStatusArgs{
153
 
                        {Tag: "unit-mysql-0", Status: status.StatusError.String(), Info: "not really"},
154
 
                        {Tag: "unit-wordpress-0", Status: status.StatusRebooting.String(), Info: "foobar"},
155
 
                        {Tag: "unit-foo-42", Status: status.StatusActive.String(), Info: "blah"},
 
153
                        {Tag: "unit-mysql-0", Status: status.Error.String(), Info: "not really"},
 
154
                        {Tag: "unit-wordpress-0", Status: status.Rebooting.String(), Info: "foobar"},
 
155
                        {Tag: "unit-foo-42", Status: status.Active.String(), Info: "blah"},
156
156
                }}
157
157
        result, err := s.uniter.SetStatus(args)
158
158
        c.Assert(err, jc.ErrorIsNil)
167
167
        // Verify mysqlUnit - no change.
168
168
        statusInfo, err := s.mysqlUnit.AgentStatus()
169
169
        c.Assert(err, jc.ErrorIsNil)
170
 
        c.Assert(statusInfo.Status, gc.Equals, status.StatusExecuting)
 
170
        c.Assert(statusInfo.Status, gc.Equals, status.Executing)
171
171
        c.Assert(statusInfo.Message, gc.Equals, "foo")
172
172
        // ...wordpressUnit is fine though.
173
173
        statusInfo, err = s.wordpressUnit.AgentStatus()
174
174
        c.Assert(err, jc.ErrorIsNil)
175
 
        c.Assert(statusInfo.Status, gc.Equals, status.StatusRebooting)
 
175
        c.Assert(statusInfo.Status, gc.Equals, status.Rebooting)
176
176
        c.Assert(statusInfo.Message, gc.Equals, "foobar")
177
177
}
178
178
 
179
179
func (s *uniterSuite) TestSetAgentStatus(c *gc.C) {
180
180
        now := time.Now()
181
181
        sInfo := status.StatusInfo{
182
 
                Status:  status.StatusExecuting,
 
182
                Status:  status.Executing,
183
183
                Message: "blah",
184
184
                Since:   &now,
185
185
        }
186
186
        err := s.wordpressUnit.SetAgentStatus(sInfo)
187
187
        c.Assert(err, jc.ErrorIsNil)
188
188
        sInfo = status.StatusInfo{
189
 
                Status:  status.StatusExecuting,
 
189
                Status:  status.Executing,
190
190
                Message: "foo",
191
191
                Since:   &now,
192
192
        }
195
195
 
196
196
        args := params.SetStatus{
197
197
                Entities: []params.EntityStatusArgs{
198
 
                        {Tag: "unit-mysql-0", Status: status.StatusError.String(), Info: "not really"},
199
 
                        {Tag: "unit-wordpress-0", Status: status.StatusExecuting.String(), Info: "foobar"},
200
 
                        {Tag: "unit-foo-42", Status: status.StatusRebooting.String(), Info: "blah"},
 
198
                        {Tag: "unit-mysql-0", Status: status.Error.String(), Info: "not really"},
 
199
                        {Tag: "unit-wordpress-0", Status: status.Executing.String(), Info: "foobar"},
 
200
                        {Tag: "unit-foo-42", Status: status.Rebooting.String(), Info: "blah"},
201
201
                }}
202
202
        result, err := s.uniter.SetAgentStatus(args)
203
203
        c.Assert(err, jc.ErrorIsNil)
212
212
        // Verify mysqlUnit - no change.
213
213
        statusInfo, err := s.mysqlUnit.AgentStatus()
214
214
        c.Assert(err, jc.ErrorIsNil)
215
 
        c.Assert(statusInfo.Status, gc.Equals, status.StatusExecuting)
 
215
        c.Assert(statusInfo.Status, gc.Equals, status.Executing)
216
216
        c.Assert(statusInfo.Message, gc.Equals, "foo")
217
217
        // ...wordpressUnit is fine though.
218
218
        statusInfo, err = s.wordpressUnit.AgentStatus()
219
219
        c.Assert(err, jc.ErrorIsNil)
220
 
        c.Assert(statusInfo.Status, gc.Equals, status.StatusExecuting)
 
220
        c.Assert(statusInfo.Status, gc.Equals, status.Executing)
221
221
        c.Assert(statusInfo.Message, gc.Equals, "foobar")
222
222
}
223
223
 
224
224
func (s *uniterSuite) TestSetUnitStatus(c *gc.C) {
225
225
        now := time.Now()
226
226
        sInfo := status.StatusInfo{
227
 
                Status:  status.StatusActive,
 
227
                Status:  status.Active,
228
228
                Message: "blah",
229
229
                Since:   &now,
230
230
        }
231
231
        err := s.wordpressUnit.SetStatus(sInfo)
232
232
        c.Assert(err, jc.ErrorIsNil)
233
233
        sInfo = status.StatusInfo{
234
 
                Status:  status.StatusTerminated,
 
234
                Status:  status.Terminated,
235
235
                Message: "foo",
236
236
                Since:   &now,
237
237
        }
240
240
 
241
241
        args := params.SetStatus{
242
242
                Entities: []params.EntityStatusArgs{
243
 
                        {Tag: "unit-mysql-0", Status: status.StatusError.String(), Info: "not really"},
244
 
                        {Tag: "unit-wordpress-0", Status: status.StatusTerminated.String(), Info: "foobar"},
245
 
                        {Tag: "unit-foo-42", Status: status.StatusActive.String(), Info: "blah"},
 
243
                        {Tag: "unit-mysql-0", Status: status.Error.String(), Info: "not really"},
 
244
                        {Tag: "unit-wordpress-0", Status: status.Terminated.String(), Info: "foobar"},
 
245
                        {Tag: "unit-foo-42", Status: status.Active.String(), Info: "blah"},
246
246
                }}
247
247
        result, err := s.uniter.SetUnitStatus(args)
248
248
        c.Assert(err, jc.ErrorIsNil)
257
257
        // Verify mysqlUnit - no change.
258
258
        statusInfo, err := s.mysqlUnit.Status()
259
259
        c.Assert(err, jc.ErrorIsNil)
260
 
        c.Assert(statusInfo.Status, gc.Equals, status.StatusTerminated)
 
260
        c.Assert(statusInfo.Status, gc.Equals, status.Terminated)
261
261
        c.Assert(statusInfo.Message, gc.Equals, "foo")
262
262
        // ...wordpressUnit is fine though.
263
263
        statusInfo, err = s.wordpressUnit.Status()
264
264
        c.Assert(err, jc.ErrorIsNil)
265
 
        c.Assert(statusInfo.Status, gc.Equals, status.StatusTerminated)
 
265
        c.Assert(statusInfo.Status, gc.Equals, status.Terminated)
266
266
        c.Assert(statusInfo.Message, gc.Equals, "foobar")
267
267
}
268
268
 
2139
2139
func (s *uniterSuite) TestUnitStatus(c *gc.C) {
2140
2140
        now := time.Now()
2141
2141
        sInfo := status.StatusInfo{
2142
 
                Status:  status.StatusMaintenance,
 
2142
                Status:  status.Maintenance,
2143
2143
                Message: "blah",
2144
2144
                Since:   &now,
2145
2145
        }
2146
2146
        err := s.wordpressUnit.SetStatus(sInfo)
2147
2147
        c.Assert(err, jc.ErrorIsNil)
2148
2148
        sInfo = status.StatusInfo{
2149
 
                Status:  status.StatusTerminated,
 
2149
                Status:  status.Terminated,
2150
2150
                Message: "foo",
2151
2151
                Since:   &now,
2152
2152
        }
2175
2175
        c.Assert(result, gc.DeepEquals, params.StatusResults{
2176
2176
                Results: []params.StatusResult{
2177
2177
                        {Error: apiservertesting.ErrUnauthorized},
2178
 
                        {Status: status.StatusMaintenance.String(), Info: "blah", Data: map[string]interface{}{}},
 
2178
                        {Status: status.Maintenance.String(), Info: "blah", Data: map[string]interface{}{}},
2179
2179
                        {Error: apiservertesting.ErrUnauthorized},
2180
2180
                        {Error: apiservertesting.ErrUnauthorized},
2181
2181
                        {Error: apiservertesting.ServerError(`"invalid" is not a valid tag`)},