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  }