github.com/juju/juju@v0.0.0-20240327075706-a90865de2538/worker/dbaccessor/package_mock_test.go (about) 1 // Code generated by MockGen. DO NOT EDIT. 2 // Source: github.com/juju/juju/worker/dbaccessor (interfaces: Logger,DBApp,NodeManager,TrackedDB,Hub,Client) 3 // 4 // Generated by this command: 5 // 6 // mockgen -package dbaccessor -destination package_mock_test.go github.com/juju/juju/worker/dbaccessor Logger,DBApp,NodeManager,TrackedDB,Hub,Client 7 // 8 9 // Package dbaccessor is a generated GoMock package. 10 package dbaccessor 11 12 import ( 13 context "context" 14 sql "database/sql" 15 reflect "reflect" 16 17 app "github.com/juju/juju/database/app" 18 dqlite "github.com/juju/juju/database/dqlite" 19 loggo "github.com/juju/loggo" 20 gomock "go.uber.org/mock/gomock" 21 ) 22 23 // MockLogger is a mock of Logger interface. 24 type MockLogger struct { 25 ctrl *gomock.Controller 26 recorder *MockLoggerMockRecorder 27 } 28 29 // MockLoggerMockRecorder is the mock recorder for MockLogger. 30 type MockLoggerMockRecorder struct { 31 mock *MockLogger 32 } 33 34 // NewMockLogger creates a new mock instance. 35 func NewMockLogger(ctrl *gomock.Controller) *MockLogger { 36 mock := &MockLogger{ctrl: ctrl} 37 mock.recorder = &MockLoggerMockRecorder{mock} 38 return mock 39 } 40 41 // EXPECT returns an object that allows the caller to indicate expected use. 42 func (m *MockLogger) EXPECT() *MockLoggerMockRecorder { 43 return m.recorder 44 } 45 46 // Debugf mocks base method. 47 func (m *MockLogger) Debugf(arg0 string, arg1 ...any) { 48 m.ctrl.T.Helper() 49 varargs := []any{arg0} 50 for _, a := range arg1 { 51 varargs = append(varargs, a) 52 } 53 m.ctrl.Call(m, "Debugf", varargs...) 54 } 55 56 // Debugf indicates an expected call of Debugf. 57 func (mr *MockLoggerMockRecorder) Debugf(arg0 any, arg1 ...any) *gomock.Call { 58 mr.mock.ctrl.T.Helper() 59 varargs := append([]any{arg0}, arg1...) 60 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Debugf", reflect.TypeOf((*MockLogger)(nil).Debugf), varargs...) 61 } 62 63 // Errorf mocks base method. 64 func (m *MockLogger) Errorf(arg0 string, arg1 ...any) { 65 m.ctrl.T.Helper() 66 varargs := []any{arg0} 67 for _, a := range arg1 { 68 varargs = append(varargs, a) 69 } 70 m.ctrl.Call(m, "Errorf", varargs...) 71 } 72 73 // Errorf indicates an expected call of Errorf. 74 func (mr *MockLoggerMockRecorder) Errorf(arg0 any, arg1 ...any) *gomock.Call { 75 mr.mock.ctrl.T.Helper() 76 varargs := append([]any{arg0}, arg1...) 77 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Errorf", reflect.TypeOf((*MockLogger)(nil).Errorf), varargs...) 78 } 79 80 // Infof mocks base method. 81 func (m *MockLogger) Infof(arg0 string, arg1 ...any) { 82 m.ctrl.T.Helper() 83 varargs := []any{arg0} 84 for _, a := range arg1 { 85 varargs = append(varargs, a) 86 } 87 m.ctrl.Call(m, "Infof", varargs...) 88 } 89 90 // Infof indicates an expected call of Infof. 91 func (mr *MockLoggerMockRecorder) Infof(arg0 any, arg1 ...any) *gomock.Call { 92 mr.mock.ctrl.T.Helper() 93 varargs := append([]any{arg0}, arg1...) 94 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Infof", reflect.TypeOf((*MockLogger)(nil).Infof), varargs...) 95 } 96 97 // IsTraceEnabled mocks base method. 98 func (m *MockLogger) IsTraceEnabled() bool { 99 m.ctrl.T.Helper() 100 ret := m.ctrl.Call(m, "IsTraceEnabled") 101 ret0, _ := ret[0].(bool) 102 return ret0 103 } 104 105 // IsTraceEnabled indicates an expected call of IsTraceEnabled. 106 func (mr *MockLoggerMockRecorder) IsTraceEnabled() *gomock.Call { 107 mr.mock.ctrl.T.Helper() 108 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsTraceEnabled", reflect.TypeOf((*MockLogger)(nil).IsTraceEnabled)) 109 } 110 111 // Logf mocks base method. 112 func (m *MockLogger) Logf(arg0 loggo.Level, arg1 string, arg2 ...any) { 113 m.ctrl.T.Helper() 114 varargs := []any{arg0, arg1} 115 for _, a := range arg2 { 116 varargs = append(varargs, a) 117 } 118 m.ctrl.Call(m, "Logf", varargs...) 119 } 120 121 // Logf indicates an expected call of Logf. 122 func (mr *MockLoggerMockRecorder) Logf(arg0, arg1 any, arg2 ...any) *gomock.Call { 123 mr.mock.ctrl.T.Helper() 124 varargs := append([]any{arg0, arg1}, arg2...) 125 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Logf", reflect.TypeOf((*MockLogger)(nil).Logf), varargs...) 126 } 127 128 // Tracef mocks base method. 129 func (m *MockLogger) Tracef(arg0 string, arg1 ...any) { 130 m.ctrl.T.Helper() 131 varargs := []any{arg0} 132 for _, a := range arg1 { 133 varargs = append(varargs, a) 134 } 135 m.ctrl.Call(m, "Tracef", varargs...) 136 } 137 138 // Tracef indicates an expected call of Tracef. 139 func (mr *MockLoggerMockRecorder) Tracef(arg0 any, arg1 ...any) *gomock.Call { 140 mr.mock.ctrl.T.Helper() 141 varargs := append([]any{arg0}, arg1...) 142 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Tracef", reflect.TypeOf((*MockLogger)(nil).Tracef), varargs...) 143 } 144 145 // Warningf mocks base method. 146 func (m *MockLogger) Warningf(arg0 string, arg1 ...any) { 147 m.ctrl.T.Helper() 148 varargs := []any{arg0} 149 for _, a := range arg1 { 150 varargs = append(varargs, a) 151 } 152 m.ctrl.Call(m, "Warningf", varargs...) 153 } 154 155 // Warningf indicates an expected call of Warningf. 156 func (mr *MockLoggerMockRecorder) Warningf(arg0 any, arg1 ...any) *gomock.Call { 157 mr.mock.ctrl.T.Helper() 158 varargs := append([]any{arg0}, arg1...) 159 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Warningf", reflect.TypeOf((*MockLogger)(nil).Warningf), varargs...) 160 } 161 162 // MockDBApp is a mock of DBApp interface. 163 type MockDBApp struct { 164 ctrl *gomock.Controller 165 recorder *MockDBAppMockRecorder 166 } 167 168 // MockDBAppMockRecorder is the mock recorder for MockDBApp. 169 type MockDBAppMockRecorder struct { 170 mock *MockDBApp 171 } 172 173 // NewMockDBApp creates a new mock instance. 174 func NewMockDBApp(ctrl *gomock.Controller) *MockDBApp { 175 mock := &MockDBApp{ctrl: ctrl} 176 mock.recorder = &MockDBAppMockRecorder{mock} 177 return mock 178 } 179 180 // EXPECT returns an object that allows the caller to indicate expected use. 181 func (m *MockDBApp) EXPECT() *MockDBAppMockRecorder { 182 return m.recorder 183 } 184 185 // Client mocks base method. 186 func (m *MockDBApp) Client(arg0 context.Context) (Client, error) { 187 m.ctrl.T.Helper() 188 ret := m.ctrl.Call(m, "Client", arg0) 189 ret0, _ := ret[0].(Client) 190 ret1, _ := ret[1].(error) 191 return ret0, ret1 192 } 193 194 // Client indicates an expected call of Client. 195 func (mr *MockDBAppMockRecorder) Client(arg0 any) *gomock.Call { 196 mr.mock.ctrl.T.Helper() 197 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Client", reflect.TypeOf((*MockDBApp)(nil).Client), arg0) 198 } 199 200 // Close mocks base method. 201 func (m *MockDBApp) Close() error { 202 m.ctrl.T.Helper() 203 ret := m.ctrl.Call(m, "Close") 204 ret0, _ := ret[0].(error) 205 return ret0 206 } 207 208 // Close indicates an expected call of Close. 209 func (mr *MockDBAppMockRecorder) Close() *gomock.Call { 210 mr.mock.ctrl.T.Helper() 211 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockDBApp)(nil).Close)) 212 } 213 214 // Handover mocks base method. 215 func (m *MockDBApp) Handover(arg0 context.Context) error { 216 m.ctrl.T.Helper() 217 ret := m.ctrl.Call(m, "Handover", arg0) 218 ret0, _ := ret[0].(error) 219 return ret0 220 } 221 222 // Handover indicates an expected call of Handover. 223 func (mr *MockDBAppMockRecorder) Handover(arg0 any) *gomock.Call { 224 mr.mock.ctrl.T.Helper() 225 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Handover", reflect.TypeOf((*MockDBApp)(nil).Handover), arg0) 226 } 227 228 // ID mocks base method. 229 func (m *MockDBApp) ID() uint64 { 230 m.ctrl.T.Helper() 231 ret := m.ctrl.Call(m, "ID") 232 ret0, _ := ret[0].(uint64) 233 return ret0 234 } 235 236 // ID indicates an expected call of ID. 237 func (mr *MockDBAppMockRecorder) ID() *gomock.Call { 238 mr.mock.ctrl.T.Helper() 239 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ID", reflect.TypeOf((*MockDBApp)(nil).ID)) 240 } 241 242 // Open mocks base method. 243 func (m *MockDBApp) Open(arg0 context.Context, arg1 string) (*sql.DB, error) { 244 m.ctrl.T.Helper() 245 ret := m.ctrl.Call(m, "Open", arg0, arg1) 246 ret0, _ := ret[0].(*sql.DB) 247 ret1, _ := ret[1].(error) 248 return ret0, ret1 249 } 250 251 // Open indicates an expected call of Open. 252 func (mr *MockDBAppMockRecorder) Open(arg0, arg1 any) *gomock.Call { 253 mr.mock.ctrl.T.Helper() 254 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Open", reflect.TypeOf((*MockDBApp)(nil).Open), arg0, arg1) 255 } 256 257 // Ready mocks base method. 258 func (m *MockDBApp) Ready(arg0 context.Context) error { 259 m.ctrl.T.Helper() 260 ret := m.ctrl.Call(m, "Ready", arg0) 261 ret0, _ := ret[0].(error) 262 return ret0 263 } 264 265 // Ready indicates an expected call of Ready. 266 func (mr *MockDBAppMockRecorder) Ready(arg0 any) *gomock.Call { 267 mr.mock.ctrl.T.Helper() 268 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Ready", reflect.TypeOf((*MockDBApp)(nil).Ready), arg0) 269 } 270 271 // MockNodeManager is a mock of NodeManager interface. 272 type MockNodeManager struct { 273 ctrl *gomock.Controller 274 recorder *MockNodeManagerMockRecorder 275 } 276 277 // MockNodeManagerMockRecorder is the mock recorder for MockNodeManager. 278 type MockNodeManagerMockRecorder struct { 279 mock *MockNodeManager 280 } 281 282 // NewMockNodeManager creates a new mock instance. 283 func NewMockNodeManager(ctrl *gomock.Controller) *MockNodeManager { 284 mock := &MockNodeManager{ctrl: ctrl} 285 mock.recorder = &MockNodeManagerMockRecorder{mock} 286 return mock 287 } 288 289 // EXPECT returns an object that allows the caller to indicate expected use. 290 func (m *MockNodeManager) EXPECT() *MockNodeManagerMockRecorder { 291 return m.recorder 292 } 293 294 // ClusterServers mocks base method. 295 func (m *MockNodeManager) ClusterServers(arg0 context.Context) ([]dqlite.NodeInfo, error) { 296 m.ctrl.T.Helper() 297 ret := m.ctrl.Call(m, "ClusterServers", arg0) 298 ret0, _ := ret[0].([]dqlite.NodeInfo) 299 ret1, _ := ret[1].(error) 300 return ret0, ret1 301 } 302 303 // ClusterServers indicates an expected call of ClusterServers. 304 func (mr *MockNodeManagerMockRecorder) ClusterServers(arg0 any) *gomock.Call { 305 mr.mock.ctrl.T.Helper() 306 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ClusterServers", reflect.TypeOf((*MockNodeManager)(nil).ClusterServers), arg0) 307 } 308 309 // EnsureDataDir mocks base method. 310 func (m *MockNodeManager) EnsureDataDir() (string, error) { 311 m.ctrl.T.Helper() 312 ret := m.ctrl.Call(m, "EnsureDataDir") 313 ret0, _ := ret[0].(string) 314 ret1, _ := ret[1].(error) 315 return ret0, ret1 316 } 317 318 // EnsureDataDir indicates an expected call of EnsureDataDir. 319 func (mr *MockNodeManagerMockRecorder) EnsureDataDir() *gomock.Call { 320 mr.mock.ctrl.T.Helper() 321 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "EnsureDataDir", reflect.TypeOf((*MockNodeManager)(nil).EnsureDataDir)) 322 } 323 324 // IsExistingNode mocks base method. 325 func (m *MockNodeManager) IsExistingNode() (bool, error) { 326 m.ctrl.T.Helper() 327 ret := m.ctrl.Call(m, "IsExistingNode") 328 ret0, _ := ret[0].(bool) 329 ret1, _ := ret[1].(error) 330 return ret0, ret1 331 } 332 333 // IsExistingNode indicates an expected call of IsExistingNode. 334 func (mr *MockNodeManagerMockRecorder) IsExistingNode() *gomock.Call { 335 mr.mock.ctrl.T.Helper() 336 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsExistingNode", reflect.TypeOf((*MockNodeManager)(nil).IsExistingNode)) 337 } 338 339 // IsLoopbackBound mocks base method. 340 func (m *MockNodeManager) IsLoopbackBound(arg0 context.Context) (bool, error) { 341 m.ctrl.T.Helper() 342 ret := m.ctrl.Call(m, "IsLoopbackBound", arg0) 343 ret0, _ := ret[0].(bool) 344 ret1, _ := ret[1].(error) 345 return ret0, ret1 346 } 347 348 // IsLoopbackBound indicates an expected call of IsLoopbackBound. 349 func (mr *MockNodeManagerMockRecorder) IsLoopbackBound(arg0 any) *gomock.Call { 350 mr.mock.ctrl.T.Helper() 351 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsLoopbackBound", reflect.TypeOf((*MockNodeManager)(nil).IsLoopbackBound), arg0) 352 } 353 354 // IsLoopbackPreferred mocks base method. 355 func (m *MockNodeManager) IsLoopbackPreferred() bool { 356 m.ctrl.T.Helper() 357 ret := m.ctrl.Call(m, "IsLoopbackPreferred") 358 ret0, _ := ret[0].(bool) 359 return ret0 360 } 361 362 // IsLoopbackPreferred indicates an expected call of IsLoopbackPreferred. 363 func (mr *MockNodeManagerMockRecorder) IsLoopbackPreferred() *gomock.Call { 364 mr.mock.ctrl.T.Helper() 365 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsLoopbackPreferred", reflect.TypeOf((*MockNodeManager)(nil).IsLoopbackPreferred)) 366 } 367 368 // SetClusterServers mocks base method. 369 func (m *MockNodeManager) SetClusterServers(arg0 context.Context, arg1 []dqlite.NodeInfo) error { 370 m.ctrl.T.Helper() 371 ret := m.ctrl.Call(m, "SetClusterServers", arg0, arg1) 372 ret0, _ := ret[0].(error) 373 return ret0 374 } 375 376 // SetClusterServers indicates an expected call of SetClusterServers. 377 func (mr *MockNodeManagerMockRecorder) SetClusterServers(arg0, arg1 any) *gomock.Call { 378 mr.mock.ctrl.T.Helper() 379 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetClusterServers", reflect.TypeOf((*MockNodeManager)(nil).SetClusterServers), arg0, arg1) 380 } 381 382 // SetClusterToLocalNode mocks base method. 383 func (m *MockNodeManager) SetClusterToLocalNode(arg0 context.Context) error { 384 m.ctrl.T.Helper() 385 ret := m.ctrl.Call(m, "SetClusterToLocalNode", arg0) 386 ret0, _ := ret[0].(error) 387 return ret0 388 } 389 390 // SetClusterToLocalNode indicates an expected call of SetClusterToLocalNode. 391 func (mr *MockNodeManagerMockRecorder) SetClusterToLocalNode(arg0 any) *gomock.Call { 392 mr.mock.ctrl.T.Helper() 393 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetClusterToLocalNode", reflect.TypeOf((*MockNodeManager)(nil).SetClusterToLocalNode), arg0) 394 } 395 396 // SetNodeInfo mocks base method. 397 func (m *MockNodeManager) SetNodeInfo(arg0 dqlite.NodeInfo) error { 398 m.ctrl.T.Helper() 399 ret := m.ctrl.Call(m, "SetNodeInfo", arg0) 400 ret0, _ := ret[0].(error) 401 return ret0 402 } 403 404 // SetNodeInfo indicates an expected call of SetNodeInfo. 405 func (mr *MockNodeManagerMockRecorder) SetNodeInfo(arg0 any) *gomock.Call { 406 mr.mock.ctrl.T.Helper() 407 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetNodeInfo", reflect.TypeOf((*MockNodeManager)(nil).SetNodeInfo), arg0) 408 } 409 410 // WithAddressOption mocks base method. 411 func (m *MockNodeManager) WithAddressOption(arg0 string) app.Option { 412 m.ctrl.T.Helper() 413 ret := m.ctrl.Call(m, "WithAddressOption", arg0) 414 ret0, _ := ret[0].(app.Option) 415 return ret0 416 } 417 418 // WithAddressOption indicates an expected call of WithAddressOption. 419 func (mr *MockNodeManagerMockRecorder) WithAddressOption(arg0 any) *gomock.Call { 420 mr.mock.ctrl.T.Helper() 421 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WithAddressOption", reflect.TypeOf((*MockNodeManager)(nil).WithAddressOption), arg0) 422 } 423 424 // WithClusterOption mocks base method. 425 func (m *MockNodeManager) WithClusterOption(arg0 []string) app.Option { 426 m.ctrl.T.Helper() 427 ret := m.ctrl.Call(m, "WithClusterOption", arg0) 428 ret0, _ := ret[0].(app.Option) 429 return ret0 430 } 431 432 // WithClusterOption indicates an expected call of WithClusterOption. 433 func (mr *MockNodeManagerMockRecorder) WithClusterOption(arg0 any) *gomock.Call { 434 mr.mock.ctrl.T.Helper() 435 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WithClusterOption", reflect.TypeOf((*MockNodeManager)(nil).WithClusterOption), arg0) 436 } 437 438 // WithLogFuncOption mocks base method. 439 func (m *MockNodeManager) WithLogFuncOption() app.Option { 440 m.ctrl.T.Helper() 441 ret := m.ctrl.Call(m, "WithLogFuncOption") 442 ret0, _ := ret[0].(app.Option) 443 return ret0 444 } 445 446 // WithLogFuncOption indicates an expected call of WithLogFuncOption. 447 func (mr *MockNodeManagerMockRecorder) WithLogFuncOption() *gomock.Call { 448 mr.mock.ctrl.T.Helper() 449 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WithLogFuncOption", reflect.TypeOf((*MockNodeManager)(nil).WithLogFuncOption)) 450 } 451 452 // WithTLSOption mocks base method. 453 func (m *MockNodeManager) WithTLSOption() (app.Option, error) { 454 m.ctrl.T.Helper() 455 ret := m.ctrl.Call(m, "WithTLSOption") 456 ret0, _ := ret[0].(app.Option) 457 ret1, _ := ret[1].(error) 458 return ret0, ret1 459 } 460 461 // WithTLSOption indicates an expected call of WithTLSOption. 462 func (mr *MockNodeManagerMockRecorder) WithTLSOption() *gomock.Call { 463 mr.mock.ctrl.T.Helper() 464 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WithTLSOption", reflect.TypeOf((*MockNodeManager)(nil).WithTLSOption)) 465 } 466 467 // WithTracingOption mocks base method. 468 func (m *MockNodeManager) WithTracingOption() app.Option { 469 m.ctrl.T.Helper() 470 ret := m.ctrl.Call(m, "WithTracingOption") 471 ret0, _ := ret[0].(app.Option) 472 return ret0 473 } 474 475 // WithTracingOption indicates an expected call of WithTracingOption. 476 func (mr *MockNodeManagerMockRecorder) WithTracingOption() *gomock.Call { 477 mr.mock.ctrl.T.Helper() 478 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WithTracingOption", reflect.TypeOf((*MockNodeManager)(nil).WithTracingOption)) 479 } 480 481 // MockTrackedDB is a mock of TrackedDB interface. 482 type MockTrackedDB struct { 483 ctrl *gomock.Controller 484 recorder *MockTrackedDBMockRecorder 485 } 486 487 // MockTrackedDBMockRecorder is the mock recorder for MockTrackedDB. 488 type MockTrackedDBMockRecorder struct { 489 mock *MockTrackedDB 490 } 491 492 // NewMockTrackedDB creates a new mock instance. 493 func NewMockTrackedDB(ctrl *gomock.Controller) *MockTrackedDB { 494 mock := &MockTrackedDB{ctrl: ctrl} 495 mock.recorder = &MockTrackedDBMockRecorder{mock} 496 return mock 497 } 498 499 // EXPECT returns an object that allows the caller to indicate expected use. 500 func (m *MockTrackedDB) EXPECT() *MockTrackedDBMockRecorder { 501 return m.recorder 502 } 503 504 // Err mocks base method. 505 func (m *MockTrackedDB) Err() error { 506 m.ctrl.T.Helper() 507 ret := m.ctrl.Call(m, "Err") 508 ret0, _ := ret[0].(error) 509 return ret0 510 } 511 512 // Err indicates an expected call of Err. 513 func (mr *MockTrackedDBMockRecorder) Err() *gomock.Call { 514 mr.mock.ctrl.T.Helper() 515 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Err", reflect.TypeOf((*MockTrackedDB)(nil).Err)) 516 } 517 518 // Kill mocks base method. 519 func (m *MockTrackedDB) Kill() { 520 m.ctrl.T.Helper() 521 m.ctrl.Call(m, "Kill") 522 } 523 524 // Kill indicates an expected call of Kill. 525 func (mr *MockTrackedDBMockRecorder) Kill() *gomock.Call { 526 mr.mock.ctrl.T.Helper() 527 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Kill", reflect.TypeOf((*MockTrackedDB)(nil).Kill)) 528 } 529 530 // Txn mocks base method. 531 func (m *MockTrackedDB) Txn(arg0 context.Context, arg1 func(context.Context, *sql.Tx) error) error { 532 m.ctrl.T.Helper() 533 ret := m.ctrl.Call(m, "Txn", arg0, arg1) 534 ret0, _ := ret[0].(error) 535 return ret0 536 } 537 538 // Txn indicates an expected call of Txn. 539 func (mr *MockTrackedDBMockRecorder) Txn(arg0, arg1 any) *gomock.Call { 540 mr.mock.ctrl.T.Helper() 541 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Txn", reflect.TypeOf((*MockTrackedDB)(nil).Txn), arg0, arg1) 542 } 543 544 // TxnNoRetry mocks base method. 545 func (m *MockTrackedDB) TxnNoRetry(arg0 context.Context, arg1 func(context.Context, *sql.Tx) error) error { 546 m.ctrl.T.Helper() 547 ret := m.ctrl.Call(m, "TxnNoRetry", arg0, arg1) 548 ret0, _ := ret[0].(error) 549 return ret0 550 } 551 552 // TxnNoRetry indicates an expected call of TxnNoRetry. 553 func (mr *MockTrackedDBMockRecorder) TxnNoRetry(arg0, arg1 any) *gomock.Call { 554 mr.mock.ctrl.T.Helper() 555 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TxnNoRetry", reflect.TypeOf((*MockTrackedDB)(nil).TxnNoRetry), arg0, arg1) 556 } 557 558 // Wait mocks base method. 559 func (m *MockTrackedDB) Wait() error { 560 m.ctrl.T.Helper() 561 ret := m.ctrl.Call(m, "Wait") 562 ret0, _ := ret[0].(error) 563 return ret0 564 } 565 566 // Wait indicates an expected call of Wait. 567 func (mr *MockTrackedDBMockRecorder) Wait() *gomock.Call { 568 mr.mock.ctrl.T.Helper() 569 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Wait", reflect.TypeOf((*MockTrackedDB)(nil).Wait)) 570 } 571 572 // MockHub is a mock of Hub interface. 573 type MockHub struct { 574 ctrl *gomock.Controller 575 recorder *MockHubMockRecorder 576 } 577 578 // MockHubMockRecorder is the mock recorder for MockHub. 579 type MockHubMockRecorder struct { 580 mock *MockHub 581 } 582 583 // NewMockHub creates a new mock instance. 584 func NewMockHub(ctrl *gomock.Controller) *MockHub { 585 mock := &MockHub{ctrl: ctrl} 586 mock.recorder = &MockHubMockRecorder{mock} 587 return mock 588 } 589 590 // EXPECT returns an object that allows the caller to indicate expected use. 591 func (m *MockHub) EXPECT() *MockHubMockRecorder { 592 return m.recorder 593 } 594 595 // Publish mocks base method. 596 func (m *MockHub) Publish(arg0 string, arg1 any) (func(), error) { 597 m.ctrl.T.Helper() 598 ret := m.ctrl.Call(m, "Publish", arg0, arg1) 599 ret0, _ := ret[0].(func()) 600 ret1, _ := ret[1].(error) 601 return ret0, ret1 602 } 603 604 // Publish indicates an expected call of Publish. 605 func (mr *MockHubMockRecorder) Publish(arg0, arg1 any) *gomock.Call { 606 mr.mock.ctrl.T.Helper() 607 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Publish", reflect.TypeOf((*MockHub)(nil).Publish), arg0, arg1) 608 } 609 610 // Subscribe mocks base method. 611 func (m *MockHub) Subscribe(arg0 string, arg1 any) (func(), error) { 612 m.ctrl.T.Helper() 613 ret := m.ctrl.Call(m, "Subscribe", arg0, arg1) 614 ret0, _ := ret[0].(func()) 615 ret1, _ := ret[1].(error) 616 return ret0, ret1 617 } 618 619 // Subscribe indicates an expected call of Subscribe. 620 func (mr *MockHubMockRecorder) Subscribe(arg0, arg1 any) *gomock.Call { 621 mr.mock.ctrl.T.Helper() 622 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Subscribe", reflect.TypeOf((*MockHub)(nil).Subscribe), arg0, arg1) 623 } 624 625 // MockClient is a mock of Client interface. 626 type MockClient struct { 627 ctrl *gomock.Controller 628 recorder *MockClientMockRecorder 629 } 630 631 // MockClientMockRecorder is the mock recorder for MockClient. 632 type MockClientMockRecorder struct { 633 mock *MockClient 634 } 635 636 // NewMockClient creates a new mock instance. 637 func NewMockClient(ctrl *gomock.Controller) *MockClient { 638 mock := &MockClient{ctrl: ctrl} 639 mock.recorder = &MockClientMockRecorder{mock} 640 return mock 641 } 642 643 // EXPECT returns an object that allows the caller to indicate expected use. 644 func (m *MockClient) EXPECT() *MockClientMockRecorder { 645 return m.recorder 646 } 647 648 // Cluster mocks base method. 649 func (m *MockClient) Cluster(arg0 context.Context) ([]dqlite.NodeInfo, error) { 650 m.ctrl.T.Helper() 651 ret := m.ctrl.Call(m, "Cluster", arg0) 652 ret0, _ := ret[0].([]dqlite.NodeInfo) 653 ret1, _ := ret[1].(error) 654 return ret0, ret1 655 } 656 657 // Cluster indicates an expected call of Cluster. 658 func (mr *MockClientMockRecorder) Cluster(arg0 any) *gomock.Call { 659 mr.mock.ctrl.T.Helper() 660 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Cluster", reflect.TypeOf((*MockClient)(nil).Cluster), arg0) 661 } 662 663 // Leader mocks base method. 664 func (m *MockClient) Leader(arg0 context.Context) (*dqlite.NodeInfo, error) { 665 m.ctrl.T.Helper() 666 ret := m.ctrl.Call(m, "Leader", arg0) 667 ret0, _ := ret[0].(*dqlite.NodeInfo) 668 ret1, _ := ret[1].(error) 669 return ret0, ret1 670 } 671 672 // Leader indicates an expected call of Leader. 673 func (mr *MockClientMockRecorder) Leader(arg0 any) *gomock.Call { 674 mr.mock.ctrl.T.Helper() 675 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Leader", reflect.TypeOf((*MockClient)(nil).Leader), arg0) 676 }