github.com/m3db/m3@v1.5.0/src/dbnode/topology/topology_mock.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: github.com/m3db/m3/src/dbnode/topology/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 topology is a generated GoMock package.
    25  package topology
    26  
    27  import (
    28  	"reflect"
    29  
    30  	"github.com/m3db/m3/src/cluster/client"
    31  	"github.com/m3db/m3/src/cluster/services"
    32  	"github.com/m3db/m3/src/dbnode/sharding"
    33  	"github.com/m3db/m3/src/x/ident"
    34  	"github.com/m3db/m3/src/x/instrument"
    35  
    36  	"github.com/golang/mock/gomock"
    37  )
    38  
    39  // MockHost is a mock of Host interface.
    40  type MockHost struct {
    41  	ctrl     *gomock.Controller
    42  	recorder *MockHostMockRecorder
    43  }
    44  
    45  // MockHostMockRecorder is the mock recorder for MockHost.
    46  type MockHostMockRecorder struct {
    47  	mock *MockHost
    48  }
    49  
    50  // NewMockHost creates a new mock instance.
    51  func NewMockHost(ctrl *gomock.Controller) *MockHost {
    52  	mock := &MockHost{ctrl: ctrl}
    53  	mock.recorder = &MockHostMockRecorder{mock}
    54  	return mock
    55  }
    56  
    57  // EXPECT returns an object that allows the caller to indicate expected use.
    58  func (m *MockHost) EXPECT() *MockHostMockRecorder {
    59  	return m.recorder
    60  }
    61  
    62  // Address mocks base method.
    63  func (m *MockHost) Address() string {
    64  	m.ctrl.T.Helper()
    65  	ret := m.ctrl.Call(m, "Address")
    66  	ret0, _ := ret[0].(string)
    67  	return ret0
    68  }
    69  
    70  // Address indicates an expected call of Address.
    71  func (mr *MockHostMockRecorder) Address() *gomock.Call {
    72  	mr.mock.ctrl.T.Helper()
    73  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Address", reflect.TypeOf((*MockHost)(nil).Address))
    74  }
    75  
    76  // ID mocks base method.
    77  func (m *MockHost) ID() string {
    78  	m.ctrl.T.Helper()
    79  	ret := m.ctrl.Call(m, "ID")
    80  	ret0, _ := ret[0].(string)
    81  	return ret0
    82  }
    83  
    84  // ID indicates an expected call of ID.
    85  func (mr *MockHostMockRecorder) ID() *gomock.Call {
    86  	mr.mock.ctrl.T.Helper()
    87  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ID", reflect.TypeOf((*MockHost)(nil).ID))
    88  }
    89  
    90  // String mocks base method.
    91  func (m *MockHost) String() string {
    92  	m.ctrl.T.Helper()
    93  	ret := m.ctrl.Call(m, "String")
    94  	ret0, _ := ret[0].(string)
    95  	return ret0
    96  }
    97  
    98  // String indicates an expected call of String.
    99  func (mr *MockHostMockRecorder) String() *gomock.Call {
   100  	mr.mock.ctrl.T.Helper()
   101  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "String", reflect.TypeOf((*MockHost)(nil).String))
   102  }
   103  
   104  // MockHostShardSet is a mock of HostShardSet interface.
   105  type MockHostShardSet struct {
   106  	ctrl     *gomock.Controller
   107  	recorder *MockHostShardSetMockRecorder
   108  }
   109  
   110  // MockHostShardSetMockRecorder is the mock recorder for MockHostShardSet.
   111  type MockHostShardSetMockRecorder struct {
   112  	mock *MockHostShardSet
   113  }
   114  
   115  // NewMockHostShardSet creates a new mock instance.
   116  func NewMockHostShardSet(ctrl *gomock.Controller) *MockHostShardSet {
   117  	mock := &MockHostShardSet{ctrl: ctrl}
   118  	mock.recorder = &MockHostShardSetMockRecorder{mock}
   119  	return mock
   120  }
   121  
   122  // EXPECT returns an object that allows the caller to indicate expected use.
   123  func (m *MockHostShardSet) EXPECT() *MockHostShardSetMockRecorder {
   124  	return m.recorder
   125  }
   126  
   127  // Host mocks base method.
   128  func (m *MockHostShardSet) Host() Host {
   129  	m.ctrl.T.Helper()
   130  	ret := m.ctrl.Call(m, "Host")
   131  	ret0, _ := ret[0].(Host)
   132  	return ret0
   133  }
   134  
   135  // Host indicates an expected call of Host.
   136  func (mr *MockHostShardSetMockRecorder) Host() *gomock.Call {
   137  	mr.mock.ctrl.T.Helper()
   138  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Host", reflect.TypeOf((*MockHostShardSet)(nil).Host))
   139  }
   140  
   141  // ShardSet mocks base method.
   142  func (m *MockHostShardSet) ShardSet() sharding.ShardSet {
   143  	m.ctrl.T.Helper()
   144  	ret := m.ctrl.Call(m, "ShardSet")
   145  	ret0, _ := ret[0].(sharding.ShardSet)
   146  	return ret0
   147  }
   148  
   149  // ShardSet indicates an expected call of ShardSet.
   150  func (mr *MockHostShardSetMockRecorder) ShardSet() *gomock.Call {
   151  	mr.mock.ctrl.T.Helper()
   152  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ShardSet", reflect.TypeOf((*MockHostShardSet)(nil).ShardSet))
   153  }
   154  
   155  // MockInitializer is a mock of Initializer interface.
   156  type MockInitializer struct {
   157  	ctrl     *gomock.Controller
   158  	recorder *MockInitializerMockRecorder
   159  }
   160  
   161  // MockInitializerMockRecorder is the mock recorder for MockInitializer.
   162  type MockInitializerMockRecorder struct {
   163  	mock *MockInitializer
   164  }
   165  
   166  // NewMockInitializer creates a new mock instance.
   167  func NewMockInitializer(ctrl *gomock.Controller) *MockInitializer {
   168  	mock := &MockInitializer{ctrl: ctrl}
   169  	mock.recorder = &MockInitializerMockRecorder{mock}
   170  	return mock
   171  }
   172  
   173  // EXPECT returns an object that allows the caller to indicate expected use.
   174  func (m *MockInitializer) EXPECT() *MockInitializerMockRecorder {
   175  	return m.recorder
   176  }
   177  
   178  // Init mocks base method.
   179  func (m *MockInitializer) Init() (Topology, error) {
   180  	m.ctrl.T.Helper()
   181  	ret := m.ctrl.Call(m, "Init")
   182  	ret0, _ := ret[0].(Topology)
   183  	ret1, _ := ret[1].(error)
   184  	return ret0, ret1
   185  }
   186  
   187  // Init indicates an expected call of Init.
   188  func (mr *MockInitializerMockRecorder) Init() *gomock.Call {
   189  	mr.mock.ctrl.T.Helper()
   190  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Init", reflect.TypeOf((*MockInitializer)(nil).Init))
   191  }
   192  
   193  // TopologyIsSet mocks base method.
   194  func (m *MockInitializer) TopologyIsSet() (bool, error) {
   195  	m.ctrl.T.Helper()
   196  	ret := m.ctrl.Call(m, "TopologyIsSet")
   197  	ret0, _ := ret[0].(bool)
   198  	ret1, _ := ret[1].(error)
   199  	return ret0, ret1
   200  }
   201  
   202  // TopologyIsSet indicates an expected call of TopologyIsSet.
   203  func (mr *MockInitializerMockRecorder) TopologyIsSet() *gomock.Call {
   204  	mr.mock.ctrl.T.Helper()
   205  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TopologyIsSet", reflect.TypeOf((*MockInitializer)(nil).TopologyIsSet))
   206  }
   207  
   208  // MockTopology is a mock of Topology interface.
   209  type MockTopology struct {
   210  	ctrl     *gomock.Controller
   211  	recorder *MockTopologyMockRecorder
   212  }
   213  
   214  // MockTopologyMockRecorder is the mock recorder for MockTopology.
   215  type MockTopologyMockRecorder struct {
   216  	mock *MockTopology
   217  }
   218  
   219  // NewMockTopology creates a new mock instance.
   220  func NewMockTopology(ctrl *gomock.Controller) *MockTopology {
   221  	mock := &MockTopology{ctrl: ctrl}
   222  	mock.recorder = &MockTopologyMockRecorder{mock}
   223  	return mock
   224  }
   225  
   226  // EXPECT returns an object that allows the caller to indicate expected use.
   227  func (m *MockTopology) EXPECT() *MockTopologyMockRecorder {
   228  	return m.recorder
   229  }
   230  
   231  // Close mocks base method.
   232  func (m *MockTopology) Close() {
   233  	m.ctrl.T.Helper()
   234  	m.ctrl.Call(m, "Close")
   235  }
   236  
   237  // Close indicates an expected call of Close.
   238  func (mr *MockTopologyMockRecorder) Close() *gomock.Call {
   239  	mr.mock.ctrl.T.Helper()
   240  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockTopology)(nil).Close))
   241  }
   242  
   243  // Get mocks base method.
   244  func (m *MockTopology) Get() Map {
   245  	m.ctrl.T.Helper()
   246  	ret := m.ctrl.Call(m, "Get")
   247  	ret0, _ := ret[0].(Map)
   248  	return ret0
   249  }
   250  
   251  // Get indicates an expected call of Get.
   252  func (mr *MockTopologyMockRecorder) Get() *gomock.Call {
   253  	mr.mock.ctrl.T.Helper()
   254  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockTopology)(nil).Get))
   255  }
   256  
   257  // Watch mocks base method.
   258  func (m *MockTopology) Watch() (MapWatch, error) {
   259  	m.ctrl.T.Helper()
   260  	ret := m.ctrl.Call(m, "Watch")
   261  	ret0, _ := ret[0].(MapWatch)
   262  	ret1, _ := ret[1].(error)
   263  	return ret0, ret1
   264  }
   265  
   266  // Watch indicates an expected call of Watch.
   267  func (mr *MockTopologyMockRecorder) Watch() *gomock.Call {
   268  	mr.mock.ctrl.T.Helper()
   269  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Watch", reflect.TypeOf((*MockTopology)(nil).Watch))
   270  }
   271  
   272  // MockDynamicTopology is a mock of DynamicTopology interface.
   273  type MockDynamicTopology struct {
   274  	ctrl     *gomock.Controller
   275  	recorder *MockDynamicTopologyMockRecorder
   276  }
   277  
   278  // MockDynamicTopologyMockRecorder is the mock recorder for MockDynamicTopology.
   279  type MockDynamicTopologyMockRecorder struct {
   280  	mock *MockDynamicTopology
   281  }
   282  
   283  // NewMockDynamicTopology creates a new mock instance.
   284  func NewMockDynamicTopology(ctrl *gomock.Controller) *MockDynamicTopology {
   285  	mock := &MockDynamicTopology{ctrl: ctrl}
   286  	mock.recorder = &MockDynamicTopologyMockRecorder{mock}
   287  	return mock
   288  }
   289  
   290  // EXPECT returns an object that allows the caller to indicate expected use.
   291  func (m *MockDynamicTopology) EXPECT() *MockDynamicTopologyMockRecorder {
   292  	return m.recorder
   293  }
   294  
   295  // Close mocks base method.
   296  func (m *MockDynamicTopology) Close() {
   297  	m.ctrl.T.Helper()
   298  	m.ctrl.Call(m, "Close")
   299  }
   300  
   301  // Close indicates an expected call of Close.
   302  func (mr *MockDynamicTopologyMockRecorder) Close() *gomock.Call {
   303  	mr.mock.ctrl.T.Helper()
   304  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockDynamicTopology)(nil).Close))
   305  }
   306  
   307  // Get mocks base method.
   308  func (m *MockDynamicTopology) Get() Map {
   309  	m.ctrl.T.Helper()
   310  	ret := m.ctrl.Call(m, "Get")
   311  	ret0, _ := ret[0].(Map)
   312  	return ret0
   313  }
   314  
   315  // Get indicates an expected call of Get.
   316  func (mr *MockDynamicTopologyMockRecorder) Get() *gomock.Call {
   317  	mr.mock.ctrl.T.Helper()
   318  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockDynamicTopology)(nil).Get))
   319  }
   320  
   321  // MarkShardsAvailable mocks base method.
   322  func (m *MockDynamicTopology) MarkShardsAvailable(instanceID string, shardIDs ...uint32) error {
   323  	m.ctrl.T.Helper()
   324  	varargs := []interface{}{instanceID}
   325  	for _, a := range shardIDs {
   326  		varargs = append(varargs, a)
   327  	}
   328  	ret := m.ctrl.Call(m, "MarkShardsAvailable", varargs...)
   329  	ret0, _ := ret[0].(error)
   330  	return ret0
   331  }
   332  
   333  // MarkShardsAvailable indicates an expected call of MarkShardsAvailable.
   334  func (mr *MockDynamicTopologyMockRecorder) MarkShardsAvailable(instanceID interface{}, shardIDs ...interface{}) *gomock.Call {
   335  	mr.mock.ctrl.T.Helper()
   336  	varargs := append([]interface{}{instanceID}, shardIDs...)
   337  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MarkShardsAvailable", reflect.TypeOf((*MockDynamicTopology)(nil).MarkShardsAvailable), varargs...)
   338  }
   339  
   340  // Watch mocks base method.
   341  func (m *MockDynamicTopology) Watch() (MapWatch, error) {
   342  	m.ctrl.T.Helper()
   343  	ret := m.ctrl.Call(m, "Watch")
   344  	ret0, _ := ret[0].(MapWatch)
   345  	ret1, _ := ret[1].(error)
   346  	return ret0, ret1
   347  }
   348  
   349  // Watch indicates an expected call of Watch.
   350  func (mr *MockDynamicTopologyMockRecorder) Watch() *gomock.Call {
   351  	mr.mock.ctrl.T.Helper()
   352  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Watch", reflect.TypeOf((*MockDynamicTopology)(nil).Watch))
   353  }
   354  
   355  // MockMapWatch is a mock of MapWatch interface.
   356  type MockMapWatch struct {
   357  	ctrl     *gomock.Controller
   358  	recorder *MockMapWatchMockRecorder
   359  }
   360  
   361  // MockMapWatchMockRecorder is the mock recorder for MockMapWatch.
   362  type MockMapWatchMockRecorder struct {
   363  	mock *MockMapWatch
   364  }
   365  
   366  // NewMockMapWatch creates a new mock instance.
   367  func NewMockMapWatch(ctrl *gomock.Controller) *MockMapWatch {
   368  	mock := &MockMapWatch{ctrl: ctrl}
   369  	mock.recorder = &MockMapWatchMockRecorder{mock}
   370  	return mock
   371  }
   372  
   373  // EXPECT returns an object that allows the caller to indicate expected use.
   374  func (m *MockMapWatch) EXPECT() *MockMapWatchMockRecorder {
   375  	return m.recorder
   376  }
   377  
   378  // C mocks base method.
   379  func (m *MockMapWatch) C() <-chan struct{} {
   380  	m.ctrl.T.Helper()
   381  	ret := m.ctrl.Call(m, "C")
   382  	ret0, _ := ret[0].(<-chan struct{})
   383  	return ret0
   384  }
   385  
   386  // C indicates an expected call of C.
   387  func (mr *MockMapWatchMockRecorder) C() *gomock.Call {
   388  	mr.mock.ctrl.T.Helper()
   389  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "C", reflect.TypeOf((*MockMapWatch)(nil).C))
   390  }
   391  
   392  // Close mocks base method.
   393  func (m *MockMapWatch) Close() {
   394  	m.ctrl.T.Helper()
   395  	m.ctrl.Call(m, "Close")
   396  }
   397  
   398  // Close indicates an expected call of Close.
   399  func (mr *MockMapWatchMockRecorder) Close() *gomock.Call {
   400  	mr.mock.ctrl.T.Helper()
   401  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockMapWatch)(nil).Close))
   402  }
   403  
   404  // Get mocks base method.
   405  func (m *MockMapWatch) Get() Map {
   406  	m.ctrl.T.Helper()
   407  	ret := m.ctrl.Call(m, "Get")
   408  	ret0, _ := ret[0].(Map)
   409  	return ret0
   410  }
   411  
   412  // Get indicates an expected call of Get.
   413  func (mr *MockMapWatchMockRecorder) Get() *gomock.Call {
   414  	mr.mock.ctrl.T.Helper()
   415  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockMapWatch)(nil).Get))
   416  }
   417  
   418  // MockMap is a mock of Map interface.
   419  type MockMap struct {
   420  	ctrl     *gomock.Controller
   421  	recorder *MockMapMockRecorder
   422  }
   423  
   424  // MockMapMockRecorder is the mock recorder for MockMap.
   425  type MockMapMockRecorder struct {
   426  	mock *MockMap
   427  }
   428  
   429  // NewMockMap creates a new mock instance.
   430  func NewMockMap(ctrl *gomock.Controller) *MockMap {
   431  	mock := &MockMap{ctrl: ctrl}
   432  	mock.recorder = &MockMapMockRecorder{mock}
   433  	return mock
   434  }
   435  
   436  // EXPECT returns an object that allows the caller to indicate expected use.
   437  func (m *MockMap) EXPECT() *MockMapMockRecorder {
   438  	return m.recorder
   439  }
   440  
   441  // HostShardSets mocks base method.
   442  func (m *MockMap) HostShardSets() []HostShardSet {
   443  	m.ctrl.T.Helper()
   444  	ret := m.ctrl.Call(m, "HostShardSets")
   445  	ret0, _ := ret[0].([]HostShardSet)
   446  	return ret0
   447  }
   448  
   449  // HostShardSets indicates an expected call of HostShardSets.
   450  func (mr *MockMapMockRecorder) HostShardSets() *gomock.Call {
   451  	mr.mock.ctrl.T.Helper()
   452  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HostShardSets", reflect.TypeOf((*MockMap)(nil).HostShardSets))
   453  }
   454  
   455  // Hosts mocks base method.
   456  func (m *MockMap) Hosts() []Host {
   457  	m.ctrl.T.Helper()
   458  	ret := m.ctrl.Call(m, "Hosts")
   459  	ret0, _ := ret[0].([]Host)
   460  	return ret0
   461  }
   462  
   463  // Hosts indicates an expected call of Hosts.
   464  func (mr *MockMapMockRecorder) Hosts() *gomock.Call {
   465  	mr.mock.ctrl.T.Helper()
   466  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Hosts", reflect.TypeOf((*MockMap)(nil).Hosts))
   467  }
   468  
   469  // HostsLen mocks base method.
   470  func (m *MockMap) HostsLen() int {
   471  	m.ctrl.T.Helper()
   472  	ret := m.ctrl.Call(m, "HostsLen")
   473  	ret0, _ := ret[0].(int)
   474  	return ret0
   475  }
   476  
   477  // HostsLen indicates an expected call of HostsLen.
   478  func (mr *MockMapMockRecorder) HostsLen() *gomock.Call {
   479  	mr.mock.ctrl.T.Helper()
   480  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HostsLen", reflect.TypeOf((*MockMap)(nil).HostsLen))
   481  }
   482  
   483  // LookupHostShardSet mocks base method.
   484  func (m *MockMap) LookupHostShardSet(hostID string) (HostShardSet, bool) {
   485  	m.ctrl.T.Helper()
   486  	ret := m.ctrl.Call(m, "LookupHostShardSet", hostID)
   487  	ret0, _ := ret[0].(HostShardSet)
   488  	ret1, _ := ret[1].(bool)
   489  	return ret0, ret1
   490  }
   491  
   492  // LookupHostShardSet indicates an expected call of LookupHostShardSet.
   493  func (mr *MockMapMockRecorder) LookupHostShardSet(hostID interface{}) *gomock.Call {
   494  	mr.mock.ctrl.T.Helper()
   495  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LookupHostShardSet", reflect.TypeOf((*MockMap)(nil).LookupHostShardSet), hostID)
   496  }
   497  
   498  // MajorityReplicas mocks base method.
   499  func (m *MockMap) MajorityReplicas() int {
   500  	m.ctrl.T.Helper()
   501  	ret := m.ctrl.Call(m, "MajorityReplicas")
   502  	ret0, _ := ret[0].(int)
   503  	return ret0
   504  }
   505  
   506  // MajorityReplicas indicates an expected call of MajorityReplicas.
   507  func (mr *MockMapMockRecorder) MajorityReplicas() *gomock.Call {
   508  	mr.mock.ctrl.T.Helper()
   509  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MajorityReplicas", reflect.TypeOf((*MockMap)(nil).MajorityReplicas))
   510  }
   511  
   512  // Replicas mocks base method.
   513  func (m *MockMap) Replicas() int {
   514  	m.ctrl.T.Helper()
   515  	ret := m.ctrl.Call(m, "Replicas")
   516  	ret0, _ := ret[0].(int)
   517  	return ret0
   518  }
   519  
   520  // Replicas indicates an expected call of Replicas.
   521  func (mr *MockMapMockRecorder) Replicas() *gomock.Call {
   522  	mr.mock.ctrl.T.Helper()
   523  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Replicas", reflect.TypeOf((*MockMap)(nil).Replicas))
   524  }
   525  
   526  // Route mocks base method.
   527  func (m *MockMap) Route(id ident.ID) (uint32, []Host, error) {
   528  	m.ctrl.T.Helper()
   529  	ret := m.ctrl.Call(m, "Route", id)
   530  	ret0, _ := ret[0].(uint32)
   531  	ret1, _ := ret[1].([]Host)
   532  	ret2, _ := ret[2].(error)
   533  	return ret0, ret1, ret2
   534  }
   535  
   536  // Route indicates an expected call of Route.
   537  func (mr *MockMapMockRecorder) Route(id interface{}) *gomock.Call {
   538  	mr.mock.ctrl.T.Helper()
   539  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Route", reflect.TypeOf((*MockMap)(nil).Route), id)
   540  }
   541  
   542  // RouteForEach mocks base method.
   543  func (m *MockMap) RouteForEach(id ident.ID, forEachFn RouteForEachFn) error {
   544  	m.ctrl.T.Helper()
   545  	ret := m.ctrl.Call(m, "RouteForEach", id, forEachFn)
   546  	ret0, _ := ret[0].(error)
   547  	return ret0
   548  }
   549  
   550  // RouteForEach indicates an expected call of RouteForEach.
   551  func (mr *MockMapMockRecorder) RouteForEach(id, forEachFn interface{}) *gomock.Call {
   552  	mr.mock.ctrl.T.Helper()
   553  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RouteForEach", reflect.TypeOf((*MockMap)(nil).RouteForEach), id, forEachFn)
   554  }
   555  
   556  // RouteShard mocks base method.
   557  func (m *MockMap) RouteShard(shard uint32) ([]Host, error) {
   558  	m.ctrl.T.Helper()
   559  	ret := m.ctrl.Call(m, "RouteShard", shard)
   560  	ret0, _ := ret[0].([]Host)
   561  	ret1, _ := ret[1].(error)
   562  	return ret0, ret1
   563  }
   564  
   565  // RouteShard indicates an expected call of RouteShard.
   566  func (mr *MockMapMockRecorder) RouteShard(shard interface{}) *gomock.Call {
   567  	mr.mock.ctrl.T.Helper()
   568  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RouteShard", reflect.TypeOf((*MockMap)(nil).RouteShard), shard)
   569  }
   570  
   571  // RouteShardForEach mocks base method.
   572  func (m *MockMap) RouteShardForEach(shard uint32, forEachFn RouteForEachFn) error {
   573  	m.ctrl.T.Helper()
   574  	ret := m.ctrl.Call(m, "RouteShardForEach", shard, forEachFn)
   575  	ret0, _ := ret[0].(error)
   576  	return ret0
   577  }
   578  
   579  // RouteShardForEach indicates an expected call of RouteShardForEach.
   580  func (mr *MockMapMockRecorder) RouteShardForEach(shard, forEachFn interface{}) *gomock.Call {
   581  	mr.mock.ctrl.T.Helper()
   582  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RouteShardForEach", reflect.TypeOf((*MockMap)(nil).RouteShardForEach), shard, forEachFn)
   583  }
   584  
   585  // ShardSet mocks base method.
   586  func (m *MockMap) ShardSet() sharding.ShardSet {
   587  	m.ctrl.T.Helper()
   588  	ret := m.ctrl.Call(m, "ShardSet")
   589  	ret0, _ := ret[0].(sharding.ShardSet)
   590  	return ret0
   591  }
   592  
   593  // ShardSet indicates an expected call of ShardSet.
   594  func (mr *MockMapMockRecorder) ShardSet() *gomock.Call {
   595  	mr.mock.ctrl.T.Helper()
   596  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ShardSet", reflect.TypeOf((*MockMap)(nil).ShardSet))
   597  }
   598  
   599  // MockStaticOptions is a mock of StaticOptions interface.
   600  type MockStaticOptions struct {
   601  	ctrl     *gomock.Controller
   602  	recorder *MockStaticOptionsMockRecorder
   603  }
   604  
   605  // MockStaticOptionsMockRecorder is the mock recorder for MockStaticOptions.
   606  type MockStaticOptionsMockRecorder struct {
   607  	mock *MockStaticOptions
   608  }
   609  
   610  // NewMockStaticOptions creates a new mock instance.
   611  func NewMockStaticOptions(ctrl *gomock.Controller) *MockStaticOptions {
   612  	mock := &MockStaticOptions{ctrl: ctrl}
   613  	mock.recorder = &MockStaticOptionsMockRecorder{mock}
   614  	return mock
   615  }
   616  
   617  // EXPECT returns an object that allows the caller to indicate expected use.
   618  func (m *MockStaticOptions) EXPECT() *MockStaticOptionsMockRecorder {
   619  	return m.recorder
   620  }
   621  
   622  // HostShardSets mocks base method.
   623  func (m *MockStaticOptions) HostShardSets() []HostShardSet {
   624  	m.ctrl.T.Helper()
   625  	ret := m.ctrl.Call(m, "HostShardSets")
   626  	ret0, _ := ret[0].([]HostShardSet)
   627  	return ret0
   628  }
   629  
   630  // HostShardSets indicates an expected call of HostShardSets.
   631  func (mr *MockStaticOptionsMockRecorder) HostShardSets() *gomock.Call {
   632  	mr.mock.ctrl.T.Helper()
   633  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HostShardSets", reflect.TypeOf((*MockStaticOptions)(nil).HostShardSets))
   634  }
   635  
   636  // Replicas mocks base method.
   637  func (m *MockStaticOptions) Replicas() int {
   638  	m.ctrl.T.Helper()
   639  	ret := m.ctrl.Call(m, "Replicas")
   640  	ret0, _ := ret[0].(int)
   641  	return ret0
   642  }
   643  
   644  // Replicas indicates an expected call of Replicas.
   645  func (mr *MockStaticOptionsMockRecorder) Replicas() *gomock.Call {
   646  	mr.mock.ctrl.T.Helper()
   647  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Replicas", reflect.TypeOf((*MockStaticOptions)(nil).Replicas))
   648  }
   649  
   650  // SetHostShardSets mocks base method.
   651  func (m *MockStaticOptions) SetHostShardSets(value []HostShardSet) StaticOptions {
   652  	m.ctrl.T.Helper()
   653  	ret := m.ctrl.Call(m, "SetHostShardSets", value)
   654  	ret0, _ := ret[0].(StaticOptions)
   655  	return ret0
   656  }
   657  
   658  // SetHostShardSets indicates an expected call of SetHostShardSets.
   659  func (mr *MockStaticOptionsMockRecorder) SetHostShardSets(value interface{}) *gomock.Call {
   660  	mr.mock.ctrl.T.Helper()
   661  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetHostShardSets", reflect.TypeOf((*MockStaticOptions)(nil).SetHostShardSets), value)
   662  }
   663  
   664  // SetReplicas mocks base method.
   665  func (m *MockStaticOptions) SetReplicas(value int) StaticOptions {
   666  	m.ctrl.T.Helper()
   667  	ret := m.ctrl.Call(m, "SetReplicas", value)
   668  	ret0, _ := ret[0].(StaticOptions)
   669  	return ret0
   670  }
   671  
   672  // SetReplicas indicates an expected call of SetReplicas.
   673  func (mr *MockStaticOptionsMockRecorder) SetReplicas(value interface{}) *gomock.Call {
   674  	mr.mock.ctrl.T.Helper()
   675  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetReplicas", reflect.TypeOf((*MockStaticOptions)(nil).SetReplicas), value)
   676  }
   677  
   678  // SetShardSet mocks base method.
   679  func (m *MockStaticOptions) SetShardSet(value sharding.ShardSet) StaticOptions {
   680  	m.ctrl.T.Helper()
   681  	ret := m.ctrl.Call(m, "SetShardSet", value)
   682  	ret0, _ := ret[0].(StaticOptions)
   683  	return ret0
   684  }
   685  
   686  // SetShardSet indicates an expected call of SetShardSet.
   687  func (mr *MockStaticOptionsMockRecorder) SetShardSet(value interface{}) *gomock.Call {
   688  	mr.mock.ctrl.T.Helper()
   689  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetShardSet", reflect.TypeOf((*MockStaticOptions)(nil).SetShardSet), value)
   690  }
   691  
   692  // ShardSet mocks base method.
   693  func (m *MockStaticOptions) ShardSet() sharding.ShardSet {
   694  	m.ctrl.T.Helper()
   695  	ret := m.ctrl.Call(m, "ShardSet")
   696  	ret0, _ := ret[0].(sharding.ShardSet)
   697  	return ret0
   698  }
   699  
   700  // ShardSet indicates an expected call of ShardSet.
   701  func (mr *MockStaticOptionsMockRecorder) ShardSet() *gomock.Call {
   702  	mr.mock.ctrl.T.Helper()
   703  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ShardSet", reflect.TypeOf((*MockStaticOptions)(nil).ShardSet))
   704  }
   705  
   706  // Validate mocks base method.
   707  func (m *MockStaticOptions) Validate() error {
   708  	m.ctrl.T.Helper()
   709  	ret := m.ctrl.Call(m, "Validate")
   710  	ret0, _ := ret[0].(error)
   711  	return ret0
   712  }
   713  
   714  // Validate indicates an expected call of Validate.
   715  func (mr *MockStaticOptionsMockRecorder) Validate() *gomock.Call {
   716  	mr.mock.ctrl.T.Helper()
   717  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Validate", reflect.TypeOf((*MockStaticOptions)(nil).Validate))
   718  }
   719  
   720  // MockDynamicOptions is a mock of DynamicOptions interface.
   721  type MockDynamicOptions struct {
   722  	ctrl     *gomock.Controller
   723  	recorder *MockDynamicOptionsMockRecorder
   724  }
   725  
   726  // MockDynamicOptionsMockRecorder is the mock recorder for MockDynamicOptions.
   727  type MockDynamicOptionsMockRecorder struct {
   728  	mock *MockDynamicOptions
   729  }
   730  
   731  // NewMockDynamicOptions creates a new mock instance.
   732  func NewMockDynamicOptions(ctrl *gomock.Controller) *MockDynamicOptions {
   733  	mock := &MockDynamicOptions{ctrl: ctrl}
   734  	mock.recorder = &MockDynamicOptionsMockRecorder{mock}
   735  	return mock
   736  }
   737  
   738  // EXPECT returns an object that allows the caller to indicate expected use.
   739  func (m *MockDynamicOptions) EXPECT() *MockDynamicOptionsMockRecorder {
   740  	return m.recorder
   741  }
   742  
   743  // ConfigServiceClient mocks base method.
   744  func (m *MockDynamicOptions) ConfigServiceClient() client.Client {
   745  	m.ctrl.T.Helper()
   746  	ret := m.ctrl.Call(m, "ConfigServiceClient")
   747  	ret0, _ := ret[0].(client.Client)
   748  	return ret0
   749  }
   750  
   751  // ConfigServiceClient indicates an expected call of ConfigServiceClient.
   752  func (mr *MockDynamicOptionsMockRecorder) ConfigServiceClient() *gomock.Call {
   753  	mr.mock.ctrl.T.Helper()
   754  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ConfigServiceClient", reflect.TypeOf((*MockDynamicOptions)(nil).ConfigServiceClient))
   755  }
   756  
   757  // HashGen mocks base method.
   758  func (m *MockDynamicOptions) HashGen() sharding.HashGen {
   759  	m.ctrl.T.Helper()
   760  	ret := m.ctrl.Call(m, "HashGen")
   761  	ret0, _ := ret[0].(sharding.HashGen)
   762  	return ret0
   763  }
   764  
   765  // HashGen indicates an expected call of HashGen.
   766  func (mr *MockDynamicOptionsMockRecorder) HashGen() *gomock.Call {
   767  	mr.mock.ctrl.T.Helper()
   768  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HashGen", reflect.TypeOf((*MockDynamicOptions)(nil).HashGen))
   769  }
   770  
   771  // InstrumentOptions mocks base method.
   772  func (m *MockDynamicOptions) InstrumentOptions() instrument.Options {
   773  	m.ctrl.T.Helper()
   774  	ret := m.ctrl.Call(m, "InstrumentOptions")
   775  	ret0, _ := ret[0].(instrument.Options)
   776  	return ret0
   777  }
   778  
   779  // InstrumentOptions indicates an expected call of InstrumentOptions.
   780  func (mr *MockDynamicOptionsMockRecorder) InstrumentOptions() *gomock.Call {
   781  	mr.mock.ctrl.T.Helper()
   782  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InstrumentOptions", reflect.TypeOf((*MockDynamicOptions)(nil).InstrumentOptions))
   783  }
   784  
   785  // QueryOptions mocks base method.
   786  func (m *MockDynamicOptions) QueryOptions() services.QueryOptions {
   787  	m.ctrl.T.Helper()
   788  	ret := m.ctrl.Call(m, "QueryOptions")
   789  	ret0, _ := ret[0].(services.QueryOptions)
   790  	return ret0
   791  }
   792  
   793  // QueryOptions indicates an expected call of QueryOptions.
   794  func (mr *MockDynamicOptionsMockRecorder) QueryOptions() *gomock.Call {
   795  	mr.mock.ctrl.T.Helper()
   796  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "QueryOptions", reflect.TypeOf((*MockDynamicOptions)(nil).QueryOptions))
   797  }
   798  
   799  // ServiceID mocks base method.
   800  func (m *MockDynamicOptions) ServiceID() services.ServiceID {
   801  	m.ctrl.T.Helper()
   802  	ret := m.ctrl.Call(m, "ServiceID")
   803  	ret0, _ := ret[0].(services.ServiceID)
   804  	return ret0
   805  }
   806  
   807  // ServiceID indicates an expected call of ServiceID.
   808  func (mr *MockDynamicOptionsMockRecorder) ServiceID() *gomock.Call {
   809  	mr.mock.ctrl.T.Helper()
   810  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServiceID", reflect.TypeOf((*MockDynamicOptions)(nil).ServiceID))
   811  }
   812  
   813  // ServicesOverrideOptions mocks base method.
   814  func (m *MockDynamicOptions) ServicesOverrideOptions() services.OverrideOptions {
   815  	m.ctrl.T.Helper()
   816  	ret := m.ctrl.Call(m, "ServicesOverrideOptions")
   817  	ret0, _ := ret[0].(services.OverrideOptions)
   818  	return ret0
   819  }
   820  
   821  // ServicesOverrideOptions indicates an expected call of ServicesOverrideOptions.
   822  func (mr *MockDynamicOptionsMockRecorder) ServicesOverrideOptions() *gomock.Call {
   823  	mr.mock.ctrl.T.Helper()
   824  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServicesOverrideOptions", reflect.TypeOf((*MockDynamicOptions)(nil).ServicesOverrideOptions))
   825  }
   826  
   827  // SetConfigServiceClient mocks base method.
   828  func (m *MockDynamicOptions) SetConfigServiceClient(c client.Client) DynamicOptions {
   829  	m.ctrl.T.Helper()
   830  	ret := m.ctrl.Call(m, "SetConfigServiceClient", c)
   831  	ret0, _ := ret[0].(DynamicOptions)
   832  	return ret0
   833  }
   834  
   835  // SetConfigServiceClient indicates an expected call of SetConfigServiceClient.
   836  func (mr *MockDynamicOptionsMockRecorder) SetConfigServiceClient(c interface{}) *gomock.Call {
   837  	mr.mock.ctrl.T.Helper()
   838  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetConfigServiceClient", reflect.TypeOf((*MockDynamicOptions)(nil).SetConfigServiceClient), c)
   839  }
   840  
   841  // SetHashGen mocks base method.
   842  func (m *MockDynamicOptions) SetHashGen(h sharding.HashGen) DynamicOptions {
   843  	m.ctrl.T.Helper()
   844  	ret := m.ctrl.Call(m, "SetHashGen", h)
   845  	ret0, _ := ret[0].(DynamicOptions)
   846  	return ret0
   847  }
   848  
   849  // SetHashGen indicates an expected call of SetHashGen.
   850  func (mr *MockDynamicOptionsMockRecorder) SetHashGen(h interface{}) *gomock.Call {
   851  	mr.mock.ctrl.T.Helper()
   852  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetHashGen", reflect.TypeOf((*MockDynamicOptions)(nil).SetHashGen), h)
   853  }
   854  
   855  // SetInstrumentOptions mocks base method.
   856  func (m *MockDynamicOptions) SetInstrumentOptions(value instrument.Options) DynamicOptions {
   857  	m.ctrl.T.Helper()
   858  	ret := m.ctrl.Call(m, "SetInstrumentOptions", value)
   859  	ret0, _ := ret[0].(DynamicOptions)
   860  	return ret0
   861  }
   862  
   863  // SetInstrumentOptions indicates an expected call of SetInstrumentOptions.
   864  func (mr *MockDynamicOptionsMockRecorder) SetInstrumentOptions(value interface{}) *gomock.Call {
   865  	mr.mock.ctrl.T.Helper()
   866  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetInstrumentOptions", reflect.TypeOf((*MockDynamicOptions)(nil).SetInstrumentOptions), value)
   867  }
   868  
   869  // SetQueryOptions mocks base method.
   870  func (m *MockDynamicOptions) SetQueryOptions(value services.QueryOptions) DynamicOptions {
   871  	m.ctrl.T.Helper()
   872  	ret := m.ctrl.Call(m, "SetQueryOptions", value)
   873  	ret0, _ := ret[0].(DynamicOptions)
   874  	return ret0
   875  }
   876  
   877  // SetQueryOptions indicates an expected call of SetQueryOptions.
   878  func (mr *MockDynamicOptionsMockRecorder) SetQueryOptions(value interface{}) *gomock.Call {
   879  	mr.mock.ctrl.T.Helper()
   880  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetQueryOptions", reflect.TypeOf((*MockDynamicOptions)(nil).SetQueryOptions), value)
   881  }
   882  
   883  // SetServiceID mocks base method.
   884  func (m *MockDynamicOptions) SetServiceID(s services.ServiceID) DynamicOptions {
   885  	m.ctrl.T.Helper()
   886  	ret := m.ctrl.Call(m, "SetServiceID", s)
   887  	ret0, _ := ret[0].(DynamicOptions)
   888  	return ret0
   889  }
   890  
   891  // SetServiceID indicates an expected call of SetServiceID.
   892  func (mr *MockDynamicOptionsMockRecorder) SetServiceID(s interface{}) *gomock.Call {
   893  	mr.mock.ctrl.T.Helper()
   894  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetServiceID", reflect.TypeOf((*MockDynamicOptions)(nil).SetServiceID), s)
   895  }
   896  
   897  // SetServicesOverrideOptions mocks base method.
   898  func (m *MockDynamicOptions) SetServicesOverrideOptions(opts services.OverrideOptions) DynamicOptions {
   899  	m.ctrl.T.Helper()
   900  	ret := m.ctrl.Call(m, "SetServicesOverrideOptions", opts)
   901  	ret0, _ := ret[0].(DynamicOptions)
   902  	return ret0
   903  }
   904  
   905  // SetServicesOverrideOptions indicates an expected call of SetServicesOverrideOptions.
   906  func (mr *MockDynamicOptionsMockRecorder) SetServicesOverrideOptions(opts interface{}) *gomock.Call {
   907  	mr.mock.ctrl.T.Helper()
   908  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetServicesOverrideOptions", reflect.TypeOf((*MockDynamicOptions)(nil).SetServicesOverrideOptions), opts)
   909  }
   910  
   911  // Validate mocks base method.
   912  func (m *MockDynamicOptions) Validate() error {
   913  	m.ctrl.T.Helper()
   914  	ret := m.ctrl.Call(m, "Validate")
   915  	ret0, _ := ret[0].(error)
   916  	return ret0
   917  }
   918  
   919  // Validate indicates an expected call of Validate.
   920  func (mr *MockDynamicOptionsMockRecorder) Validate() *gomock.Call {
   921  	mr.mock.ctrl.T.Helper()
   922  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Validate", reflect.TypeOf((*MockDynamicOptions)(nil).Validate))
   923  }
   924  
   925  // MockMapProvider is a mock of MapProvider interface.
   926  type MockMapProvider struct {
   927  	ctrl     *gomock.Controller
   928  	recorder *MockMapProviderMockRecorder
   929  }
   930  
   931  // MockMapProviderMockRecorder is the mock recorder for MockMapProvider.
   932  type MockMapProviderMockRecorder struct {
   933  	mock *MockMapProvider
   934  }
   935  
   936  // NewMockMapProvider creates a new mock instance.
   937  func NewMockMapProvider(ctrl *gomock.Controller) *MockMapProvider {
   938  	mock := &MockMapProvider{ctrl: ctrl}
   939  	mock.recorder = &MockMapProviderMockRecorder{mock}
   940  	return mock
   941  }
   942  
   943  // EXPECT returns an object that allows the caller to indicate expected use.
   944  func (m *MockMapProvider) EXPECT() *MockMapProviderMockRecorder {
   945  	return m.recorder
   946  }
   947  
   948  // TopologyMap mocks base method.
   949  func (m *MockMapProvider) TopologyMap() (Map, error) {
   950  	m.ctrl.T.Helper()
   951  	ret := m.ctrl.Call(m, "TopologyMap")
   952  	ret0, _ := ret[0].(Map)
   953  	ret1, _ := ret[1].(error)
   954  	return ret0, ret1
   955  }
   956  
   957  // TopologyMap indicates an expected call of TopologyMap.
   958  func (mr *MockMapProviderMockRecorder) TopologyMap() *gomock.Call {
   959  	mr.mock.ctrl.T.Helper()
   960  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TopologyMap", reflect.TypeOf((*MockMapProvider)(nil).TopologyMap))
   961  }