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  }