github.com/m3db/m3@v1.5.0/src/cluster/services/services_mock.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: github.com/m3db/m3/src/cluster/services/types.go
     3  
     4  // Copyright (c) 2021 Uber Technologies, Inc.
     5  //
     6  // Permission is hereby granted, free of charge, to any person obtaining a copy
     7  // of this software and associated documentation files (the "Software"), to deal
     8  // in the Software without restriction, including without limitation the rights
     9  // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
    10  // copies of the Software, and to permit persons to whom the Software is
    11  // furnished to do so, subject to the following conditions:
    12  //
    13  // The above copyright notice and this permission notice shall be included in
    14  // all copies or substantial portions of the Software.
    15  //
    16  // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
    17  // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
    18  // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
    19  // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
    20  // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
    21  // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
    22  // THE SOFTWARE.
    23  
    24  // Package services is a generated GoMock package.
    25  package services
    26  
    27  import (
    28  	"reflect"
    29  	"time"
    30  
    31  	"github.com/m3db/m3/src/cluster/generated/proto/metadatapb"
    32  	"github.com/m3db/m3/src/cluster/placement"
    33  	"github.com/m3db/m3/src/cluster/services/leader/campaign"
    34  	"github.com/m3db/m3/src/cluster/shard"
    35  	"github.com/m3db/m3/src/x/instrument"
    36  	"github.com/m3db/m3/src/x/watch"
    37  
    38  	"github.com/golang/mock/gomock"
    39  )
    40  
    41  // MockServices is a mock of Services interface.
    42  type MockServices struct {
    43  	ctrl     *gomock.Controller
    44  	recorder *MockServicesMockRecorder
    45  }
    46  
    47  // MockServicesMockRecorder is the mock recorder for MockServices.
    48  type MockServicesMockRecorder struct {
    49  	mock *MockServices
    50  }
    51  
    52  // NewMockServices creates a new mock instance.
    53  func NewMockServices(ctrl *gomock.Controller) *MockServices {
    54  	mock := &MockServices{ctrl: ctrl}
    55  	mock.recorder = &MockServicesMockRecorder{mock}
    56  	return mock
    57  }
    58  
    59  // EXPECT returns an object that allows the caller to indicate expected use.
    60  func (m *MockServices) EXPECT() *MockServicesMockRecorder {
    61  	return m.recorder
    62  }
    63  
    64  // Advertise mocks base method.
    65  func (m *MockServices) Advertise(ad Advertisement) error {
    66  	m.ctrl.T.Helper()
    67  	ret := m.ctrl.Call(m, "Advertise", ad)
    68  	ret0, _ := ret[0].(error)
    69  	return ret0
    70  }
    71  
    72  // Advertise indicates an expected call of Advertise.
    73  func (mr *MockServicesMockRecorder) Advertise(ad interface{}) *gomock.Call {
    74  	mr.mock.ctrl.T.Helper()
    75  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Advertise", reflect.TypeOf((*MockServices)(nil).Advertise), ad)
    76  }
    77  
    78  // DeleteMetadata mocks base method.
    79  func (m *MockServices) DeleteMetadata(sid ServiceID) error {
    80  	m.ctrl.T.Helper()
    81  	ret := m.ctrl.Call(m, "DeleteMetadata", sid)
    82  	ret0, _ := ret[0].(error)
    83  	return ret0
    84  }
    85  
    86  // DeleteMetadata indicates an expected call of DeleteMetadata.
    87  func (mr *MockServicesMockRecorder) DeleteMetadata(sid interface{}) *gomock.Call {
    88  	mr.mock.ctrl.T.Helper()
    89  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteMetadata", reflect.TypeOf((*MockServices)(nil).DeleteMetadata), sid)
    90  }
    91  
    92  // HeartbeatService mocks base method.
    93  func (m *MockServices) HeartbeatService(service ServiceID) (HeartbeatService, error) {
    94  	m.ctrl.T.Helper()
    95  	ret := m.ctrl.Call(m, "HeartbeatService", service)
    96  	ret0, _ := ret[0].(HeartbeatService)
    97  	ret1, _ := ret[1].(error)
    98  	return ret0, ret1
    99  }
   100  
   101  // HeartbeatService indicates an expected call of HeartbeatService.
   102  func (mr *MockServicesMockRecorder) HeartbeatService(service interface{}) *gomock.Call {
   103  	mr.mock.ctrl.T.Helper()
   104  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HeartbeatService", reflect.TypeOf((*MockServices)(nil).HeartbeatService), service)
   105  }
   106  
   107  // LeaderService mocks base method.
   108  func (m *MockServices) LeaderService(service ServiceID, opts ElectionOptions) (LeaderService, error) {
   109  	m.ctrl.T.Helper()
   110  	ret := m.ctrl.Call(m, "LeaderService", service, opts)
   111  	ret0, _ := ret[0].(LeaderService)
   112  	ret1, _ := ret[1].(error)
   113  	return ret0, ret1
   114  }
   115  
   116  // LeaderService indicates an expected call of LeaderService.
   117  func (mr *MockServicesMockRecorder) LeaderService(service, opts interface{}) *gomock.Call {
   118  	mr.mock.ctrl.T.Helper()
   119  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LeaderService", reflect.TypeOf((*MockServices)(nil).LeaderService), service, opts)
   120  }
   121  
   122  // Metadata mocks base method.
   123  func (m *MockServices) Metadata(sid ServiceID) (Metadata, error) {
   124  	m.ctrl.T.Helper()
   125  	ret := m.ctrl.Call(m, "Metadata", sid)
   126  	ret0, _ := ret[0].(Metadata)
   127  	ret1, _ := ret[1].(error)
   128  	return ret0, ret1
   129  }
   130  
   131  // Metadata indicates an expected call of Metadata.
   132  func (mr *MockServicesMockRecorder) Metadata(sid interface{}) *gomock.Call {
   133  	mr.mock.ctrl.T.Helper()
   134  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Metadata", reflect.TypeOf((*MockServices)(nil).Metadata), sid)
   135  }
   136  
   137  // PlacementService mocks base method.
   138  func (m *MockServices) PlacementService(sid ServiceID, popts placement.Options) (placement.Service, error) {
   139  	m.ctrl.T.Helper()
   140  	ret := m.ctrl.Call(m, "PlacementService", sid, popts)
   141  	ret0, _ := ret[0].(placement.Service)
   142  	ret1, _ := ret[1].(error)
   143  	return ret0, ret1
   144  }
   145  
   146  // PlacementService indicates an expected call of PlacementService.
   147  func (mr *MockServicesMockRecorder) PlacementService(sid, popts interface{}) *gomock.Call {
   148  	mr.mock.ctrl.T.Helper()
   149  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PlacementService", reflect.TypeOf((*MockServices)(nil).PlacementService), sid, popts)
   150  }
   151  
   152  // Query mocks base method.
   153  func (m *MockServices) Query(service ServiceID, opts QueryOptions) (Service, error) {
   154  	m.ctrl.T.Helper()
   155  	ret := m.ctrl.Call(m, "Query", service, opts)
   156  	ret0, _ := ret[0].(Service)
   157  	ret1, _ := ret[1].(error)
   158  	return ret0, ret1
   159  }
   160  
   161  // Query indicates an expected call of Query.
   162  func (mr *MockServicesMockRecorder) Query(service, opts interface{}) *gomock.Call {
   163  	mr.mock.ctrl.T.Helper()
   164  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Query", reflect.TypeOf((*MockServices)(nil).Query), service, opts)
   165  }
   166  
   167  // SetMetadata mocks base method.
   168  func (m_2 *MockServices) SetMetadata(sid ServiceID, m Metadata) error {
   169  	m_2.ctrl.T.Helper()
   170  	ret := m_2.ctrl.Call(m_2, "SetMetadata", sid, m)
   171  	ret0, _ := ret[0].(error)
   172  	return ret0
   173  }
   174  
   175  // SetMetadata indicates an expected call of SetMetadata.
   176  func (mr *MockServicesMockRecorder) SetMetadata(sid, m interface{}) *gomock.Call {
   177  	mr.mock.ctrl.T.Helper()
   178  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetMetadata", reflect.TypeOf((*MockServices)(nil).SetMetadata), sid, m)
   179  }
   180  
   181  // Unadvertise mocks base method.
   182  func (m *MockServices) Unadvertise(service ServiceID, id string) error {
   183  	m.ctrl.T.Helper()
   184  	ret := m.ctrl.Call(m, "Unadvertise", service, id)
   185  	ret0, _ := ret[0].(error)
   186  	return ret0
   187  }
   188  
   189  // Unadvertise indicates an expected call of Unadvertise.
   190  func (mr *MockServicesMockRecorder) Unadvertise(service, id interface{}) *gomock.Call {
   191  	mr.mock.ctrl.T.Helper()
   192  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Unadvertise", reflect.TypeOf((*MockServices)(nil).Unadvertise), service, id)
   193  }
   194  
   195  // Watch mocks base method.
   196  func (m *MockServices) Watch(service ServiceID, opts QueryOptions) (Watch, error) {
   197  	m.ctrl.T.Helper()
   198  	ret := m.ctrl.Call(m, "Watch", service, opts)
   199  	ret0, _ := ret[0].(Watch)
   200  	ret1, _ := ret[1].(error)
   201  	return ret0, ret1
   202  }
   203  
   204  // Watch indicates an expected call of Watch.
   205  func (mr *MockServicesMockRecorder) Watch(service, opts interface{}) *gomock.Call {
   206  	mr.mock.ctrl.T.Helper()
   207  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Watch", reflect.TypeOf((*MockServices)(nil).Watch), service, opts)
   208  }
   209  
   210  // MockOptions is a mock of Options interface.
   211  type MockOptions struct {
   212  	ctrl     *gomock.Controller
   213  	recorder *MockOptionsMockRecorder
   214  }
   215  
   216  // MockOptionsMockRecorder is the mock recorder for MockOptions.
   217  type MockOptionsMockRecorder struct {
   218  	mock *MockOptions
   219  }
   220  
   221  // NewMockOptions creates a new mock instance.
   222  func NewMockOptions(ctrl *gomock.Controller) *MockOptions {
   223  	mock := &MockOptions{ctrl: ctrl}
   224  	mock.recorder = &MockOptionsMockRecorder{mock}
   225  	return mock
   226  }
   227  
   228  // EXPECT returns an object that allows the caller to indicate expected use.
   229  func (m *MockOptions) EXPECT() *MockOptionsMockRecorder {
   230  	return m.recorder
   231  }
   232  
   233  // HeartbeatGen mocks base method.
   234  func (m *MockOptions) HeartbeatGen() HeartbeatGen {
   235  	m.ctrl.T.Helper()
   236  	ret := m.ctrl.Call(m, "HeartbeatGen")
   237  	ret0, _ := ret[0].(HeartbeatGen)
   238  	return ret0
   239  }
   240  
   241  // HeartbeatGen indicates an expected call of HeartbeatGen.
   242  func (mr *MockOptionsMockRecorder) HeartbeatGen() *gomock.Call {
   243  	mr.mock.ctrl.T.Helper()
   244  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HeartbeatGen", reflect.TypeOf((*MockOptions)(nil).HeartbeatGen))
   245  }
   246  
   247  // InitTimeout mocks base method.
   248  func (m *MockOptions) InitTimeout() time.Duration {
   249  	m.ctrl.T.Helper()
   250  	ret := m.ctrl.Call(m, "InitTimeout")
   251  	ret0, _ := ret[0].(time.Duration)
   252  	return ret0
   253  }
   254  
   255  // InitTimeout indicates an expected call of InitTimeout.
   256  func (mr *MockOptionsMockRecorder) InitTimeout() *gomock.Call {
   257  	mr.mock.ctrl.T.Helper()
   258  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InitTimeout", reflect.TypeOf((*MockOptions)(nil).InitTimeout))
   259  }
   260  
   261  // InstrumentsOptions mocks base method.
   262  func (m *MockOptions) InstrumentsOptions() instrument.Options {
   263  	m.ctrl.T.Helper()
   264  	ret := m.ctrl.Call(m, "InstrumentsOptions")
   265  	ret0, _ := ret[0].(instrument.Options)
   266  	return ret0
   267  }
   268  
   269  // InstrumentsOptions indicates an expected call of InstrumentsOptions.
   270  func (mr *MockOptionsMockRecorder) InstrumentsOptions() *gomock.Call {
   271  	mr.mock.ctrl.T.Helper()
   272  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InstrumentsOptions", reflect.TypeOf((*MockOptions)(nil).InstrumentsOptions))
   273  }
   274  
   275  // KVGen mocks base method.
   276  func (m *MockOptions) KVGen() KVGen {
   277  	m.ctrl.T.Helper()
   278  	ret := m.ctrl.Call(m, "KVGen")
   279  	ret0, _ := ret[0].(KVGen)
   280  	return ret0
   281  }
   282  
   283  // KVGen indicates an expected call of KVGen.
   284  func (mr *MockOptionsMockRecorder) KVGen() *gomock.Call {
   285  	mr.mock.ctrl.T.Helper()
   286  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "KVGen", reflect.TypeOf((*MockOptions)(nil).KVGen))
   287  }
   288  
   289  // LeaderGen mocks base method.
   290  func (m *MockOptions) LeaderGen() LeaderGen {
   291  	m.ctrl.T.Helper()
   292  	ret := m.ctrl.Call(m, "LeaderGen")
   293  	ret0, _ := ret[0].(LeaderGen)
   294  	return ret0
   295  }
   296  
   297  // LeaderGen indicates an expected call of LeaderGen.
   298  func (mr *MockOptionsMockRecorder) LeaderGen() *gomock.Call {
   299  	mr.mock.ctrl.T.Helper()
   300  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LeaderGen", reflect.TypeOf((*MockOptions)(nil).LeaderGen))
   301  }
   302  
   303  // NamespaceOptions mocks base method.
   304  func (m *MockOptions) NamespaceOptions() NamespaceOptions {
   305  	m.ctrl.T.Helper()
   306  	ret := m.ctrl.Call(m, "NamespaceOptions")
   307  	ret0, _ := ret[0].(NamespaceOptions)
   308  	return ret0
   309  }
   310  
   311  // NamespaceOptions indicates an expected call of NamespaceOptions.
   312  func (mr *MockOptionsMockRecorder) NamespaceOptions() *gomock.Call {
   313  	mr.mock.ctrl.T.Helper()
   314  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NamespaceOptions", reflect.TypeOf((*MockOptions)(nil).NamespaceOptions))
   315  }
   316  
   317  // SetHeartbeatGen mocks base method.
   318  func (m *MockOptions) SetHeartbeatGen(gen HeartbeatGen) Options {
   319  	m.ctrl.T.Helper()
   320  	ret := m.ctrl.Call(m, "SetHeartbeatGen", gen)
   321  	ret0, _ := ret[0].(Options)
   322  	return ret0
   323  }
   324  
   325  // SetHeartbeatGen indicates an expected call of SetHeartbeatGen.
   326  func (mr *MockOptionsMockRecorder) SetHeartbeatGen(gen interface{}) *gomock.Call {
   327  	mr.mock.ctrl.T.Helper()
   328  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetHeartbeatGen", reflect.TypeOf((*MockOptions)(nil).SetHeartbeatGen), gen)
   329  }
   330  
   331  // SetInitTimeout mocks base method.
   332  func (m *MockOptions) SetInitTimeout(t time.Duration) Options {
   333  	m.ctrl.T.Helper()
   334  	ret := m.ctrl.Call(m, "SetInitTimeout", t)
   335  	ret0, _ := ret[0].(Options)
   336  	return ret0
   337  }
   338  
   339  // SetInitTimeout indicates an expected call of SetInitTimeout.
   340  func (mr *MockOptionsMockRecorder) SetInitTimeout(t interface{}) *gomock.Call {
   341  	mr.mock.ctrl.T.Helper()
   342  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetInitTimeout", reflect.TypeOf((*MockOptions)(nil).SetInitTimeout), t)
   343  }
   344  
   345  // SetInstrumentsOptions mocks base method.
   346  func (m *MockOptions) SetInstrumentsOptions(iopts instrument.Options) Options {
   347  	m.ctrl.T.Helper()
   348  	ret := m.ctrl.Call(m, "SetInstrumentsOptions", iopts)
   349  	ret0, _ := ret[0].(Options)
   350  	return ret0
   351  }
   352  
   353  // SetInstrumentsOptions indicates an expected call of SetInstrumentsOptions.
   354  func (mr *MockOptionsMockRecorder) SetInstrumentsOptions(iopts interface{}) *gomock.Call {
   355  	mr.mock.ctrl.T.Helper()
   356  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetInstrumentsOptions", reflect.TypeOf((*MockOptions)(nil).SetInstrumentsOptions), iopts)
   357  }
   358  
   359  // SetKVGen mocks base method.
   360  func (m *MockOptions) SetKVGen(gen KVGen) Options {
   361  	m.ctrl.T.Helper()
   362  	ret := m.ctrl.Call(m, "SetKVGen", gen)
   363  	ret0, _ := ret[0].(Options)
   364  	return ret0
   365  }
   366  
   367  // SetKVGen indicates an expected call of SetKVGen.
   368  func (mr *MockOptionsMockRecorder) SetKVGen(gen interface{}) *gomock.Call {
   369  	mr.mock.ctrl.T.Helper()
   370  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetKVGen", reflect.TypeOf((*MockOptions)(nil).SetKVGen), gen)
   371  }
   372  
   373  // SetLeaderGen mocks base method.
   374  func (m *MockOptions) SetLeaderGen(gen LeaderGen) Options {
   375  	m.ctrl.T.Helper()
   376  	ret := m.ctrl.Call(m, "SetLeaderGen", gen)
   377  	ret0, _ := ret[0].(Options)
   378  	return ret0
   379  }
   380  
   381  // SetLeaderGen indicates an expected call of SetLeaderGen.
   382  func (mr *MockOptionsMockRecorder) SetLeaderGen(gen interface{}) *gomock.Call {
   383  	mr.mock.ctrl.T.Helper()
   384  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetLeaderGen", reflect.TypeOf((*MockOptions)(nil).SetLeaderGen), gen)
   385  }
   386  
   387  // SetNamespaceOptions mocks base method.
   388  func (m *MockOptions) SetNamespaceOptions(opts NamespaceOptions) Options {
   389  	m.ctrl.T.Helper()
   390  	ret := m.ctrl.Call(m, "SetNamespaceOptions", opts)
   391  	ret0, _ := ret[0].(Options)
   392  	return ret0
   393  }
   394  
   395  // SetNamespaceOptions indicates an expected call of SetNamespaceOptions.
   396  func (mr *MockOptionsMockRecorder) SetNamespaceOptions(opts interface{}) *gomock.Call {
   397  	mr.mock.ctrl.T.Helper()
   398  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetNamespaceOptions", reflect.TypeOf((*MockOptions)(nil).SetNamespaceOptions), opts)
   399  }
   400  
   401  // Validate mocks base method.
   402  func (m *MockOptions) Validate() error {
   403  	m.ctrl.T.Helper()
   404  	ret := m.ctrl.Call(m, "Validate")
   405  	ret0, _ := ret[0].(error)
   406  	return ret0
   407  }
   408  
   409  // Validate indicates an expected call of Validate.
   410  func (mr *MockOptionsMockRecorder) Validate() *gomock.Call {
   411  	mr.mock.ctrl.T.Helper()
   412  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Validate", reflect.TypeOf((*MockOptions)(nil).Validate))
   413  }
   414  
   415  // MockNamespaceOptions is a mock of NamespaceOptions interface.
   416  type MockNamespaceOptions struct {
   417  	ctrl     *gomock.Controller
   418  	recorder *MockNamespaceOptionsMockRecorder
   419  }
   420  
   421  // MockNamespaceOptionsMockRecorder is the mock recorder for MockNamespaceOptions.
   422  type MockNamespaceOptionsMockRecorder struct {
   423  	mock *MockNamespaceOptions
   424  }
   425  
   426  // NewMockNamespaceOptions creates a new mock instance.
   427  func NewMockNamespaceOptions(ctrl *gomock.Controller) *MockNamespaceOptions {
   428  	mock := &MockNamespaceOptions{ctrl: ctrl}
   429  	mock.recorder = &MockNamespaceOptionsMockRecorder{mock}
   430  	return mock
   431  }
   432  
   433  // EXPECT returns an object that allows the caller to indicate expected use.
   434  func (m *MockNamespaceOptions) EXPECT() *MockNamespaceOptionsMockRecorder {
   435  	return m.recorder
   436  }
   437  
   438  // MetadataNamespace mocks base method.
   439  func (m *MockNamespaceOptions) MetadataNamespace() string {
   440  	m.ctrl.T.Helper()
   441  	ret := m.ctrl.Call(m, "MetadataNamespace")
   442  	ret0, _ := ret[0].(string)
   443  	return ret0
   444  }
   445  
   446  // MetadataNamespace indicates an expected call of MetadataNamespace.
   447  func (mr *MockNamespaceOptionsMockRecorder) MetadataNamespace() *gomock.Call {
   448  	mr.mock.ctrl.T.Helper()
   449  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MetadataNamespace", reflect.TypeOf((*MockNamespaceOptions)(nil).MetadataNamespace))
   450  }
   451  
   452  // PlacementNamespace mocks base method.
   453  func (m *MockNamespaceOptions) PlacementNamespace() string {
   454  	m.ctrl.T.Helper()
   455  	ret := m.ctrl.Call(m, "PlacementNamespace")
   456  	ret0, _ := ret[0].(string)
   457  	return ret0
   458  }
   459  
   460  // PlacementNamespace indicates an expected call of PlacementNamespace.
   461  func (mr *MockNamespaceOptionsMockRecorder) PlacementNamespace() *gomock.Call {
   462  	mr.mock.ctrl.T.Helper()
   463  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PlacementNamespace", reflect.TypeOf((*MockNamespaceOptions)(nil).PlacementNamespace))
   464  }
   465  
   466  // SetMetadataNamespace mocks base method.
   467  func (m *MockNamespaceOptions) SetMetadataNamespace(v string) NamespaceOptions {
   468  	m.ctrl.T.Helper()
   469  	ret := m.ctrl.Call(m, "SetMetadataNamespace", v)
   470  	ret0, _ := ret[0].(NamespaceOptions)
   471  	return ret0
   472  }
   473  
   474  // SetMetadataNamespace indicates an expected call of SetMetadataNamespace.
   475  func (mr *MockNamespaceOptionsMockRecorder) SetMetadataNamespace(v interface{}) *gomock.Call {
   476  	mr.mock.ctrl.T.Helper()
   477  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetMetadataNamespace", reflect.TypeOf((*MockNamespaceOptions)(nil).SetMetadataNamespace), v)
   478  }
   479  
   480  // SetPlacementNamespace mocks base method.
   481  func (m *MockNamespaceOptions) SetPlacementNamespace(v string) NamespaceOptions {
   482  	m.ctrl.T.Helper()
   483  	ret := m.ctrl.Call(m, "SetPlacementNamespace", v)
   484  	ret0, _ := ret[0].(NamespaceOptions)
   485  	return ret0
   486  }
   487  
   488  // SetPlacementNamespace indicates an expected call of SetPlacementNamespace.
   489  func (mr *MockNamespaceOptionsMockRecorder) SetPlacementNamespace(v interface{}) *gomock.Call {
   490  	mr.mock.ctrl.T.Helper()
   491  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetPlacementNamespace", reflect.TypeOf((*MockNamespaceOptions)(nil).SetPlacementNamespace), v)
   492  }
   493  
   494  // MockOverrideOptions is a mock of OverrideOptions interface.
   495  type MockOverrideOptions struct {
   496  	ctrl     *gomock.Controller
   497  	recorder *MockOverrideOptionsMockRecorder
   498  }
   499  
   500  // MockOverrideOptionsMockRecorder is the mock recorder for MockOverrideOptions.
   501  type MockOverrideOptionsMockRecorder struct {
   502  	mock *MockOverrideOptions
   503  }
   504  
   505  // NewMockOverrideOptions creates a new mock instance.
   506  func NewMockOverrideOptions(ctrl *gomock.Controller) *MockOverrideOptions {
   507  	mock := &MockOverrideOptions{ctrl: ctrl}
   508  	mock.recorder = &MockOverrideOptionsMockRecorder{mock}
   509  	return mock
   510  }
   511  
   512  // EXPECT returns an object that allows the caller to indicate expected use.
   513  func (m *MockOverrideOptions) EXPECT() *MockOverrideOptionsMockRecorder {
   514  	return m.recorder
   515  }
   516  
   517  // NamespaceOptions mocks base method.
   518  func (m *MockOverrideOptions) NamespaceOptions() NamespaceOptions {
   519  	m.ctrl.T.Helper()
   520  	ret := m.ctrl.Call(m, "NamespaceOptions")
   521  	ret0, _ := ret[0].(NamespaceOptions)
   522  	return ret0
   523  }
   524  
   525  // NamespaceOptions indicates an expected call of NamespaceOptions.
   526  func (mr *MockOverrideOptionsMockRecorder) NamespaceOptions() *gomock.Call {
   527  	mr.mock.ctrl.T.Helper()
   528  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NamespaceOptions", reflect.TypeOf((*MockOverrideOptions)(nil).NamespaceOptions))
   529  }
   530  
   531  // SetNamespaceOptions mocks base method.
   532  func (m *MockOverrideOptions) SetNamespaceOptions(opts NamespaceOptions) OverrideOptions {
   533  	m.ctrl.T.Helper()
   534  	ret := m.ctrl.Call(m, "SetNamespaceOptions", opts)
   535  	ret0, _ := ret[0].(OverrideOptions)
   536  	return ret0
   537  }
   538  
   539  // SetNamespaceOptions indicates an expected call of SetNamespaceOptions.
   540  func (mr *MockOverrideOptionsMockRecorder) SetNamespaceOptions(opts interface{}) *gomock.Call {
   541  	mr.mock.ctrl.T.Helper()
   542  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetNamespaceOptions", reflect.TypeOf((*MockOverrideOptions)(nil).SetNamespaceOptions), opts)
   543  }
   544  
   545  // MockWatch is a mock of Watch interface.
   546  type MockWatch struct {
   547  	ctrl     *gomock.Controller
   548  	recorder *MockWatchMockRecorder
   549  }
   550  
   551  // MockWatchMockRecorder is the mock recorder for MockWatch.
   552  type MockWatchMockRecorder struct {
   553  	mock *MockWatch
   554  }
   555  
   556  // NewMockWatch creates a new mock instance.
   557  func NewMockWatch(ctrl *gomock.Controller) *MockWatch {
   558  	mock := &MockWatch{ctrl: ctrl}
   559  	mock.recorder = &MockWatchMockRecorder{mock}
   560  	return mock
   561  }
   562  
   563  // EXPECT returns an object that allows the caller to indicate expected use.
   564  func (m *MockWatch) EXPECT() *MockWatchMockRecorder {
   565  	return m.recorder
   566  }
   567  
   568  // C mocks base method.
   569  func (m *MockWatch) C() <-chan struct{} {
   570  	m.ctrl.T.Helper()
   571  	ret := m.ctrl.Call(m, "C")
   572  	ret0, _ := ret[0].(<-chan struct{})
   573  	return ret0
   574  }
   575  
   576  // C indicates an expected call of C.
   577  func (mr *MockWatchMockRecorder) C() *gomock.Call {
   578  	mr.mock.ctrl.T.Helper()
   579  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "C", reflect.TypeOf((*MockWatch)(nil).C))
   580  }
   581  
   582  // Close mocks base method.
   583  func (m *MockWatch) Close() {
   584  	m.ctrl.T.Helper()
   585  	m.ctrl.Call(m, "Close")
   586  }
   587  
   588  // Close indicates an expected call of Close.
   589  func (mr *MockWatchMockRecorder) Close() *gomock.Call {
   590  	mr.mock.ctrl.T.Helper()
   591  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockWatch)(nil).Close))
   592  }
   593  
   594  // Get mocks base method.
   595  func (m *MockWatch) Get() Service {
   596  	m.ctrl.T.Helper()
   597  	ret := m.ctrl.Call(m, "Get")
   598  	ret0, _ := ret[0].(Service)
   599  	return ret0
   600  }
   601  
   602  // Get indicates an expected call of Get.
   603  func (mr *MockWatchMockRecorder) Get() *gomock.Call {
   604  	mr.mock.ctrl.T.Helper()
   605  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockWatch)(nil).Get))
   606  }
   607  
   608  // MockService is a mock of Service interface.
   609  type MockService struct {
   610  	ctrl     *gomock.Controller
   611  	recorder *MockServiceMockRecorder
   612  }
   613  
   614  // MockServiceMockRecorder is the mock recorder for MockService.
   615  type MockServiceMockRecorder struct {
   616  	mock *MockService
   617  }
   618  
   619  // NewMockService creates a new mock instance.
   620  func NewMockService(ctrl *gomock.Controller) *MockService {
   621  	mock := &MockService{ctrl: ctrl}
   622  	mock.recorder = &MockServiceMockRecorder{mock}
   623  	return mock
   624  }
   625  
   626  // EXPECT returns an object that allows the caller to indicate expected use.
   627  func (m *MockService) EXPECT() *MockServiceMockRecorder {
   628  	return m.recorder
   629  }
   630  
   631  // Instance mocks base method.
   632  func (m *MockService) Instance(instanceID string) (ServiceInstance, error) {
   633  	m.ctrl.T.Helper()
   634  	ret := m.ctrl.Call(m, "Instance", instanceID)
   635  	ret0, _ := ret[0].(ServiceInstance)
   636  	ret1, _ := ret[1].(error)
   637  	return ret0, ret1
   638  }
   639  
   640  // Instance indicates an expected call of Instance.
   641  func (mr *MockServiceMockRecorder) Instance(instanceID interface{}) *gomock.Call {
   642  	mr.mock.ctrl.T.Helper()
   643  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Instance", reflect.TypeOf((*MockService)(nil).Instance), instanceID)
   644  }
   645  
   646  // Instances mocks base method.
   647  func (m *MockService) Instances() []ServiceInstance {
   648  	m.ctrl.T.Helper()
   649  	ret := m.ctrl.Call(m, "Instances")
   650  	ret0, _ := ret[0].([]ServiceInstance)
   651  	return ret0
   652  }
   653  
   654  // Instances indicates an expected call of Instances.
   655  func (mr *MockServiceMockRecorder) Instances() *gomock.Call {
   656  	mr.mock.ctrl.T.Helper()
   657  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Instances", reflect.TypeOf((*MockService)(nil).Instances))
   658  }
   659  
   660  // Replication mocks base method.
   661  func (m *MockService) Replication() ServiceReplication {
   662  	m.ctrl.T.Helper()
   663  	ret := m.ctrl.Call(m, "Replication")
   664  	ret0, _ := ret[0].(ServiceReplication)
   665  	return ret0
   666  }
   667  
   668  // Replication indicates an expected call of Replication.
   669  func (mr *MockServiceMockRecorder) Replication() *gomock.Call {
   670  	mr.mock.ctrl.T.Helper()
   671  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Replication", reflect.TypeOf((*MockService)(nil).Replication))
   672  }
   673  
   674  // SetInstances mocks base method.
   675  func (m *MockService) SetInstances(insts []ServiceInstance) Service {
   676  	m.ctrl.T.Helper()
   677  	ret := m.ctrl.Call(m, "SetInstances", insts)
   678  	ret0, _ := ret[0].(Service)
   679  	return ret0
   680  }
   681  
   682  // SetInstances indicates an expected call of SetInstances.
   683  func (mr *MockServiceMockRecorder) SetInstances(insts interface{}) *gomock.Call {
   684  	mr.mock.ctrl.T.Helper()
   685  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetInstances", reflect.TypeOf((*MockService)(nil).SetInstances), insts)
   686  }
   687  
   688  // SetReplication mocks base method.
   689  func (m *MockService) SetReplication(r ServiceReplication) Service {
   690  	m.ctrl.T.Helper()
   691  	ret := m.ctrl.Call(m, "SetReplication", r)
   692  	ret0, _ := ret[0].(Service)
   693  	return ret0
   694  }
   695  
   696  // SetReplication indicates an expected call of SetReplication.
   697  func (mr *MockServiceMockRecorder) SetReplication(r interface{}) *gomock.Call {
   698  	mr.mock.ctrl.T.Helper()
   699  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetReplication", reflect.TypeOf((*MockService)(nil).SetReplication), r)
   700  }
   701  
   702  // SetSharding mocks base method.
   703  func (m *MockService) SetSharding(s ServiceSharding) Service {
   704  	m.ctrl.T.Helper()
   705  	ret := m.ctrl.Call(m, "SetSharding", s)
   706  	ret0, _ := ret[0].(Service)
   707  	return ret0
   708  }
   709  
   710  // SetSharding indicates an expected call of SetSharding.
   711  func (mr *MockServiceMockRecorder) SetSharding(s interface{}) *gomock.Call {
   712  	mr.mock.ctrl.T.Helper()
   713  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetSharding", reflect.TypeOf((*MockService)(nil).SetSharding), s)
   714  }
   715  
   716  // Sharding mocks base method.
   717  func (m *MockService) Sharding() ServiceSharding {
   718  	m.ctrl.T.Helper()
   719  	ret := m.ctrl.Call(m, "Sharding")
   720  	ret0, _ := ret[0].(ServiceSharding)
   721  	return ret0
   722  }
   723  
   724  // Sharding indicates an expected call of Sharding.
   725  func (mr *MockServiceMockRecorder) Sharding() *gomock.Call {
   726  	mr.mock.ctrl.T.Helper()
   727  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Sharding", reflect.TypeOf((*MockService)(nil).Sharding))
   728  }
   729  
   730  // MockServiceReplication is a mock of ServiceReplication interface.
   731  type MockServiceReplication struct {
   732  	ctrl     *gomock.Controller
   733  	recorder *MockServiceReplicationMockRecorder
   734  }
   735  
   736  // MockServiceReplicationMockRecorder is the mock recorder for MockServiceReplication.
   737  type MockServiceReplicationMockRecorder struct {
   738  	mock *MockServiceReplication
   739  }
   740  
   741  // NewMockServiceReplication creates a new mock instance.
   742  func NewMockServiceReplication(ctrl *gomock.Controller) *MockServiceReplication {
   743  	mock := &MockServiceReplication{ctrl: ctrl}
   744  	mock.recorder = &MockServiceReplicationMockRecorder{mock}
   745  	return mock
   746  }
   747  
   748  // EXPECT returns an object that allows the caller to indicate expected use.
   749  func (m *MockServiceReplication) EXPECT() *MockServiceReplicationMockRecorder {
   750  	return m.recorder
   751  }
   752  
   753  // Replicas mocks base method.
   754  func (m *MockServiceReplication) Replicas() int {
   755  	m.ctrl.T.Helper()
   756  	ret := m.ctrl.Call(m, "Replicas")
   757  	ret0, _ := ret[0].(int)
   758  	return ret0
   759  }
   760  
   761  // Replicas indicates an expected call of Replicas.
   762  func (mr *MockServiceReplicationMockRecorder) Replicas() *gomock.Call {
   763  	mr.mock.ctrl.T.Helper()
   764  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Replicas", reflect.TypeOf((*MockServiceReplication)(nil).Replicas))
   765  }
   766  
   767  // SetReplicas mocks base method.
   768  func (m *MockServiceReplication) SetReplicas(r int) ServiceReplication {
   769  	m.ctrl.T.Helper()
   770  	ret := m.ctrl.Call(m, "SetReplicas", r)
   771  	ret0, _ := ret[0].(ServiceReplication)
   772  	return ret0
   773  }
   774  
   775  // SetReplicas indicates an expected call of SetReplicas.
   776  func (mr *MockServiceReplicationMockRecorder) SetReplicas(r interface{}) *gomock.Call {
   777  	mr.mock.ctrl.T.Helper()
   778  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetReplicas", reflect.TypeOf((*MockServiceReplication)(nil).SetReplicas), r)
   779  }
   780  
   781  // MockServiceSharding is a mock of ServiceSharding interface.
   782  type MockServiceSharding struct {
   783  	ctrl     *gomock.Controller
   784  	recorder *MockServiceShardingMockRecorder
   785  }
   786  
   787  // MockServiceShardingMockRecorder is the mock recorder for MockServiceSharding.
   788  type MockServiceShardingMockRecorder struct {
   789  	mock *MockServiceSharding
   790  }
   791  
   792  // NewMockServiceSharding creates a new mock instance.
   793  func NewMockServiceSharding(ctrl *gomock.Controller) *MockServiceSharding {
   794  	mock := &MockServiceSharding{ctrl: ctrl}
   795  	mock.recorder = &MockServiceShardingMockRecorder{mock}
   796  	return mock
   797  }
   798  
   799  // EXPECT returns an object that allows the caller to indicate expected use.
   800  func (m *MockServiceSharding) EXPECT() *MockServiceShardingMockRecorder {
   801  	return m.recorder
   802  }
   803  
   804  // IsSharded mocks base method.
   805  func (m *MockServiceSharding) IsSharded() bool {
   806  	m.ctrl.T.Helper()
   807  	ret := m.ctrl.Call(m, "IsSharded")
   808  	ret0, _ := ret[0].(bool)
   809  	return ret0
   810  }
   811  
   812  // IsSharded indicates an expected call of IsSharded.
   813  func (mr *MockServiceShardingMockRecorder) IsSharded() *gomock.Call {
   814  	mr.mock.ctrl.T.Helper()
   815  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsSharded", reflect.TypeOf((*MockServiceSharding)(nil).IsSharded))
   816  }
   817  
   818  // NumShards mocks base method.
   819  func (m *MockServiceSharding) NumShards() int {
   820  	m.ctrl.T.Helper()
   821  	ret := m.ctrl.Call(m, "NumShards")
   822  	ret0, _ := ret[0].(int)
   823  	return ret0
   824  }
   825  
   826  // NumShards indicates an expected call of NumShards.
   827  func (mr *MockServiceShardingMockRecorder) NumShards() *gomock.Call {
   828  	mr.mock.ctrl.T.Helper()
   829  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NumShards", reflect.TypeOf((*MockServiceSharding)(nil).NumShards))
   830  }
   831  
   832  // SetIsSharded mocks base method.
   833  func (m *MockServiceSharding) SetIsSharded(s bool) ServiceSharding {
   834  	m.ctrl.T.Helper()
   835  	ret := m.ctrl.Call(m, "SetIsSharded", s)
   836  	ret0, _ := ret[0].(ServiceSharding)
   837  	return ret0
   838  }
   839  
   840  // SetIsSharded indicates an expected call of SetIsSharded.
   841  func (mr *MockServiceShardingMockRecorder) SetIsSharded(s interface{}) *gomock.Call {
   842  	mr.mock.ctrl.T.Helper()
   843  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetIsSharded", reflect.TypeOf((*MockServiceSharding)(nil).SetIsSharded), s)
   844  }
   845  
   846  // SetNumShards mocks base method.
   847  func (m *MockServiceSharding) SetNumShards(n int) ServiceSharding {
   848  	m.ctrl.T.Helper()
   849  	ret := m.ctrl.Call(m, "SetNumShards", n)
   850  	ret0, _ := ret[0].(ServiceSharding)
   851  	return ret0
   852  }
   853  
   854  // SetNumShards indicates an expected call of SetNumShards.
   855  func (mr *MockServiceShardingMockRecorder) SetNumShards(n interface{}) *gomock.Call {
   856  	mr.mock.ctrl.T.Helper()
   857  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetNumShards", reflect.TypeOf((*MockServiceSharding)(nil).SetNumShards), n)
   858  }
   859  
   860  // MockServiceInstance is a mock of ServiceInstance interface.
   861  type MockServiceInstance struct {
   862  	ctrl     *gomock.Controller
   863  	recorder *MockServiceInstanceMockRecorder
   864  }
   865  
   866  // MockServiceInstanceMockRecorder is the mock recorder for MockServiceInstance.
   867  type MockServiceInstanceMockRecorder struct {
   868  	mock *MockServiceInstance
   869  }
   870  
   871  // NewMockServiceInstance creates a new mock instance.
   872  func NewMockServiceInstance(ctrl *gomock.Controller) *MockServiceInstance {
   873  	mock := &MockServiceInstance{ctrl: ctrl}
   874  	mock.recorder = &MockServiceInstanceMockRecorder{mock}
   875  	return mock
   876  }
   877  
   878  // EXPECT returns an object that allows the caller to indicate expected use.
   879  func (m *MockServiceInstance) EXPECT() *MockServiceInstanceMockRecorder {
   880  	return m.recorder
   881  }
   882  
   883  // Endpoint mocks base method.
   884  func (m *MockServiceInstance) Endpoint() string {
   885  	m.ctrl.T.Helper()
   886  	ret := m.ctrl.Call(m, "Endpoint")
   887  	ret0, _ := ret[0].(string)
   888  	return ret0
   889  }
   890  
   891  // Endpoint indicates an expected call of Endpoint.
   892  func (mr *MockServiceInstanceMockRecorder) Endpoint() *gomock.Call {
   893  	mr.mock.ctrl.T.Helper()
   894  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Endpoint", reflect.TypeOf((*MockServiceInstance)(nil).Endpoint))
   895  }
   896  
   897  // InstanceID mocks base method.
   898  func (m *MockServiceInstance) InstanceID() string {
   899  	m.ctrl.T.Helper()
   900  	ret := m.ctrl.Call(m, "InstanceID")
   901  	ret0, _ := ret[0].(string)
   902  	return ret0
   903  }
   904  
   905  // InstanceID indicates an expected call of InstanceID.
   906  func (mr *MockServiceInstanceMockRecorder) InstanceID() *gomock.Call {
   907  	mr.mock.ctrl.T.Helper()
   908  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InstanceID", reflect.TypeOf((*MockServiceInstance)(nil).InstanceID))
   909  }
   910  
   911  // ServiceID mocks base method.
   912  func (m *MockServiceInstance) ServiceID() ServiceID {
   913  	m.ctrl.T.Helper()
   914  	ret := m.ctrl.Call(m, "ServiceID")
   915  	ret0, _ := ret[0].(ServiceID)
   916  	return ret0
   917  }
   918  
   919  // ServiceID indicates an expected call of ServiceID.
   920  func (mr *MockServiceInstanceMockRecorder) ServiceID() *gomock.Call {
   921  	mr.mock.ctrl.T.Helper()
   922  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServiceID", reflect.TypeOf((*MockServiceInstance)(nil).ServiceID))
   923  }
   924  
   925  // SetEndpoint mocks base method.
   926  func (m *MockServiceInstance) SetEndpoint(e string) ServiceInstance {
   927  	m.ctrl.T.Helper()
   928  	ret := m.ctrl.Call(m, "SetEndpoint", e)
   929  	ret0, _ := ret[0].(ServiceInstance)
   930  	return ret0
   931  }
   932  
   933  // SetEndpoint indicates an expected call of SetEndpoint.
   934  func (mr *MockServiceInstanceMockRecorder) SetEndpoint(e interface{}) *gomock.Call {
   935  	mr.mock.ctrl.T.Helper()
   936  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetEndpoint", reflect.TypeOf((*MockServiceInstance)(nil).SetEndpoint), e)
   937  }
   938  
   939  // SetInstanceID mocks base method.
   940  func (m *MockServiceInstance) SetInstanceID(id string) ServiceInstance {
   941  	m.ctrl.T.Helper()
   942  	ret := m.ctrl.Call(m, "SetInstanceID", id)
   943  	ret0, _ := ret[0].(ServiceInstance)
   944  	return ret0
   945  }
   946  
   947  // SetInstanceID indicates an expected call of SetInstanceID.
   948  func (mr *MockServiceInstanceMockRecorder) SetInstanceID(id interface{}) *gomock.Call {
   949  	mr.mock.ctrl.T.Helper()
   950  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetInstanceID", reflect.TypeOf((*MockServiceInstance)(nil).SetInstanceID), id)
   951  }
   952  
   953  // SetServiceID mocks base method.
   954  func (m *MockServiceInstance) SetServiceID(service ServiceID) ServiceInstance {
   955  	m.ctrl.T.Helper()
   956  	ret := m.ctrl.Call(m, "SetServiceID", service)
   957  	ret0, _ := ret[0].(ServiceInstance)
   958  	return ret0
   959  }
   960  
   961  // SetServiceID indicates an expected call of SetServiceID.
   962  func (mr *MockServiceInstanceMockRecorder) SetServiceID(service interface{}) *gomock.Call {
   963  	mr.mock.ctrl.T.Helper()
   964  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetServiceID", reflect.TypeOf((*MockServiceInstance)(nil).SetServiceID), service)
   965  }
   966  
   967  // SetShards mocks base method.
   968  func (m *MockServiceInstance) SetShards(s shard.Shards) ServiceInstance {
   969  	m.ctrl.T.Helper()
   970  	ret := m.ctrl.Call(m, "SetShards", s)
   971  	ret0, _ := ret[0].(ServiceInstance)
   972  	return ret0
   973  }
   974  
   975  // SetShards indicates an expected call of SetShards.
   976  func (mr *MockServiceInstanceMockRecorder) SetShards(s interface{}) *gomock.Call {
   977  	mr.mock.ctrl.T.Helper()
   978  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetShards", reflect.TypeOf((*MockServiceInstance)(nil).SetShards), s)
   979  }
   980  
   981  // Shards mocks base method.
   982  func (m *MockServiceInstance) Shards() shard.Shards {
   983  	m.ctrl.T.Helper()
   984  	ret := m.ctrl.Call(m, "Shards")
   985  	ret0, _ := ret[0].(shard.Shards)
   986  	return ret0
   987  }
   988  
   989  // Shards indicates an expected call of Shards.
   990  func (mr *MockServiceInstanceMockRecorder) Shards() *gomock.Call {
   991  	mr.mock.ctrl.T.Helper()
   992  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Shards", reflect.TypeOf((*MockServiceInstance)(nil).Shards))
   993  }
   994  
   995  // MockAdvertisement is a mock of Advertisement interface.
   996  type MockAdvertisement struct {
   997  	ctrl     *gomock.Controller
   998  	recorder *MockAdvertisementMockRecorder
   999  }
  1000  
  1001  // MockAdvertisementMockRecorder is the mock recorder for MockAdvertisement.
  1002  type MockAdvertisementMockRecorder struct {
  1003  	mock *MockAdvertisement
  1004  }
  1005  
  1006  // NewMockAdvertisement creates a new mock instance.
  1007  func NewMockAdvertisement(ctrl *gomock.Controller) *MockAdvertisement {
  1008  	mock := &MockAdvertisement{ctrl: ctrl}
  1009  	mock.recorder = &MockAdvertisementMockRecorder{mock}
  1010  	return mock
  1011  }
  1012  
  1013  // EXPECT returns an object that allows the caller to indicate expected use.
  1014  func (m *MockAdvertisement) EXPECT() *MockAdvertisementMockRecorder {
  1015  	return m.recorder
  1016  }
  1017  
  1018  // Health mocks base method.
  1019  func (m *MockAdvertisement) Health() func() error {
  1020  	m.ctrl.T.Helper()
  1021  	ret := m.ctrl.Call(m, "Health")
  1022  	ret0, _ := ret[0].(func() error)
  1023  	return ret0
  1024  }
  1025  
  1026  // Health indicates an expected call of Health.
  1027  func (mr *MockAdvertisementMockRecorder) Health() *gomock.Call {
  1028  	mr.mock.ctrl.T.Helper()
  1029  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Health", reflect.TypeOf((*MockAdvertisement)(nil).Health))
  1030  }
  1031  
  1032  // PlacementInstance mocks base method.
  1033  func (m *MockAdvertisement) PlacementInstance() placement.Instance {
  1034  	m.ctrl.T.Helper()
  1035  	ret := m.ctrl.Call(m, "PlacementInstance")
  1036  	ret0, _ := ret[0].(placement.Instance)
  1037  	return ret0
  1038  }
  1039  
  1040  // PlacementInstance indicates an expected call of PlacementInstance.
  1041  func (mr *MockAdvertisementMockRecorder) PlacementInstance() *gomock.Call {
  1042  	mr.mock.ctrl.T.Helper()
  1043  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PlacementInstance", reflect.TypeOf((*MockAdvertisement)(nil).PlacementInstance))
  1044  }
  1045  
  1046  // ServiceID mocks base method.
  1047  func (m *MockAdvertisement) ServiceID() ServiceID {
  1048  	m.ctrl.T.Helper()
  1049  	ret := m.ctrl.Call(m, "ServiceID")
  1050  	ret0, _ := ret[0].(ServiceID)
  1051  	return ret0
  1052  }
  1053  
  1054  // ServiceID indicates an expected call of ServiceID.
  1055  func (mr *MockAdvertisementMockRecorder) ServiceID() *gomock.Call {
  1056  	mr.mock.ctrl.T.Helper()
  1057  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServiceID", reflect.TypeOf((*MockAdvertisement)(nil).ServiceID))
  1058  }
  1059  
  1060  // SetHealth mocks base method.
  1061  func (m *MockAdvertisement) SetHealth(health func() error) Advertisement {
  1062  	m.ctrl.T.Helper()
  1063  	ret := m.ctrl.Call(m, "SetHealth", health)
  1064  	ret0, _ := ret[0].(Advertisement)
  1065  	return ret0
  1066  }
  1067  
  1068  // SetHealth indicates an expected call of SetHealth.
  1069  func (mr *MockAdvertisementMockRecorder) SetHealth(health interface{}) *gomock.Call {
  1070  	mr.mock.ctrl.T.Helper()
  1071  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetHealth", reflect.TypeOf((*MockAdvertisement)(nil).SetHealth), health)
  1072  }
  1073  
  1074  // SetPlacementInstance mocks base method.
  1075  func (m *MockAdvertisement) SetPlacementInstance(p placement.Instance) Advertisement {
  1076  	m.ctrl.T.Helper()
  1077  	ret := m.ctrl.Call(m, "SetPlacementInstance", p)
  1078  	ret0, _ := ret[0].(Advertisement)
  1079  	return ret0
  1080  }
  1081  
  1082  // SetPlacementInstance indicates an expected call of SetPlacementInstance.
  1083  func (mr *MockAdvertisementMockRecorder) SetPlacementInstance(p interface{}) *gomock.Call {
  1084  	mr.mock.ctrl.T.Helper()
  1085  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetPlacementInstance", reflect.TypeOf((*MockAdvertisement)(nil).SetPlacementInstance), p)
  1086  }
  1087  
  1088  // SetServiceID mocks base method.
  1089  func (m *MockAdvertisement) SetServiceID(service ServiceID) Advertisement {
  1090  	m.ctrl.T.Helper()
  1091  	ret := m.ctrl.Call(m, "SetServiceID", service)
  1092  	ret0, _ := ret[0].(Advertisement)
  1093  	return ret0
  1094  }
  1095  
  1096  // SetServiceID indicates an expected call of SetServiceID.
  1097  func (mr *MockAdvertisementMockRecorder) SetServiceID(service interface{}) *gomock.Call {
  1098  	mr.mock.ctrl.T.Helper()
  1099  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetServiceID", reflect.TypeOf((*MockAdvertisement)(nil).SetServiceID), service)
  1100  }
  1101  
  1102  // MockServiceID is a mock of ServiceID interface.
  1103  type MockServiceID struct {
  1104  	ctrl     *gomock.Controller
  1105  	recorder *MockServiceIDMockRecorder
  1106  }
  1107  
  1108  // MockServiceIDMockRecorder is the mock recorder for MockServiceID.
  1109  type MockServiceIDMockRecorder struct {
  1110  	mock *MockServiceID
  1111  }
  1112  
  1113  // NewMockServiceID creates a new mock instance.
  1114  func NewMockServiceID(ctrl *gomock.Controller) *MockServiceID {
  1115  	mock := &MockServiceID{ctrl: ctrl}
  1116  	mock.recorder = &MockServiceIDMockRecorder{mock}
  1117  	return mock
  1118  }
  1119  
  1120  // EXPECT returns an object that allows the caller to indicate expected use.
  1121  func (m *MockServiceID) EXPECT() *MockServiceIDMockRecorder {
  1122  	return m.recorder
  1123  }
  1124  
  1125  // Environment mocks base method.
  1126  func (m *MockServiceID) Environment() string {
  1127  	m.ctrl.T.Helper()
  1128  	ret := m.ctrl.Call(m, "Environment")
  1129  	ret0, _ := ret[0].(string)
  1130  	return ret0
  1131  }
  1132  
  1133  // Environment indicates an expected call of Environment.
  1134  func (mr *MockServiceIDMockRecorder) Environment() *gomock.Call {
  1135  	mr.mock.ctrl.T.Helper()
  1136  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Environment", reflect.TypeOf((*MockServiceID)(nil).Environment))
  1137  }
  1138  
  1139  // Equal mocks base method.
  1140  func (m *MockServiceID) Equal(value ServiceID) bool {
  1141  	m.ctrl.T.Helper()
  1142  	ret := m.ctrl.Call(m, "Equal", value)
  1143  	ret0, _ := ret[0].(bool)
  1144  	return ret0
  1145  }
  1146  
  1147  // Equal indicates an expected call of Equal.
  1148  func (mr *MockServiceIDMockRecorder) Equal(value interface{}) *gomock.Call {
  1149  	mr.mock.ctrl.T.Helper()
  1150  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Equal", reflect.TypeOf((*MockServiceID)(nil).Equal), value)
  1151  }
  1152  
  1153  // Name mocks base method.
  1154  func (m *MockServiceID) Name() string {
  1155  	m.ctrl.T.Helper()
  1156  	ret := m.ctrl.Call(m, "Name")
  1157  	ret0, _ := ret[0].(string)
  1158  	return ret0
  1159  }
  1160  
  1161  // Name indicates an expected call of Name.
  1162  func (mr *MockServiceIDMockRecorder) Name() *gomock.Call {
  1163  	mr.mock.ctrl.T.Helper()
  1164  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Name", reflect.TypeOf((*MockServiceID)(nil).Name))
  1165  }
  1166  
  1167  // SetEnvironment mocks base method.
  1168  func (m *MockServiceID) SetEnvironment(env string) ServiceID {
  1169  	m.ctrl.T.Helper()
  1170  	ret := m.ctrl.Call(m, "SetEnvironment", env)
  1171  	ret0, _ := ret[0].(ServiceID)
  1172  	return ret0
  1173  }
  1174  
  1175  // SetEnvironment indicates an expected call of SetEnvironment.
  1176  func (mr *MockServiceIDMockRecorder) SetEnvironment(env interface{}) *gomock.Call {
  1177  	mr.mock.ctrl.T.Helper()
  1178  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetEnvironment", reflect.TypeOf((*MockServiceID)(nil).SetEnvironment), env)
  1179  }
  1180  
  1181  // SetName mocks base method.
  1182  func (m *MockServiceID) SetName(s string) ServiceID {
  1183  	m.ctrl.T.Helper()
  1184  	ret := m.ctrl.Call(m, "SetName", s)
  1185  	ret0, _ := ret[0].(ServiceID)
  1186  	return ret0
  1187  }
  1188  
  1189  // SetName indicates an expected call of SetName.
  1190  func (mr *MockServiceIDMockRecorder) SetName(s interface{}) *gomock.Call {
  1191  	mr.mock.ctrl.T.Helper()
  1192  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetName", reflect.TypeOf((*MockServiceID)(nil).SetName), s)
  1193  }
  1194  
  1195  // SetZone mocks base method.
  1196  func (m *MockServiceID) SetZone(zone string) ServiceID {
  1197  	m.ctrl.T.Helper()
  1198  	ret := m.ctrl.Call(m, "SetZone", zone)
  1199  	ret0, _ := ret[0].(ServiceID)
  1200  	return ret0
  1201  }
  1202  
  1203  // SetZone indicates an expected call of SetZone.
  1204  func (mr *MockServiceIDMockRecorder) SetZone(zone interface{}) *gomock.Call {
  1205  	mr.mock.ctrl.T.Helper()
  1206  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetZone", reflect.TypeOf((*MockServiceID)(nil).SetZone), zone)
  1207  }
  1208  
  1209  // String mocks base method.
  1210  func (m *MockServiceID) String() string {
  1211  	m.ctrl.T.Helper()
  1212  	ret := m.ctrl.Call(m, "String")
  1213  	ret0, _ := ret[0].(string)
  1214  	return ret0
  1215  }
  1216  
  1217  // String indicates an expected call of String.
  1218  func (mr *MockServiceIDMockRecorder) String() *gomock.Call {
  1219  	mr.mock.ctrl.T.Helper()
  1220  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "String", reflect.TypeOf((*MockServiceID)(nil).String))
  1221  }
  1222  
  1223  // Zone mocks base method.
  1224  func (m *MockServiceID) Zone() string {
  1225  	m.ctrl.T.Helper()
  1226  	ret := m.ctrl.Call(m, "Zone")
  1227  	ret0, _ := ret[0].(string)
  1228  	return ret0
  1229  }
  1230  
  1231  // Zone indicates an expected call of Zone.
  1232  func (mr *MockServiceIDMockRecorder) Zone() *gomock.Call {
  1233  	mr.mock.ctrl.T.Helper()
  1234  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Zone", reflect.TypeOf((*MockServiceID)(nil).Zone))
  1235  }
  1236  
  1237  // MockQueryOptions is a mock of QueryOptions interface.
  1238  type MockQueryOptions struct {
  1239  	ctrl     *gomock.Controller
  1240  	recorder *MockQueryOptionsMockRecorder
  1241  }
  1242  
  1243  // MockQueryOptionsMockRecorder is the mock recorder for MockQueryOptions.
  1244  type MockQueryOptionsMockRecorder struct {
  1245  	mock *MockQueryOptions
  1246  }
  1247  
  1248  // NewMockQueryOptions creates a new mock instance.
  1249  func NewMockQueryOptions(ctrl *gomock.Controller) *MockQueryOptions {
  1250  	mock := &MockQueryOptions{ctrl: ctrl}
  1251  	mock.recorder = &MockQueryOptionsMockRecorder{mock}
  1252  	return mock
  1253  }
  1254  
  1255  // EXPECT returns an object that allows the caller to indicate expected use.
  1256  func (m *MockQueryOptions) EXPECT() *MockQueryOptionsMockRecorder {
  1257  	return m.recorder
  1258  }
  1259  
  1260  // IncludeUnhealthy mocks base method.
  1261  func (m *MockQueryOptions) IncludeUnhealthy() bool {
  1262  	m.ctrl.T.Helper()
  1263  	ret := m.ctrl.Call(m, "IncludeUnhealthy")
  1264  	ret0, _ := ret[0].(bool)
  1265  	return ret0
  1266  }
  1267  
  1268  // IncludeUnhealthy indicates an expected call of IncludeUnhealthy.
  1269  func (mr *MockQueryOptionsMockRecorder) IncludeUnhealthy() *gomock.Call {
  1270  	mr.mock.ctrl.T.Helper()
  1271  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IncludeUnhealthy", reflect.TypeOf((*MockQueryOptions)(nil).IncludeUnhealthy))
  1272  }
  1273  
  1274  // InterruptedCh mocks base method.
  1275  func (m *MockQueryOptions) InterruptedCh() <-chan struct{} {
  1276  	m.ctrl.T.Helper()
  1277  	ret := m.ctrl.Call(m, "InterruptedCh")
  1278  	ret0, _ := ret[0].(<-chan struct{})
  1279  	return ret0
  1280  }
  1281  
  1282  // InterruptedCh indicates an expected call of InterruptedCh.
  1283  func (mr *MockQueryOptionsMockRecorder) InterruptedCh() *gomock.Call {
  1284  	mr.mock.ctrl.T.Helper()
  1285  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InterruptedCh", reflect.TypeOf((*MockQueryOptions)(nil).InterruptedCh))
  1286  }
  1287  
  1288  // SetIncludeUnhealthy mocks base method.
  1289  func (m *MockQueryOptions) SetIncludeUnhealthy(h bool) QueryOptions {
  1290  	m.ctrl.T.Helper()
  1291  	ret := m.ctrl.Call(m, "SetIncludeUnhealthy", h)
  1292  	ret0, _ := ret[0].(QueryOptions)
  1293  	return ret0
  1294  }
  1295  
  1296  // SetIncludeUnhealthy indicates an expected call of SetIncludeUnhealthy.
  1297  func (mr *MockQueryOptionsMockRecorder) SetIncludeUnhealthy(h interface{}) *gomock.Call {
  1298  	mr.mock.ctrl.T.Helper()
  1299  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetIncludeUnhealthy", reflect.TypeOf((*MockQueryOptions)(nil).SetIncludeUnhealthy), h)
  1300  }
  1301  
  1302  // SetInterruptedCh mocks base method.
  1303  func (m *MockQueryOptions) SetInterruptedCh(value <-chan struct{}) QueryOptions {
  1304  	m.ctrl.T.Helper()
  1305  	ret := m.ctrl.Call(m, "SetInterruptedCh", value)
  1306  	ret0, _ := ret[0].(QueryOptions)
  1307  	return ret0
  1308  }
  1309  
  1310  // SetInterruptedCh indicates an expected call of SetInterruptedCh.
  1311  func (mr *MockQueryOptionsMockRecorder) SetInterruptedCh(value interface{}) *gomock.Call {
  1312  	mr.mock.ctrl.T.Helper()
  1313  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetInterruptedCh", reflect.TypeOf((*MockQueryOptions)(nil).SetInterruptedCh), value)
  1314  }
  1315  
  1316  // MockMetadata is a mock of Metadata interface.
  1317  type MockMetadata struct {
  1318  	ctrl     *gomock.Controller
  1319  	recorder *MockMetadataMockRecorder
  1320  }
  1321  
  1322  // MockMetadataMockRecorder is the mock recorder for MockMetadata.
  1323  type MockMetadataMockRecorder struct {
  1324  	mock *MockMetadata
  1325  }
  1326  
  1327  // NewMockMetadata creates a new mock instance.
  1328  func NewMockMetadata(ctrl *gomock.Controller) *MockMetadata {
  1329  	mock := &MockMetadata{ctrl: ctrl}
  1330  	mock.recorder = &MockMetadataMockRecorder{mock}
  1331  	return mock
  1332  }
  1333  
  1334  // EXPECT returns an object that allows the caller to indicate expected use.
  1335  func (m *MockMetadata) EXPECT() *MockMetadataMockRecorder {
  1336  	return m.recorder
  1337  }
  1338  
  1339  // HeartbeatInterval mocks base method.
  1340  func (m *MockMetadata) HeartbeatInterval() time.Duration {
  1341  	m.ctrl.T.Helper()
  1342  	ret := m.ctrl.Call(m, "HeartbeatInterval")
  1343  	ret0, _ := ret[0].(time.Duration)
  1344  	return ret0
  1345  }
  1346  
  1347  // HeartbeatInterval indicates an expected call of HeartbeatInterval.
  1348  func (mr *MockMetadataMockRecorder) HeartbeatInterval() *gomock.Call {
  1349  	mr.mock.ctrl.T.Helper()
  1350  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HeartbeatInterval", reflect.TypeOf((*MockMetadata)(nil).HeartbeatInterval))
  1351  }
  1352  
  1353  // LivenessInterval mocks base method.
  1354  func (m *MockMetadata) LivenessInterval() time.Duration {
  1355  	m.ctrl.T.Helper()
  1356  	ret := m.ctrl.Call(m, "LivenessInterval")
  1357  	ret0, _ := ret[0].(time.Duration)
  1358  	return ret0
  1359  }
  1360  
  1361  // LivenessInterval indicates an expected call of LivenessInterval.
  1362  func (mr *MockMetadataMockRecorder) LivenessInterval() *gomock.Call {
  1363  	mr.mock.ctrl.T.Helper()
  1364  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LivenessInterval", reflect.TypeOf((*MockMetadata)(nil).LivenessInterval))
  1365  }
  1366  
  1367  // Port mocks base method.
  1368  func (m *MockMetadata) Port() uint32 {
  1369  	m.ctrl.T.Helper()
  1370  	ret := m.ctrl.Call(m, "Port")
  1371  	ret0, _ := ret[0].(uint32)
  1372  	return ret0
  1373  }
  1374  
  1375  // Port indicates an expected call of Port.
  1376  func (mr *MockMetadataMockRecorder) Port() *gomock.Call {
  1377  	mr.mock.ctrl.T.Helper()
  1378  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Port", reflect.TypeOf((*MockMetadata)(nil).Port))
  1379  }
  1380  
  1381  // Proto mocks base method.
  1382  func (m *MockMetadata) Proto() (*metadatapb.Metadata, error) {
  1383  	m.ctrl.T.Helper()
  1384  	ret := m.ctrl.Call(m, "Proto")
  1385  	ret0, _ := ret[0].(*metadatapb.Metadata)
  1386  	ret1, _ := ret[1].(error)
  1387  	return ret0, ret1
  1388  }
  1389  
  1390  // Proto indicates an expected call of Proto.
  1391  func (mr *MockMetadataMockRecorder) Proto() *gomock.Call {
  1392  	mr.mock.ctrl.T.Helper()
  1393  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Proto", reflect.TypeOf((*MockMetadata)(nil).Proto))
  1394  }
  1395  
  1396  // SetHeartbeatInterval mocks base method.
  1397  func (m *MockMetadata) SetHeartbeatInterval(h time.Duration) Metadata {
  1398  	m.ctrl.T.Helper()
  1399  	ret := m.ctrl.Call(m, "SetHeartbeatInterval", h)
  1400  	ret0, _ := ret[0].(Metadata)
  1401  	return ret0
  1402  }
  1403  
  1404  // SetHeartbeatInterval indicates an expected call of SetHeartbeatInterval.
  1405  func (mr *MockMetadataMockRecorder) SetHeartbeatInterval(h interface{}) *gomock.Call {
  1406  	mr.mock.ctrl.T.Helper()
  1407  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetHeartbeatInterval", reflect.TypeOf((*MockMetadata)(nil).SetHeartbeatInterval), h)
  1408  }
  1409  
  1410  // SetLivenessInterval mocks base method.
  1411  func (m *MockMetadata) SetLivenessInterval(l time.Duration) Metadata {
  1412  	m.ctrl.T.Helper()
  1413  	ret := m.ctrl.Call(m, "SetLivenessInterval", l)
  1414  	ret0, _ := ret[0].(Metadata)
  1415  	return ret0
  1416  }
  1417  
  1418  // SetLivenessInterval indicates an expected call of SetLivenessInterval.
  1419  func (mr *MockMetadataMockRecorder) SetLivenessInterval(l interface{}) *gomock.Call {
  1420  	mr.mock.ctrl.T.Helper()
  1421  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetLivenessInterval", reflect.TypeOf((*MockMetadata)(nil).SetLivenessInterval), l)
  1422  }
  1423  
  1424  // SetPort mocks base method.
  1425  func (m *MockMetadata) SetPort(p uint32) Metadata {
  1426  	m.ctrl.T.Helper()
  1427  	ret := m.ctrl.Call(m, "SetPort", p)
  1428  	ret0, _ := ret[0].(Metadata)
  1429  	return ret0
  1430  }
  1431  
  1432  // SetPort indicates an expected call of SetPort.
  1433  func (mr *MockMetadataMockRecorder) SetPort(p interface{}) *gomock.Call {
  1434  	mr.mock.ctrl.T.Helper()
  1435  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetPort", reflect.TypeOf((*MockMetadata)(nil).SetPort), p)
  1436  }
  1437  
  1438  // String mocks base method.
  1439  func (m *MockMetadata) String() string {
  1440  	m.ctrl.T.Helper()
  1441  	ret := m.ctrl.Call(m, "String")
  1442  	ret0, _ := ret[0].(string)
  1443  	return ret0
  1444  }
  1445  
  1446  // String indicates an expected call of String.
  1447  func (mr *MockMetadataMockRecorder) String() *gomock.Call {
  1448  	mr.mock.ctrl.T.Helper()
  1449  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "String", reflect.TypeOf((*MockMetadata)(nil).String))
  1450  }
  1451  
  1452  // MockHeartbeatService is a mock of HeartbeatService interface.
  1453  type MockHeartbeatService struct {
  1454  	ctrl     *gomock.Controller
  1455  	recorder *MockHeartbeatServiceMockRecorder
  1456  }
  1457  
  1458  // MockHeartbeatServiceMockRecorder is the mock recorder for MockHeartbeatService.
  1459  type MockHeartbeatServiceMockRecorder struct {
  1460  	mock *MockHeartbeatService
  1461  }
  1462  
  1463  // NewMockHeartbeatService creates a new mock instance.
  1464  func NewMockHeartbeatService(ctrl *gomock.Controller) *MockHeartbeatService {
  1465  	mock := &MockHeartbeatService{ctrl: ctrl}
  1466  	mock.recorder = &MockHeartbeatServiceMockRecorder{mock}
  1467  	return mock
  1468  }
  1469  
  1470  // EXPECT returns an object that allows the caller to indicate expected use.
  1471  func (m *MockHeartbeatService) EXPECT() *MockHeartbeatServiceMockRecorder {
  1472  	return m.recorder
  1473  }
  1474  
  1475  // Delete mocks base method.
  1476  func (m *MockHeartbeatService) Delete(instance string) error {
  1477  	m.ctrl.T.Helper()
  1478  	ret := m.ctrl.Call(m, "Delete", instance)
  1479  	ret0, _ := ret[0].(error)
  1480  	return ret0
  1481  }
  1482  
  1483  // Delete indicates an expected call of Delete.
  1484  func (mr *MockHeartbeatServiceMockRecorder) Delete(instance interface{}) *gomock.Call {
  1485  	mr.mock.ctrl.T.Helper()
  1486  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockHeartbeatService)(nil).Delete), instance)
  1487  }
  1488  
  1489  // Get mocks base method.
  1490  func (m *MockHeartbeatService) Get() ([]string, error) {
  1491  	m.ctrl.T.Helper()
  1492  	ret := m.ctrl.Call(m, "Get")
  1493  	ret0, _ := ret[0].([]string)
  1494  	ret1, _ := ret[1].(error)
  1495  	return ret0, ret1
  1496  }
  1497  
  1498  // Get indicates an expected call of Get.
  1499  func (mr *MockHeartbeatServiceMockRecorder) Get() *gomock.Call {
  1500  	mr.mock.ctrl.T.Helper()
  1501  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockHeartbeatService)(nil).Get))
  1502  }
  1503  
  1504  // GetInstances mocks base method.
  1505  func (m *MockHeartbeatService) GetInstances() ([]placement.Instance, error) {
  1506  	m.ctrl.T.Helper()
  1507  	ret := m.ctrl.Call(m, "GetInstances")
  1508  	ret0, _ := ret[0].([]placement.Instance)
  1509  	ret1, _ := ret[1].(error)
  1510  	return ret0, ret1
  1511  }
  1512  
  1513  // GetInstances indicates an expected call of GetInstances.
  1514  func (mr *MockHeartbeatServiceMockRecorder) GetInstances() *gomock.Call {
  1515  	mr.mock.ctrl.T.Helper()
  1516  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetInstances", reflect.TypeOf((*MockHeartbeatService)(nil).GetInstances))
  1517  }
  1518  
  1519  // Heartbeat mocks base method.
  1520  func (m *MockHeartbeatService) Heartbeat(instance placement.Instance, ttl time.Duration) error {
  1521  	m.ctrl.T.Helper()
  1522  	ret := m.ctrl.Call(m, "Heartbeat", instance, ttl)
  1523  	ret0, _ := ret[0].(error)
  1524  	return ret0
  1525  }
  1526  
  1527  // Heartbeat indicates an expected call of Heartbeat.
  1528  func (mr *MockHeartbeatServiceMockRecorder) Heartbeat(instance, ttl interface{}) *gomock.Call {
  1529  	mr.mock.ctrl.T.Helper()
  1530  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Heartbeat", reflect.TypeOf((*MockHeartbeatService)(nil).Heartbeat), instance, ttl)
  1531  }
  1532  
  1533  // Watch mocks base method.
  1534  func (m *MockHeartbeatService) Watch() (watch.Watch, error) {
  1535  	m.ctrl.T.Helper()
  1536  	ret := m.ctrl.Call(m, "Watch")
  1537  	ret0, _ := ret[0].(watch.Watch)
  1538  	ret1, _ := ret[1].(error)
  1539  	return ret0, ret1
  1540  }
  1541  
  1542  // Watch indicates an expected call of Watch.
  1543  func (mr *MockHeartbeatServiceMockRecorder) Watch() *gomock.Call {
  1544  	mr.mock.ctrl.T.Helper()
  1545  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Watch", reflect.TypeOf((*MockHeartbeatService)(nil).Watch))
  1546  }
  1547  
  1548  // MockElectionOptions is a mock of ElectionOptions interface.
  1549  type MockElectionOptions struct {
  1550  	ctrl     *gomock.Controller
  1551  	recorder *MockElectionOptionsMockRecorder
  1552  }
  1553  
  1554  // MockElectionOptionsMockRecorder is the mock recorder for MockElectionOptions.
  1555  type MockElectionOptionsMockRecorder struct {
  1556  	mock *MockElectionOptions
  1557  }
  1558  
  1559  // NewMockElectionOptions creates a new mock instance.
  1560  func NewMockElectionOptions(ctrl *gomock.Controller) *MockElectionOptions {
  1561  	mock := &MockElectionOptions{ctrl: ctrl}
  1562  	mock.recorder = &MockElectionOptionsMockRecorder{mock}
  1563  	return mock
  1564  }
  1565  
  1566  // EXPECT returns an object that allows the caller to indicate expected use.
  1567  func (m *MockElectionOptions) EXPECT() *MockElectionOptionsMockRecorder {
  1568  	return m.recorder
  1569  }
  1570  
  1571  // LeaderTimeout mocks base method.
  1572  func (m *MockElectionOptions) LeaderTimeout() time.Duration {
  1573  	m.ctrl.T.Helper()
  1574  	ret := m.ctrl.Call(m, "LeaderTimeout")
  1575  	ret0, _ := ret[0].(time.Duration)
  1576  	return ret0
  1577  }
  1578  
  1579  // LeaderTimeout indicates an expected call of LeaderTimeout.
  1580  func (mr *MockElectionOptionsMockRecorder) LeaderTimeout() *gomock.Call {
  1581  	mr.mock.ctrl.T.Helper()
  1582  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LeaderTimeout", reflect.TypeOf((*MockElectionOptions)(nil).LeaderTimeout))
  1583  }
  1584  
  1585  // ResignTimeout mocks base method.
  1586  func (m *MockElectionOptions) ResignTimeout() time.Duration {
  1587  	m.ctrl.T.Helper()
  1588  	ret := m.ctrl.Call(m, "ResignTimeout")
  1589  	ret0, _ := ret[0].(time.Duration)
  1590  	return ret0
  1591  }
  1592  
  1593  // ResignTimeout indicates an expected call of ResignTimeout.
  1594  func (mr *MockElectionOptionsMockRecorder) ResignTimeout() *gomock.Call {
  1595  	mr.mock.ctrl.T.Helper()
  1596  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ResignTimeout", reflect.TypeOf((*MockElectionOptions)(nil).ResignTimeout))
  1597  }
  1598  
  1599  // SetLeaderTimeout mocks base method.
  1600  func (m *MockElectionOptions) SetLeaderTimeout(t time.Duration) ElectionOptions {
  1601  	m.ctrl.T.Helper()
  1602  	ret := m.ctrl.Call(m, "SetLeaderTimeout", t)
  1603  	ret0, _ := ret[0].(ElectionOptions)
  1604  	return ret0
  1605  }
  1606  
  1607  // SetLeaderTimeout indicates an expected call of SetLeaderTimeout.
  1608  func (mr *MockElectionOptionsMockRecorder) SetLeaderTimeout(t interface{}) *gomock.Call {
  1609  	mr.mock.ctrl.T.Helper()
  1610  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetLeaderTimeout", reflect.TypeOf((*MockElectionOptions)(nil).SetLeaderTimeout), t)
  1611  }
  1612  
  1613  // SetResignTimeout mocks base method.
  1614  func (m *MockElectionOptions) SetResignTimeout(t time.Duration) ElectionOptions {
  1615  	m.ctrl.T.Helper()
  1616  	ret := m.ctrl.Call(m, "SetResignTimeout", t)
  1617  	ret0, _ := ret[0].(ElectionOptions)
  1618  	return ret0
  1619  }
  1620  
  1621  // SetResignTimeout indicates an expected call of SetResignTimeout.
  1622  func (mr *MockElectionOptionsMockRecorder) SetResignTimeout(t interface{}) *gomock.Call {
  1623  	mr.mock.ctrl.T.Helper()
  1624  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetResignTimeout", reflect.TypeOf((*MockElectionOptions)(nil).SetResignTimeout), t)
  1625  }
  1626  
  1627  // SetTTLSecs mocks base method.
  1628  func (m *MockElectionOptions) SetTTLSecs(ttl int) ElectionOptions {
  1629  	m.ctrl.T.Helper()
  1630  	ret := m.ctrl.Call(m, "SetTTLSecs", ttl)
  1631  	ret0, _ := ret[0].(ElectionOptions)
  1632  	return ret0
  1633  }
  1634  
  1635  // SetTTLSecs indicates an expected call of SetTTLSecs.
  1636  func (mr *MockElectionOptionsMockRecorder) SetTTLSecs(ttl interface{}) *gomock.Call {
  1637  	mr.mock.ctrl.T.Helper()
  1638  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetTTLSecs", reflect.TypeOf((*MockElectionOptions)(nil).SetTTLSecs), ttl)
  1639  }
  1640  
  1641  // TTLSecs mocks base method.
  1642  func (m *MockElectionOptions) TTLSecs() int {
  1643  	m.ctrl.T.Helper()
  1644  	ret := m.ctrl.Call(m, "TTLSecs")
  1645  	ret0, _ := ret[0].(int)
  1646  	return ret0
  1647  }
  1648  
  1649  // TTLSecs indicates an expected call of TTLSecs.
  1650  func (mr *MockElectionOptionsMockRecorder) TTLSecs() *gomock.Call {
  1651  	mr.mock.ctrl.T.Helper()
  1652  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TTLSecs", reflect.TypeOf((*MockElectionOptions)(nil).TTLSecs))
  1653  }
  1654  
  1655  // MockCampaignOptions is a mock of CampaignOptions interface.
  1656  type MockCampaignOptions struct {
  1657  	ctrl     *gomock.Controller
  1658  	recorder *MockCampaignOptionsMockRecorder
  1659  }
  1660  
  1661  // MockCampaignOptionsMockRecorder is the mock recorder for MockCampaignOptions.
  1662  type MockCampaignOptionsMockRecorder struct {
  1663  	mock *MockCampaignOptions
  1664  }
  1665  
  1666  // NewMockCampaignOptions creates a new mock instance.
  1667  func NewMockCampaignOptions(ctrl *gomock.Controller) *MockCampaignOptions {
  1668  	mock := &MockCampaignOptions{ctrl: ctrl}
  1669  	mock.recorder = &MockCampaignOptionsMockRecorder{mock}
  1670  	return mock
  1671  }
  1672  
  1673  // EXPECT returns an object that allows the caller to indicate expected use.
  1674  func (m *MockCampaignOptions) EXPECT() *MockCampaignOptionsMockRecorder {
  1675  	return m.recorder
  1676  }
  1677  
  1678  // LeaderValue mocks base method.
  1679  func (m *MockCampaignOptions) LeaderValue() string {
  1680  	m.ctrl.T.Helper()
  1681  	ret := m.ctrl.Call(m, "LeaderValue")
  1682  	ret0, _ := ret[0].(string)
  1683  	return ret0
  1684  }
  1685  
  1686  // LeaderValue indicates an expected call of LeaderValue.
  1687  func (mr *MockCampaignOptionsMockRecorder) LeaderValue() *gomock.Call {
  1688  	mr.mock.ctrl.T.Helper()
  1689  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LeaderValue", reflect.TypeOf((*MockCampaignOptions)(nil).LeaderValue))
  1690  }
  1691  
  1692  // SetLeaderValue mocks base method.
  1693  func (m *MockCampaignOptions) SetLeaderValue(v string) CampaignOptions {
  1694  	m.ctrl.T.Helper()
  1695  	ret := m.ctrl.Call(m, "SetLeaderValue", v)
  1696  	ret0, _ := ret[0].(CampaignOptions)
  1697  	return ret0
  1698  }
  1699  
  1700  // SetLeaderValue indicates an expected call of SetLeaderValue.
  1701  func (mr *MockCampaignOptionsMockRecorder) SetLeaderValue(v interface{}) *gomock.Call {
  1702  	mr.mock.ctrl.T.Helper()
  1703  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetLeaderValue", reflect.TypeOf((*MockCampaignOptions)(nil).SetLeaderValue), v)
  1704  }
  1705  
  1706  // MockLeaderService is a mock of LeaderService interface.
  1707  type MockLeaderService struct {
  1708  	ctrl     *gomock.Controller
  1709  	recorder *MockLeaderServiceMockRecorder
  1710  }
  1711  
  1712  // MockLeaderServiceMockRecorder is the mock recorder for MockLeaderService.
  1713  type MockLeaderServiceMockRecorder struct {
  1714  	mock *MockLeaderService
  1715  }
  1716  
  1717  // NewMockLeaderService creates a new mock instance.
  1718  func NewMockLeaderService(ctrl *gomock.Controller) *MockLeaderService {
  1719  	mock := &MockLeaderService{ctrl: ctrl}
  1720  	mock.recorder = &MockLeaderServiceMockRecorder{mock}
  1721  	return mock
  1722  }
  1723  
  1724  // EXPECT returns an object that allows the caller to indicate expected use.
  1725  func (m *MockLeaderService) EXPECT() *MockLeaderServiceMockRecorder {
  1726  	return m.recorder
  1727  }
  1728  
  1729  // Campaign mocks base method.
  1730  func (m *MockLeaderService) Campaign(electionID string, opts CampaignOptions) (<-chan campaign.Status, error) {
  1731  	m.ctrl.T.Helper()
  1732  	ret := m.ctrl.Call(m, "Campaign", electionID, opts)
  1733  	ret0, _ := ret[0].(<-chan campaign.Status)
  1734  	ret1, _ := ret[1].(error)
  1735  	return ret0, ret1
  1736  }
  1737  
  1738  // Campaign indicates an expected call of Campaign.
  1739  func (mr *MockLeaderServiceMockRecorder) Campaign(electionID, opts interface{}) *gomock.Call {
  1740  	mr.mock.ctrl.T.Helper()
  1741  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Campaign", reflect.TypeOf((*MockLeaderService)(nil).Campaign), electionID, opts)
  1742  }
  1743  
  1744  // Close mocks base method.
  1745  func (m *MockLeaderService) Close() error {
  1746  	m.ctrl.T.Helper()
  1747  	ret := m.ctrl.Call(m, "Close")
  1748  	ret0, _ := ret[0].(error)
  1749  	return ret0
  1750  }
  1751  
  1752  // Close indicates an expected call of Close.
  1753  func (mr *MockLeaderServiceMockRecorder) Close() *gomock.Call {
  1754  	mr.mock.ctrl.T.Helper()
  1755  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockLeaderService)(nil).Close))
  1756  }
  1757  
  1758  // Leader mocks base method.
  1759  func (m *MockLeaderService) Leader(electionID string) (string, error) {
  1760  	m.ctrl.T.Helper()
  1761  	ret := m.ctrl.Call(m, "Leader", electionID)
  1762  	ret0, _ := ret[0].(string)
  1763  	ret1, _ := ret[1].(error)
  1764  	return ret0, ret1
  1765  }
  1766  
  1767  // Leader indicates an expected call of Leader.
  1768  func (mr *MockLeaderServiceMockRecorder) Leader(electionID interface{}) *gomock.Call {
  1769  	mr.mock.ctrl.T.Helper()
  1770  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Leader", reflect.TypeOf((*MockLeaderService)(nil).Leader), electionID)
  1771  }
  1772  
  1773  // Observe mocks base method.
  1774  func (m *MockLeaderService) Observe(electionID string) (<-chan string, error) {
  1775  	m.ctrl.T.Helper()
  1776  	ret := m.ctrl.Call(m, "Observe", electionID)
  1777  	ret0, _ := ret[0].(<-chan string)
  1778  	ret1, _ := ret[1].(error)
  1779  	return ret0, ret1
  1780  }
  1781  
  1782  // Observe indicates an expected call of Observe.
  1783  func (mr *MockLeaderServiceMockRecorder) Observe(electionID interface{}) *gomock.Call {
  1784  	mr.mock.ctrl.T.Helper()
  1785  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Observe", reflect.TypeOf((*MockLeaderService)(nil).Observe), electionID)
  1786  }
  1787  
  1788  // Resign mocks base method.
  1789  func (m *MockLeaderService) Resign(electionID string) error {
  1790  	m.ctrl.T.Helper()
  1791  	ret := m.ctrl.Call(m, "Resign", electionID)
  1792  	ret0, _ := ret[0].(error)
  1793  	return ret0
  1794  }
  1795  
  1796  // Resign indicates an expected call of Resign.
  1797  func (mr *MockLeaderServiceMockRecorder) Resign(electionID interface{}) *gomock.Call {
  1798  	mr.mock.ctrl.T.Helper()
  1799  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Resign", reflect.TypeOf((*MockLeaderService)(nil).Resign), electionID)
  1800  }