github.com/xmidt-org/webpa-common@v1.11.9/service/mocks.go (about) 1 package service 2 3 import ( 4 "github.com/go-kit/kit/sd" 5 "github.com/stretchr/testify/mock" 6 "github.com/xmidt-org/webpa-common/xmetrics" 7 ) 8 9 // MockAccessor is a mocked Accessor 10 type MockAccessor struct { 11 mock.Mock 12 } 13 14 var _ Accessor = (*MockAccessor)(nil) 15 16 func (m *MockAccessor) Get(v []byte) (string, error) { 17 arguments := m.Called(v) 18 return arguments.String(0), arguments.Error(1) 19 } 20 21 // MockRegistrar is a stretchr/testify mocked sd.Registrar 22 type MockRegistrar struct { 23 mock.Mock 24 } 25 26 var _ sd.Registrar = (*MockRegistrar)(nil) 27 28 func (m *MockRegistrar) Register() { 29 m.Called() 30 } 31 32 func (m *MockRegistrar) Deregister() { 33 m.Called() 34 } 35 36 // MockInstancer is a stretchr/testify mocked sd.Instancer 37 type MockInstancer struct { 38 mock.Mock 39 } 40 41 var _ sd.Instancer = (*MockInstancer)(nil) 42 43 func (m *MockInstancer) Register(events chan<- sd.Event) { 44 m.Called(events) 45 } 46 47 func (m *MockInstancer) Deregister(events chan<- sd.Event) { 48 m.Called(events) 49 } 50 51 func (m *MockInstancer) Stop() { 52 m.Called() 53 } 54 55 // MockEnvironment is a stretchr/testify mocked Environment 56 type MockEnvironment struct { 57 mock.Mock 58 } 59 60 var _ Environment = (*MockEnvironment)(nil) 61 62 func (m *MockEnvironment) Register() { 63 m.Called() 64 } 65 66 func (m *MockEnvironment) Deregister() { 67 m.Called() 68 } 69 70 func (m *MockEnvironment) Close() error { 71 return m.Called().Error(0) 72 } 73 74 func (m *MockEnvironment) IsRegistered(instance string) bool { 75 return m.Called(instance).Bool(0) 76 } 77 78 func (m *MockEnvironment) DefaultScheme() string { 79 return m.Called().String(0) 80 } 81 82 func (m *MockEnvironment) Instancers() Instancers { 83 return m.Called().Get(0).(Instancers) 84 } 85 86 func (m *MockEnvironment) UpdateInstancers(currentKeys map[string]bool, instancersToAdd Instancers) { 87 m.Called(currentKeys, instancersToAdd) 88 } 89 90 func (m *MockEnvironment) AccessorFactory() AccessorFactory { 91 return m.Called().Get(0).(AccessorFactory) 92 } 93 94 func (m *MockEnvironment) Closed() <-chan struct{} { 95 return m.Called().Get(0).(<-chan struct{}) 96 } 97 98 func (m *MockEnvironment) Provider() xmetrics.Registry { 99 if m.Called().Get(1).(bool) { 100 return m.Called().Get(0).(xmetrics.Registry) 101 } 102 103 return nil 104 105 }