github.com/cloud-green/juju@v0.0.0-20151002100041-a00291338d3d/worker/machiner/mock_test.go (about) 1 // Copyright 2015 Canonical Ltd. 2 // Licensed under the AGPLv3, see LICENCE file for details. 3 4 package machiner_test 5 6 import ( 7 "github.com/juju/names" 8 gitjujutesting "github.com/juju/testing" 9 10 "github.com/juju/juju/api/watcher" 11 "github.com/juju/juju/apiserver/params" 12 "github.com/juju/juju/network" 13 "github.com/juju/juju/worker/machiner" 14 ) 15 16 type mockWatcher struct { 17 gitjujutesting.Stub 18 changes chan struct{} 19 } 20 21 func (w *mockWatcher) Changes() <-chan struct{} { 22 w.MethodCall(w, "Changes") 23 return w.changes 24 } 25 26 func (w *mockWatcher) Stop() error { 27 w.MethodCall(w, "Stop") 28 return w.NextErr() 29 } 30 31 func (w *mockWatcher) Err() error { 32 w.MethodCall(w, "Err") 33 return w.NextErr() 34 } 35 36 type mockMachine struct { 37 machiner.Machine 38 gitjujutesting.Stub 39 watcher mockWatcher 40 life params.Life 41 } 42 43 func (m *mockMachine) Refresh() error { 44 m.MethodCall(m, "Refresh") 45 return m.NextErr() 46 } 47 48 func (m *mockMachine) Life() params.Life { 49 m.MethodCall(m, "Life") 50 return m.life 51 } 52 53 func (m *mockMachine) EnsureDead() error { 54 m.MethodCall(m, "EnsureDead") 55 return m.NextErr() 56 } 57 58 func (m *mockMachine) SetMachineAddresses(addresses []network.Address) error { 59 m.MethodCall(m, "SetMachineAddresses", addresses) 60 return m.NextErr() 61 } 62 63 func (m *mockMachine) SetStatus(status params.Status, info string, data map[string]interface{}) error { 64 m.MethodCall(m, "SetStatus", status, info, data) 65 return m.NextErr() 66 } 67 68 func (m *mockMachine) Watch() (watcher.NotifyWatcher, error) { 69 m.MethodCall(m, "Watch") 70 if err := m.NextErr(); err != nil { 71 return nil, err 72 } 73 return &m.watcher, nil 74 } 75 76 type mockMachineAccessor struct { 77 gitjujutesting.Stub 78 machine mockMachine 79 } 80 81 func (a *mockMachineAccessor) Machine(tag names.MachineTag) (machiner.Machine, error) { 82 a.MethodCall(a, "Machine", tag) 83 if err := a.NextErr(); err != nil { 84 return nil, err 85 } 86 return &a.machine, nil 87 }