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  }