← Back to branch summary

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

« back to all changes in this revision

Viewing changes to src/github.com/juju/juju/apiserver/instancepoller/mock_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:
src/github.com/juju/juju/apiserver/instancepoller/mock_test.go
17
17
        "github.com/juju/juju/instance"
18
18
        "github.com/juju/juju/network"
19
19
        "github.com/juju/juju/state"
 
20
        "github.com/juju/juju/status"
20
21
)
21
22
 
22
23
// mockState implements StateInterface and allows inspection of called
206
207
type machineInfo struct {
207
208
        id                string
208
209
        instanceId        instance.Id
209
 
        status            state.StatusInfo
210
 
        instanceStatus    string
 
210
        status            status.StatusInfo
 
211
        instanceStatus    status.StatusInfo
211
212
        providerAddresses []network.Address
212
213
        life              state.Life
213
214
        isManual          bool
264
265
}
265
266
 
266
267
// InstanceStatus implements StateMachine.
267
 
func (m *mockMachine) InstanceStatus() (string, error) {
 
268
func (m *mockMachine) InstanceStatus() (status.StatusInfo, error) {
268
269
        m.mu.Lock()
269
270
        defer m.mu.Unlock()
270
271
 
271
272
        m.MethodCall(m, "InstanceStatus")
272
273
        if err := m.NextErr(); err != nil {
273
 
                return "", err
 
274
                return status.StatusInfo{}, err
274
275
        }
275
276
        return m.instanceStatus, nil
276
277
}
277
278
 
278
279
// SetInstanceStatus implements StateMachine.
279
 
func (m *mockMachine) SetInstanceStatus(status string) error {
 
280
func (m *mockMachine) SetInstanceStatus(instanceStatus status.Status, info string, data map[string]interface{}) error {
280
281
        m.mu.Lock()
281
282
        defer m.mu.Unlock()
282
283
 
283
 
        m.MethodCall(m, "SetInstanceStatus", status)
 
284
        m.MethodCall(m, "SetInstanceStatus", instanceStatus)
284
285
        if err := m.NextErr(); err != nil {
285
286
                return err
286
287
        }
287
 
        m.instanceStatus = status
 
288
        m.instanceStatus = status.StatusInfo{
 
289
                Status:  instanceStatus,
 
290
                Message: info,
 
291
                Data:    data,
 
292
        }
288
293
        return nil
289
294
}
290
295
 
308
313
}
309
314
 
310
315
// Status implements StateMachine.
311
 
func (m *mockMachine) Status() (state.StatusInfo, error) {
 
316
func (m *mockMachine) Status() (status.StatusInfo, error) {
312
317
        m.mu.Lock()
313
318
        defer m.mu.Unlock()
314
319