github.com/m3db/m3@v1.5.1-0.20231129193456-75a402aa583b/src/dbnode/topology/topology_mock.go (about) 1 // Code generated by MockGen. DO NOT EDIT. 2 // Source: ../../topology/types.go 3 4 // Copyright (c) 2023 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 // LookupInitializingHostPair mocks base method. 499 func (m *MockMap) LookupInitializingHostPair(leavingHostID string, id uint32) (string, bool) { 500 m.ctrl.T.Helper() 501 ret := m.ctrl.Call(m, "LookupInitializingHostPair", leavingHostID, id) 502 ret0, _ := ret[0].(string) 503 ret1, _ := ret[1].(bool) 504 return ret0, ret1 505 } 506 507 // LookupInitializingHostPair indicates an expected call of LookupInitializingHostPair. 508 func (mr *MockMapMockRecorder) LookupInitializingHostPair(leavingHostID, id interface{}) *gomock.Call { 509 mr.mock.ctrl.T.Helper() 510 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LookupInitializingHostPair", reflect.TypeOf((*MockMap)(nil).LookupInitializingHostPair), leavingHostID, id) 511 } 512 513 // MajorityReplicas mocks base method. 514 func (m *MockMap) MajorityReplicas() int { 515 m.ctrl.T.Helper() 516 ret := m.ctrl.Call(m, "MajorityReplicas") 517 ret0, _ := ret[0].(int) 518 return ret0 519 } 520 521 // MajorityReplicas indicates an expected call of MajorityReplicas. 522 func (mr *MockMapMockRecorder) MajorityReplicas() *gomock.Call { 523 mr.mock.ctrl.T.Helper() 524 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MajorityReplicas", reflect.TypeOf((*MockMap)(nil).MajorityReplicas)) 525 } 526 527 // Replicas mocks base method. 528 func (m *MockMap) Replicas() int { 529 m.ctrl.T.Helper() 530 ret := m.ctrl.Call(m, "Replicas") 531 ret0, _ := ret[0].(int) 532 return ret0 533 } 534 535 // Replicas indicates an expected call of Replicas. 536 func (mr *MockMapMockRecorder) Replicas() *gomock.Call { 537 mr.mock.ctrl.T.Helper() 538 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Replicas", reflect.TypeOf((*MockMap)(nil).Replicas)) 539 } 540 541 // Route mocks base method. 542 func (m *MockMap) Route(id ident.ID) (uint32, []Host, error) { 543 m.ctrl.T.Helper() 544 ret := m.ctrl.Call(m, "Route", id) 545 ret0, _ := ret[0].(uint32) 546 ret1, _ := ret[1].([]Host) 547 ret2, _ := ret[2].(error) 548 return ret0, ret1, ret2 549 } 550 551 // Route indicates an expected call of Route. 552 func (mr *MockMapMockRecorder) Route(id interface{}) *gomock.Call { 553 mr.mock.ctrl.T.Helper() 554 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Route", reflect.TypeOf((*MockMap)(nil).Route), id) 555 } 556 557 // RouteForEach mocks base method. 558 func (m *MockMap) RouteForEach(id ident.ID, forEachFn RouteForEachFn) error { 559 m.ctrl.T.Helper() 560 ret := m.ctrl.Call(m, "RouteForEach", id, forEachFn) 561 ret0, _ := ret[0].(error) 562 return ret0 563 } 564 565 // RouteForEach indicates an expected call of RouteForEach. 566 func (mr *MockMapMockRecorder) RouteForEach(id, forEachFn interface{}) *gomock.Call { 567 mr.mock.ctrl.T.Helper() 568 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RouteForEach", reflect.TypeOf((*MockMap)(nil).RouteForEach), id, forEachFn) 569 } 570 571 // RouteShard mocks base method. 572 func (m *MockMap) RouteShard(shard uint32) ([]Host, error) { 573 m.ctrl.T.Helper() 574 ret := m.ctrl.Call(m, "RouteShard", shard) 575 ret0, _ := ret[0].([]Host) 576 ret1, _ := ret[1].(error) 577 return ret0, ret1 578 } 579 580 // RouteShard indicates an expected call of RouteShard. 581 func (mr *MockMapMockRecorder) RouteShard(shard interface{}) *gomock.Call { 582 mr.mock.ctrl.T.Helper() 583 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RouteShard", reflect.TypeOf((*MockMap)(nil).RouteShard), shard) 584 } 585 586 // RouteShardForEach mocks base method. 587 func (m *MockMap) RouteShardForEach(shard uint32, forEachFn RouteForEachFn) error { 588 m.ctrl.T.Helper() 589 ret := m.ctrl.Call(m, "RouteShardForEach", shard, forEachFn) 590 ret0, _ := ret[0].(error) 591 return ret0 592 } 593 594 // RouteShardForEach indicates an expected call of RouteShardForEach. 595 func (mr *MockMapMockRecorder) RouteShardForEach(shard, forEachFn interface{}) *gomock.Call { 596 mr.mock.ctrl.T.Helper() 597 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RouteShardForEach", reflect.TypeOf((*MockMap)(nil).RouteShardForEach), shard, forEachFn) 598 } 599 600 // ShardSet mocks base method. 601 func (m *MockMap) ShardSet() sharding.ShardSet { 602 m.ctrl.T.Helper() 603 ret := m.ctrl.Call(m, "ShardSet") 604 ret0, _ := ret[0].(sharding.ShardSet) 605 return ret0 606 } 607 608 // ShardSet indicates an expected call of ShardSet. 609 func (mr *MockMapMockRecorder) ShardSet() *gomock.Call { 610 mr.mock.ctrl.T.Helper() 611 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ShardSet", reflect.TypeOf((*MockMap)(nil).ShardSet)) 612 } 613 614 // MockStaticOptions is a mock of StaticOptions interface. 615 type MockStaticOptions struct { 616 ctrl *gomock.Controller 617 recorder *MockStaticOptionsMockRecorder 618 } 619 620 // MockStaticOptionsMockRecorder is the mock recorder for MockStaticOptions. 621 type MockStaticOptionsMockRecorder struct { 622 mock *MockStaticOptions 623 } 624 625 // NewMockStaticOptions creates a new mock instance. 626 func NewMockStaticOptions(ctrl *gomock.Controller) *MockStaticOptions { 627 mock := &MockStaticOptions{ctrl: ctrl} 628 mock.recorder = &MockStaticOptionsMockRecorder{mock} 629 return mock 630 } 631 632 // EXPECT returns an object that allows the caller to indicate expected use. 633 func (m *MockStaticOptions) EXPECT() *MockStaticOptionsMockRecorder { 634 return m.recorder 635 } 636 637 // HostShardSets mocks base method. 638 func (m *MockStaticOptions) HostShardSets() []HostShardSet { 639 m.ctrl.T.Helper() 640 ret := m.ctrl.Call(m, "HostShardSets") 641 ret0, _ := ret[0].([]HostShardSet) 642 return ret0 643 } 644 645 // HostShardSets indicates an expected call of HostShardSets. 646 func (mr *MockStaticOptionsMockRecorder) HostShardSets() *gomock.Call { 647 mr.mock.ctrl.T.Helper() 648 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HostShardSets", reflect.TypeOf((*MockStaticOptions)(nil).HostShardSets)) 649 } 650 651 // Replicas mocks base method. 652 func (m *MockStaticOptions) Replicas() int { 653 m.ctrl.T.Helper() 654 ret := m.ctrl.Call(m, "Replicas") 655 ret0, _ := ret[0].(int) 656 return ret0 657 } 658 659 // Replicas indicates an expected call of Replicas. 660 func (mr *MockStaticOptionsMockRecorder) Replicas() *gomock.Call { 661 mr.mock.ctrl.T.Helper() 662 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Replicas", reflect.TypeOf((*MockStaticOptions)(nil).Replicas)) 663 } 664 665 // SetHostShardSets mocks base method. 666 func (m *MockStaticOptions) SetHostShardSets(value []HostShardSet) StaticOptions { 667 m.ctrl.T.Helper() 668 ret := m.ctrl.Call(m, "SetHostShardSets", value) 669 ret0, _ := ret[0].(StaticOptions) 670 return ret0 671 } 672 673 // SetHostShardSets indicates an expected call of SetHostShardSets. 674 func (mr *MockStaticOptionsMockRecorder) SetHostShardSets(value interface{}) *gomock.Call { 675 mr.mock.ctrl.T.Helper() 676 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetHostShardSets", reflect.TypeOf((*MockStaticOptions)(nil).SetHostShardSets), value) 677 } 678 679 // SetReplicas mocks base method. 680 func (m *MockStaticOptions) SetReplicas(value int) StaticOptions { 681 m.ctrl.T.Helper() 682 ret := m.ctrl.Call(m, "SetReplicas", value) 683 ret0, _ := ret[0].(StaticOptions) 684 return ret0 685 } 686 687 // SetReplicas indicates an expected call of SetReplicas. 688 func (mr *MockStaticOptionsMockRecorder) SetReplicas(value interface{}) *gomock.Call { 689 mr.mock.ctrl.T.Helper() 690 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetReplicas", reflect.TypeOf((*MockStaticOptions)(nil).SetReplicas), value) 691 } 692 693 // SetShardSet mocks base method. 694 func (m *MockStaticOptions) SetShardSet(value sharding.ShardSet) StaticOptions { 695 m.ctrl.T.Helper() 696 ret := m.ctrl.Call(m, "SetShardSet", value) 697 ret0, _ := ret[0].(StaticOptions) 698 return ret0 699 } 700 701 // SetShardSet indicates an expected call of SetShardSet. 702 func (mr *MockStaticOptionsMockRecorder) SetShardSet(value interface{}) *gomock.Call { 703 mr.mock.ctrl.T.Helper() 704 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetShardSet", reflect.TypeOf((*MockStaticOptions)(nil).SetShardSet), value) 705 } 706 707 // ShardSet mocks base method. 708 func (m *MockStaticOptions) ShardSet() sharding.ShardSet { 709 m.ctrl.T.Helper() 710 ret := m.ctrl.Call(m, "ShardSet") 711 ret0, _ := ret[0].(sharding.ShardSet) 712 return ret0 713 } 714 715 // ShardSet indicates an expected call of ShardSet. 716 func (mr *MockStaticOptionsMockRecorder) ShardSet() *gomock.Call { 717 mr.mock.ctrl.T.Helper() 718 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ShardSet", reflect.TypeOf((*MockStaticOptions)(nil).ShardSet)) 719 } 720 721 // Validate mocks base method. 722 func (m *MockStaticOptions) Validate() error { 723 m.ctrl.T.Helper() 724 ret := m.ctrl.Call(m, "Validate") 725 ret0, _ := ret[0].(error) 726 return ret0 727 } 728 729 // Validate indicates an expected call of Validate. 730 func (mr *MockStaticOptionsMockRecorder) Validate() *gomock.Call { 731 mr.mock.ctrl.T.Helper() 732 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Validate", reflect.TypeOf((*MockStaticOptions)(nil).Validate)) 733 } 734 735 // MockDynamicOptions is a mock of DynamicOptions interface. 736 type MockDynamicOptions struct { 737 ctrl *gomock.Controller 738 recorder *MockDynamicOptionsMockRecorder 739 } 740 741 // MockDynamicOptionsMockRecorder is the mock recorder for MockDynamicOptions. 742 type MockDynamicOptionsMockRecorder struct { 743 mock *MockDynamicOptions 744 } 745 746 // NewMockDynamicOptions creates a new mock instance. 747 func NewMockDynamicOptions(ctrl *gomock.Controller) *MockDynamicOptions { 748 mock := &MockDynamicOptions{ctrl: ctrl} 749 mock.recorder = &MockDynamicOptionsMockRecorder{mock} 750 return mock 751 } 752 753 // EXPECT returns an object that allows the caller to indicate expected use. 754 func (m *MockDynamicOptions) EXPECT() *MockDynamicOptionsMockRecorder { 755 return m.recorder 756 } 757 758 // ConfigServiceClient mocks base method. 759 func (m *MockDynamicOptions) ConfigServiceClient() client.Client { 760 m.ctrl.T.Helper() 761 ret := m.ctrl.Call(m, "ConfigServiceClient") 762 ret0, _ := ret[0].(client.Client) 763 return ret0 764 } 765 766 // ConfigServiceClient indicates an expected call of ConfigServiceClient. 767 func (mr *MockDynamicOptionsMockRecorder) ConfigServiceClient() *gomock.Call { 768 mr.mock.ctrl.T.Helper() 769 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ConfigServiceClient", reflect.TypeOf((*MockDynamicOptions)(nil).ConfigServiceClient)) 770 } 771 772 // HashGen mocks base method. 773 func (m *MockDynamicOptions) HashGen() sharding.HashGen { 774 m.ctrl.T.Helper() 775 ret := m.ctrl.Call(m, "HashGen") 776 ret0, _ := ret[0].(sharding.HashGen) 777 return ret0 778 } 779 780 // HashGen indicates an expected call of HashGen. 781 func (mr *MockDynamicOptionsMockRecorder) HashGen() *gomock.Call { 782 mr.mock.ctrl.T.Helper() 783 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "HashGen", reflect.TypeOf((*MockDynamicOptions)(nil).HashGen)) 784 } 785 786 // InstrumentOptions mocks base method. 787 func (m *MockDynamicOptions) InstrumentOptions() instrument.Options { 788 m.ctrl.T.Helper() 789 ret := m.ctrl.Call(m, "InstrumentOptions") 790 ret0, _ := ret[0].(instrument.Options) 791 return ret0 792 } 793 794 // InstrumentOptions indicates an expected call of InstrumentOptions. 795 func (mr *MockDynamicOptionsMockRecorder) InstrumentOptions() *gomock.Call { 796 mr.mock.ctrl.T.Helper() 797 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "InstrumentOptions", reflect.TypeOf((*MockDynamicOptions)(nil).InstrumentOptions)) 798 } 799 800 // QueryOptions mocks base method. 801 func (m *MockDynamicOptions) QueryOptions() services.QueryOptions { 802 m.ctrl.T.Helper() 803 ret := m.ctrl.Call(m, "QueryOptions") 804 ret0, _ := ret[0].(services.QueryOptions) 805 return ret0 806 } 807 808 // QueryOptions indicates an expected call of QueryOptions. 809 func (mr *MockDynamicOptionsMockRecorder) QueryOptions() *gomock.Call { 810 mr.mock.ctrl.T.Helper() 811 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "QueryOptions", reflect.TypeOf((*MockDynamicOptions)(nil).QueryOptions)) 812 } 813 814 // ServiceID mocks base method. 815 func (m *MockDynamicOptions) ServiceID() services.ServiceID { 816 m.ctrl.T.Helper() 817 ret := m.ctrl.Call(m, "ServiceID") 818 ret0, _ := ret[0].(services.ServiceID) 819 return ret0 820 } 821 822 // ServiceID indicates an expected call of ServiceID. 823 func (mr *MockDynamicOptionsMockRecorder) ServiceID() *gomock.Call { 824 mr.mock.ctrl.T.Helper() 825 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServiceID", reflect.TypeOf((*MockDynamicOptions)(nil).ServiceID)) 826 } 827 828 // ServicesOverrideOptions mocks base method. 829 func (m *MockDynamicOptions) ServicesOverrideOptions() services.OverrideOptions { 830 m.ctrl.T.Helper() 831 ret := m.ctrl.Call(m, "ServicesOverrideOptions") 832 ret0, _ := ret[0].(services.OverrideOptions) 833 return ret0 834 } 835 836 // ServicesOverrideOptions indicates an expected call of ServicesOverrideOptions. 837 func (mr *MockDynamicOptionsMockRecorder) ServicesOverrideOptions() *gomock.Call { 838 mr.mock.ctrl.T.Helper() 839 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServicesOverrideOptions", reflect.TypeOf((*MockDynamicOptions)(nil).ServicesOverrideOptions)) 840 } 841 842 // SetConfigServiceClient mocks base method. 843 func (m *MockDynamicOptions) SetConfigServiceClient(c client.Client) DynamicOptions { 844 m.ctrl.T.Helper() 845 ret := m.ctrl.Call(m, "SetConfigServiceClient", c) 846 ret0, _ := ret[0].(DynamicOptions) 847 return ret0 848 } 849 850 // SetConfigServiceClient indicates an expected call of SetConfigServiceClient. 851 func (mr *MockDynamicOptionsMockRecorder) SetConfigServiceClient(c interface{}) *gomock.Call { 852 mr.mock.ctrl.T.Helper() 853 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetConfigServiceClient", reflect.TypeOf((*MockDynamicOptions)(nil).SetConfigServiceClient), c) 854 } 855 856 // SetHashGen mocks base method. 857 func (m *MockDynamicOptions) SetHashGen(h sharding.HashGen) DynamicOptions { 858 m.ctrl.T.Helper() 859 ret := m.ctrl.Call(m, "SetHashGen", h) 860 ret0, _ := ret[0].(DynamicOptions) 861 return ret0 862 } 863 864 // SetHashGen indicates an expected call of SetHashGen. 865 func (mr *MockDynamicOptionsMockRecorder) SetHashGen(h interface{}) *gomock.Call { 866 mr.mock.ctrl.T.Helper() 867 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetHashGen", reflect.TypeOf((*MockDynamicOptions)(nil).SetHashGen), h) 868 } 869 870 // SetInstrumentOptions mocks base method. 871 func (m *MockDynamicOptions) SetInstrumentOptions(value instrument.Options) DynamicOptions { 872 m.ctrl.T.Helper() 873 ret := m.ctrl.Call(m, "SetInstrumentOptions", value) 874 ret0, _ := ret[0].(DynamicOptions) 875 return ret0 876 } 877 878 // SetInstrumentOptions indicates an expected call of SetInstrumentOptions. 879 func (mr *MockDynamicOptionsMockRecorder) SetInstrumentOptions(value interface{}) *gomock.Call { 880 mr.mock.ctrl.T.Helper() 881 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetInstrumentOptions", reflect.TypeOf((*MockDynamicOptions)(nil).SetInstrumentOptions), value) 882 } 883 884 // SetQueryOptions mocks base method. 885 func (m *MockDynamicOptions) SetQueryOptions(value services.QueryOptions) DynamicOptions { 886 m.ctrl.T.Helper() 887 ret := m.ctrl.Call(m, "SetQueryOptions", value) 888 ret0, _ := ret[0].(DynamicOptions) 889 return ret0 890 } 891 892 // SetQueryOptions indicates an expected call of SetQueryOptions. 893 func (mr *MockDynamicOptionsMockRecorder) SetQueryOptions(value interface{}) *gomock.Call { 894 mr.mock.ctrl.T.Helper() 895 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetQueryOptions", reflect.TypeOf((*MockDynamicOptions)(nil).SetQueryOptions), value) 896 } 897 898 // SetServiceID mocks base method. 899 func (m *MockDynamicOptions) SetServiceID(s services.ServiceID) DynamicOptions { 900 m.ctrl.T.Helper() 901 ret := m.ctrl.Call(m, "SetServiceID", s) 902 ret0, _ := ret[0].(DynamicOptions) 903 return ret0 904 } 905 906 // SetServiceID indicates an expected call of SetServiceID. 907 func (mr *MockDynamicOptionsMockRecorder) SetServiceID(s interface{}) *gomock.Call { 908 mr.mock.ctrl.T.Helper() 909 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetServiceID", reflect.TypeOf((*MockDynamicOptions)(nil).SetServiceID), s) 910 } 911 912 // SetServicesOverrideOptions mocks base method. 913 func (m *MockDynamicOptions) SetServicesOverrideOptions(opts services.OverrideOptions) DynamicOptions { 914 m.ctrl.T.Helper() 915 ret := m.ctrl.Call(m, "SetServicesOverrideOptions", opts) 916 ret0, _ := ret[0].(DynamicOptions) 917 return ret0 918 } 919 920 // SetServicesOverrideOptions indicates an expected call of SetServicesOverrideOptions. 921 func (mr *MockDynamicOptionsMockRecorder) SetServicesOverrideOptions(opts interface{}) *gomock.Call { 922 mr.mock.ctrl.T.Helper() 923 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetServicesOverrideOptions", reflect.TypeOf((*MockDynamicOptions)(nil).SetServicesOverrideOptions), opts) 924 } 925 926 // Validate mocks base method. 927 func (m *MockDynamicOptions) Validate() error { 928 m.ctrl.T.Helper() 929 ret := m.ctrl.Call(m, "Validate") 930 ret0, _ := ret[0].(error) 931 return ret0 932 } 933 934 // Validate indicates an expected call of Validate. 935 func (mr *MockDynamicOptionsMockRecorder) Validate() *gomock.Call { 936 mr.mock.ctrl.T.Helper() 937 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Validate", reflect.TypeOf((*MockDynamicOptions)(nil).Validate)) 938 } 939 940 // MockMapProvider is a mock of MapProvider interface. 941 type MockMapProvider struct { 942 ctrl *gomock.Controller 943 recorder *MockMapProviderMockRecorder 944 } 945 946 // MockMapProviderMockRecorder is the mock recorder for MockMapProvider. 947 type MockMapProviderMockRecorder struct { 948 mock *MockMapProvider 949 } 950 951 // NewMockMapProvider creates a new mock instance. 952 func NewMockMapProvider(ctrl *gomock.Controller) *MockMapProvider { 953 mock := &MockMapProvider{ctrl: ctrl} 954 mock.recorder = &MockMapProviderMockRecorder{mock} 955 return mock 956 } 957 958 // EXPECT returns an object that allows the caller to indicate expected use. 959 func (m *MockMapProvider) EXPECT() *MockMapProviderMockRecorder { 960 return m.recorder 961 } 962 963 // TopologyMap mocks base method. 964 func (m *MockMapProvider) TopologyMap() (Map, error) { 965 m.ctrl.T.Helper() 966 ret := m.ctrl.Call(m, "TopologyMap") 967 ret0, _ := ret[0].(Map) 968 ret1, _ := ret[1].(error) 969 return ret0, ret1 970 } 971 972 // TopologyMap indicates an expected call of TopologyMap. 973 func (mr *MockMapProviderMockRecorder) TopologyMap() *gomock.Call { 974 mr.mock.ctrl.T.Helper() 975 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TopologyMap", reflect.TypeOf((*MockMapProvider)(nil).TopologyMap)) 976 }