github.com/xmidt-org/webpa-common@v1.11.9/service/zk/mocks_test.go (about) 1 package zk 2 3 import ( 4 "github.com/go-kit/kit/log" 5 gokitzk "github.com/go-kit/kit/sd/zk" 6 zkclient "github.com/samuel/go-zookeeper/zk" 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 = gokitzk.NewClient 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(servers []string, logger log.Logger, options ...gokitzk.Option) (gokitzk.Client, error) { 29 arguments := m.Called(servers, logger, options) 30 31 first, _ := arguments.Get(0).(gokitzk.Client) 32 return first, arguments.Error(1) 33 } 34 35 type mockClient struct { 36 mock.Mock 37 } 38 39 func (m *mockClient) GetEntries(path string) ([]string, <-chan zkclient.Event, error) { 40 arguments := m.Called(path) 41 return arguments.Get(0).([]string), 42 arguments.Get(1).(<-chan zkclient.Event), 43 arguments.Error(2) 44 } 45 46 func (m *mockClient) CreateParentNodes(path string) error { 47 return m.Called(path).Error(0) 48 } 49 50 func (m *mockClient) Register(s *gokitzk.Service) error { 51 return m.Called(s).Error(0) 52 } 53 54 func (m *mockClient) Deregister(s *gokitzk.Service) error { 55 return m.Called(s).Error(0) 56 } 57 58 func (m *mockClient) Stop() { 59 m.Called() 60 }