github.com/xmidt-org/webpa-common@v1.11.9/service/consul/mocks_test.go (about) 1 package consul 2 3 import ( 4 "time" 5 6 "github.com/hashicorp/consul/api" 7 "github.com/stretchr/testify/mock" 8 ) 9 10 // resekClientFactory resets the global singleton factory function 11 // to its original value. This function is handy as a defer for tests. 12 func resetClientFactory() { 13 clientFactory = defaultClientFactory 14 } 15 16 // prepareMockClientFactory creates a new mockClientFactory and sets up this package 17 // to use it. 18 func prepareMockClientFactory() *mockClientFactory { 19 m := new(mockClientFactory) 20 clientFactory = m.NewClient 21 return m 22 } 23 24 type mockClientFactory struct { 25 mock.Mock 26 } 27 28 func (m *mockClientFactory) NewClient(c *api.Client) (Client, ttlUpdater) { 29 arguments := m.Called(c) 30 return arguments.Get(0).(Client), 31 arguments.Get(1).(ttlUpdater) 32 } 33 34 func resetTickerFactory() { 35 tickerFactory = defaultTickerFactory 36 } 37 38 func prepareMockTickerFactory() *mockTickerFactory { 39 m := new(mockTickerFactory) 40 tickerFactory = m.NewTicker 41 return m 42 } 43 44 type mockTickerFactory struct { 45 mock.Mock 46 } 47 48 func (m *mockTickerFactory) NewTicker(d time.Duration) (<-chan time.Time, func()) { 49 arguments := m.Called(d) 50 return arguments.Get(0).(<-chan time.Time), arguments.Get(1).(func()) 51 } 52 53 type mockClient struct { 54 mock.Mock 55 } 56 57 var _ Client = (*mockClient)(nil) 58 59 func (m *mockClient) Register(r *api.AgentServiceRegistration) error { 60 return m.Called(r).Error(0) 61 } 62 63 func (m *mockClient) Deregister(r *api.AgentServiceRegistration) error { 64 return m.Called(r).Error(0) 65 } 66 67 func (m *mockClient) Service(service, tag string, passingOnly bool, queryOpts *api.QueryOptions) ([]*api.ServiceEntry, *api.QueryMeta, error) { 68 var ( 69 arguments = m.Called(service, tag, passingOnly, queryOpts) 70 first, _ = arguments.Get(0).([]*api.ServiceEntry) 71 second, _ = arguments.Get(1).(*api.QueryMeta) 72 ) 73 74 return first, second, arguments.Error(2) 75 } 76 77 func (m *mockClient) Datacenters() ([]string, error) { 78 arguments := m.Called() 79 first, _ := arguments.Get(0).([]string) 80 return first, arguments.Error(1) 81 } 82 83 type mockTTLUpdater struct { 84 mock.Mock 85 } 86 87 func (m *mockTTLUpdater) UpdateTTL(checkID, output, status string) error { 88 return m.Called(checkID, output, status).Error(0) 89 }