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 }