github.com/matrixorigin/matrixone@v1.2.0/pkg/frontend/test/txn_mock.go (about) 1 // Code generated by MockGen. DO NOT EDIT. 2 // Source: ../../../pkg/txn/client/types.go 3 4 // Package mock_frontend is a generated GoMock package. 5 package mock_frontend 6 7 import ( 8 context "context" 9 reflect "reflect" 10 time "time" 11 12 gomock "github.com/golang/mock/gomock" 13 lock "github.com/matrixorigin/matrixone/pkg/pb/lock" 14 timestamp "github.com/matrixorigin/matrixone/pkg/pb/timestamp" 15 txn "github.com/matrixorigin/matrixone/pkg/pb/txn" 16 client "github.com/matrixorigin/matrixone/pkg/txn/client" 17 rpc "github.com/matrixorigin/matrixone/pkg/txn/rpc" 18 ) 19 20 // MockTxnTimestampAware is a mock of TxnTimestampAware interface. 21 type MockTxnTimestampAware struct { 22 ctrl *gomock.Controller 23 recorder *MockTxnTimestampAwareMockRecorder 24 } 25 26 // MockTxnTimestampAwareMockRecorder is the mock recorder for MockTxnTimestampAware. 27 type MockTxnTimestampAwareMockRecorder struct { 28 mock *MockTxnTimestampAware 29 } 30 31 // NewMockTxnTimestampAware creates a new mock instance. 32 func NewMockTxnTimestampAware(ctrl *gomock.Controller) *MockTxnTimestampAware { 33 mock := &MockTxnTimestampAware{ctrl: ctrl} 34 mock.recorder = &MockTxnTimestampAwareMockRecorder{mock} 35 return mock 36 } 37 38 // EXPECT returns an object that allows the caller to indicate expected use. 39 func (m *MockTxnTimestampAware) EXPECT() *MockTxnTimestampAwareMockRecorder { 40 return m.recorder 41 } 42 43 // GetLatestCommitTS mocks base method. 44 func (m *MockTxnTimestampAware) GetLatestCommitTS() timestamp.Timestamp { 45 m.ctrl.T.Helper() 46 ret := m.ctrl.Call(m, "GetLatestCommitTS") 47 ret0, _ := ret[0].(timestamp.Timestamp) 48 return ret0 49 } 50 51 // GetLatestCommitTS indicates an expected call of GetLatestCommitTS. 52 func (mr *MockTxnTimestampAwareMockRecorder) GetLatestCommitTS() *gomock.Call { 53 mr.mock.ctrl.T.Helper() 54 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLatestCommitTS", reflect.TypeOf((*MockTxnTimestampAware)(nil).GetLatestCommitTS)) 55 } 56 57 // GetSyncLatestCommitTSTimes mocks base method. 58 func (m *MockTxnTimestampAware) GetSyncLatestCommitTSTimes() uint64 { 59 m.ctrl.T.Helper() 60 ret := m.ctrl.Call(m, "GetSyncLatestCommitTSTimes") 61 ret0, _ := ret[0].(uint64) 62 return ret0 63 } 64 65 // GetSyncLatestCommitTSTimes indicates an expected call of GetSyncLatestCommitTSTimes. 66 func (mr *MockTxnTimestampAwareMockRecorder) GetSyncLatestCommitTSTimes() *gomock.Call { 67 mr.mock.ctrl.T.Helper() 68 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSyncLatestCommitTSTimes", reflect.TypeOf((*MockTxnTimestampAware)(nil).GetSyncLatestCommitTSTimes)) 69 } 70 71 // MinTimestamp mocks base method. 72 func (m *MockTxnTimestampAware) MinTimestamp() timestamp.Timestamp { 73 m.ctrl.T.Helper() 74 ret := m.ctrl.Call(m, "MinTimestamp") 75 ret0, _ := ret[0].(timestamp.Timestamp) 76 return ret0 77 } 78 79 // MinTimestamp indicates an expected call of MinTimestamp. 80 func (mr *MockTxnTimestampAwareMockRecorder) MinTimestamp() *gomock.Call { 81 mr.mock.ctrl.T.Helper() 82 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MinTimestamp", reflect.TypeOf((*MockTxnTimestampAware)(nil).MinTimestamp)) 83 } 84 85 // SyncLatestCommitTS mocks base method. 86 func (m *MockTxnTimestampAware) SyncLatestCommitTS(arg0 timestamp.Timestamp) { 87 m.ctrl.T.Helper() 88 m.ctrl.Call(m, "SyncLatestCommitTS", arg0) 89 } 90 91 // SyncLatestCommitTS indicates an expected call of SyncLatestCommitTS. 92 func (mr *MockTxnTimestampAwareMockRecorder) SyncLatestCommitTS(arg0 interface{}) *gomock.Call { 93 mr.mock.ctrl.T.Helper() 94 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SyncLatestCommitTS", reflect.TypeOf((*MockTxnTimestampAware)(nil).SyncLatestCommitTS), arg0) 95 } 96 97 // WaitLogTailAppliedAt mocks base method. 98 func (m *MockTxnTimestampAware) WaitLogTailAppliedAt(ctx context.Context, ts timestamp.Timestamp) (timestamp.Timestamp, error) { 99 m.ctrl.T.Helper() 100 ret := m.ctrl.Call(m, "WaitLogTailAppliedAt", ctx, ts) 101 ret0, _ := ret[0].(timestamp.Timestamp) 102 ret1, _ := ret[1].(error) 103 return ret0, ret1 104 } 105 106 // WaitLogTailAppliedAt indicates an expected call of WaitLogTailAppliedAt. 107 func (mr *MockTxnTimestampAwareMockRecorder) WaitLogTailAppliedAt(ctx, ts interface{}) *gomock.Call { 108 mr.mock.ctrl.T.Helper() 109 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WaitLogTailAppliedAt", reflect.TypeOf((*MockTxnTimestampAware)(nil).WaitLogTailAppliedAt), ctx, ts) 110 } 111 112 // MockTxnAction is a mock of TxnAction interface. 113 type MockTxnAction struct { 114 ctrl *gomock.Controller 115 recorder *MockTxnActionMockRecorder 116 } 117 118 // MockTxnActionMockRecorder is the mock recorder for MockTxnAction. 119 type MockTxnActionMockRecorder struct { 120 mock *MockTxnAction 121 } 122 123 // NewMockTxnAction creates a new mock instance. 124 func NewMockTxnAction(ctrl *gomock.Controller) *MockTxnAction { 125 mock := &MockTxnAction{ctrl: ctrl} 126 mock.recorder = &MockTxnActionMockRecorder{mock} 127 return mock 128 } 129 130 // EXPECT returns an object that allows the caller to indicate expected use. 131 func (m *MockTxnAction) EXPECT() *MockTxnActionMockRecorder { 132 return m.recorder 133 } 134 135 // AbortAllRunningTxn mocks base method. 136 func (m *MockTxnAction) AbortAllRunningTxn() { 137 m.ctrl.T.Helper() 138 m.ctrl.Call(m, "AbortAllRunningTxn") 139 } 140 141 // AbortAllRunningTxn indicates an expected call of AbortAllRunningTxn. 142 func (mr *MockTxnActionMockRecorder) AbortAllRunningTxn() *gomock.Call { 143 mr.mock.ctrl.T.Helper() 144 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AbortAllRunningTxn", reflect.TypeOf((*MockTxnAction)(nil).AbortAllRunningTxn)) 145 } 146 147 // Pause mocks base method. 148 func (m *MockTxnAction) Pause() { 149 m.ctrl.T.Helper() 150 m.ctrl.Call(m, "Pause") 151 } 152 153 // Pause indicates an expected call of Pause. 154 func (mr *MockTxnActionMockRecorder) Pause() *gomock.Call { 155 mr.mock.ctrl.T.Helper() 156 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Pause", reflect.TypeOf((*MockTxnAction)(nil).Pause)) 157 } 158 159 // Resume mocks base method. 160 func (m *MockTxnAction) Resume() { 161 m.ctrl.T.Helper() 162 m.ctrl.Call(m, "Resume") 163 } 164 165 // Resume indicates an expected call of Resume. 166 func (mr *MockTxnActionMockRecorder) Resume() *gomock.Call { 167 mr.mock.ctrl.T.Helper() 168 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Resume", reflect.TypeOf((*MockTxnAction)(nil).Resume)) 169 } 170 171 // MockTxnClient is a mock of TxnClient interface. 172 type MockTxnClient struct { 173 ctrl *gomock.Controller 174 recorder *MockTxnClientMockRecorder 175 } 176 177 // MockTxnClientMockRecorder is the mock recorder for MockTxnClient. 178 type MockTxnClientMockRecorder struct { 179 mock *MockTxnClient 180 } 181 182 // NewMockTxnClient creates a new mock instance. 183 func NewMockTxnClient(ctrl *gomock.Controller) *MockTxnClient { 184 mock := &MockTxnClient{ctrl: ctrl} 185 mock.recorder = &MockTxnClientMockRecorder{mock} 186 return mock 187 } 188 189 // EXPECT returns an object that allows the caller to indicate expected use. 190 func (m *MockTxnClient) EXPECT() *MockTxnClientMockRecorder { 191 return m.recorder 192 } 193 194 // AbortAllRunningTxn mocks base method. 195 func (m *MockTxnClient) AbortAllRunningTxn() { 196 m.ctrl.T.Helper() 197 m.ctrl.Call(m, "AbortAllRunningTxn") 198 } 199 200 // AbortAllRunningTxn indicates an expected call of AbortAllRunningTxn. 201 func (mr *MockTxnClientMockRecorder) AbortAllRunningTxn() *gomock.Call { 202 mr.mock.ctrl.T.Helper() 203 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AbortAllRunningTxn", reflect.TypeOf((*MockTxnClient)(nil).AbortAllRunningTxn)) 204 } 205 206 // CNBasedConsistencyEnabled mocks base method. 207 func (m *MockTxnClient) CNBasedConsistencyEnabled() bool { 208 m.ctrl.T.Helper() 209 ret := m.ctrl.Call(m, "CNBasedConsistencyEnabled") 210 ret0, _ := ret[0].(bool) 211 return ret0 212 } 213 214 // CNBasedConsistencyEnabled indicates an expected call of CNBasedConsistencyEnabled. 215 func (mr *MockTxnClientMockRecorder) CNBasedConsistencyEnabled() *gomock.Call { 216 mr.mock.ctrl.T.Helper() 217 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CNBasedConsistencyEnabled", reflect.TypeOf((*MockTxnClient)(nil).CNBasedConsistencyEnabled)) 218 } 219 220 // Close mocks base method. 221 func (m *MockTxnClient) Close() error { 222 m.ctrl.T.Helper() 223 ret := m.ctrl.Call(m, "Close") 224 ret0, _ := ret[0].(error) 225 return ret0 226 } 227 228 // Close indicates an expected call of Close. 229 func (mr *MockTxnClientMockRecorder) Close() *gomock.Call { 230 mr.mock.ctrl.T.Helper() 231 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockTxnClient)(nil).Close)) 232 } 233 234 // GetLatestCommitTS mocks base method. 235 func (m *MockTxnClient) GetLatestCommitTS() timestamp.Timestamp { 236 m.ctrl.T.Helper() 237 ret := m.ctrl.Call(m, "GetLatestCommitTS") 238 ret0, _ := ret[0].(timestamp.Timestamp) 239 return ret0 240 } 241 242 // GetLatestCommitTS indicates an expected call of GetLatestCommitTS. 243 func (mr *MockTxnClientMockRecorder) GetLatestCommitTS() *gomock.Call { 244 mr.mock.ctrl.T.Helper() 245 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetLatestCommitTS", reflect.TypeOf((*MockTxnClient)(nil).GetLatestCommitTS)) 246 } 247 248 // GetState mocks base method. 249 func (m *MockTxnClient) GetState() client.TxnState { 250 m.ctrl.T.Helper() 251 ret := m.ctrl.Call(m, "GetState") 252 ret0, _ := ret[0].(client.TxnState) 253 return ret0 254 } 255 256 // GetState indicates an expected call of GetState. 257 func (mr *MockTxnClientMockRecorder) GetState() *gomock.Call { 258 mr.mock.ctrl.T.Helper() 259 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetState", reflect.TypeOf((*MockTxnClient)(nil).GetState)) 260 } 261 262 // GetSyncLatestCommitTSTimes mocks base method. 263 func (m *MockTxnClient) GetSyncLatestCommitTSTimes() uint64 { 264 m.ctrl.T.Helper() 265 ret := m.ctrl.Call(m, "GetSyncLatestCommitTSTimes") 266 ret0, _ := ret[0].(uint64) 267 return ret0 268 } 269 270 // GetSyncLatestCommitTSTimes indicates an expected call of GetSyncLatestCommitTSTimes. 271 func (mr *MockTxnClientMockRecorder) GetSyncLatestCommitTSTimes() *gomock.Call { 272 mr.mock.ctrl.T.Helper() 273 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSyncLatestCommitTSTimes", reflect.TypeOf((*MockTxnClient)(nil).GetSyncLatestCommitTSTimes)) 274 } 275 276 // IterTxns mocks base method. 277 func (m *MockTxnClient) IterTxns(arg0 func(client.TxnOverview) bool) { 278 m.ctrl.T.Helper() 279 m.ctrl.Call(m, "IterTxns", arg0) 280 } 281 282 // IterTxns indicates an expected call of IterTxns. 283 func (mr *MockTxnClientMockRecorder) IterTxns(arg0 interface{}) *gomock.Call { 284 mr.mock.ctrl.T.Helper() 285 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IterTxns", reflect.TypeOf((*MockTxnClient)(nil).IterTxns), arg0) 286 } 287 288 // MinTimestamp mocks base method. 289 func (m *MockTxnClient) MinTimestamp() timestamp.Timestamp { 290 m.ctrl.T.Helper() 291 ret := m.ctrl.Call(m, "MinTimestamp") 292 ret0, _ := ret[0].(timestamp.Timestamp) 293 return ret0 294 } 295 296 // MinTimestamp indicates an expected call of MinTimestamp. 297 func (mr *MockTxnClientMockRecorder) MinTimestamp() *gomock.Call { 298 mr.mock.ctrl.T.Helper() 299 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MinTimestamp", reflect.TypeOf((*MockTxnClient)(nil).MinTimestamp)) 300 } 301 302 // New mocks base method. 303 func (m *MockTxnClient) New(ctx context.Context, commitTS timestamp.Timestamp, options ...client.TxnOption) (client.TxnOperator, error) { 304 m.ctrl.T.Helper() 305 varargs := []interface{}{ctx, commitTS} 306 for _, a := range options { 307 varargs = append(varargs, a) 308 } 309 ret := m.ctrl.Call(m, "New", varargs...) 310 ret0, _ := ret[0].(client.TxnOperator) 311 ret1, _ := ret[1].(error) 312 return ret0, ret1 313 } 314 315 // New indicates an expected call of New. 316 func (mr *MockTxnClientMockRecorder) New(ctx, commitTS interface{}, options ...interface{}) *gomock.Call { 317 mr.mock.ctrl.T.Helper() 318 varargs := append([]interface{}{ctx, commitTS}, options...) 319 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "New", reflect.TypeOf((*MockTxnClient)(nil).New), varargs...) 320 } 321 322 // NewWithSnapshot mocks base method. 323 func (m *MockTxnClient) NewWithSnapshot(snapshot []byte) (client.TxnOperator, error) { 324 m.ctrl.T.Helper() 325 ret := m.ctrl.Call(m, "NewWithSnapshot", snapshot) 326 ret0, _ := ret[0].(client.TxnOperator) 327 ret1, _ := ret[1].(error) 328 return ret0, ret1 329 } 330 331 // NewWithSnapshot indicates an expected call of NewWithSnapshot. 332 func (mr *MockTxnClientMockRecorder) NewWithSnapshot(snapshot interface{}) *gomock.Call { 333 mr.mock.ctrl.T.Helper() 334 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NewWithSnapshot", reflect.TypeOf((*MockTxnClient)(nil).NewWithSnapshot), snapshot) 335 } 336 337 // Pause mocks base method. 338 func (m *MockTxnClient) Pause() { 339 m.ctrl.T.Helper() 340 m.ctrl.Call(m, "Pause") 341 } 342 343 // Pause indicates an expected call of Pause. 344 func (mr *MockTxnClientMockRecorder) Pause() *gomock.Call { 345 mr.mock.ctrl.T.Helper() 346 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Pause", reflect.TypeOf((*MockTxnClient)(nil).Pause)) 347 } 348 349 // RefreshExpressionEnabled mocks base method. 350 func (m *MockTxnClient) RefreshExpressionEnabled() bool { 351 m.ctrl.T.Helper() 352 ret := m.ctrl.Call(m, "RefreshExpressionEnabled") 353 ret0, _ := ret[0].(bool) 354 return ret0 355 } 356 357 // RefreshExpressionEnabled indicates an expected call of RefreshExpressionEnabled. 358 func (mr *MockTxnClientMockRecorder) RefreshExpressionEnabled() *gomock.Call { 359 mr.mock.ctrl.T.Helper() 360 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RefreshExpressionEnabled", reflect.TypeOf((*MockTxnClient)(nil).RefreshExpressionEnabled)) 361 } 362 363 // Resume mocks base method. 364 func (m *MockTxnClient) Resume() { 365 m.ctrl.T.Helper() 366 m.ctrl.Call(m, "Resume") 367 } 368 369 // Resume indicates an expected call of Resume. 370 func (mr *MockTxnClientMockRecorder) Resume() *gomock.Call { 371 mr.mock.ctrl.T.Helper() 372 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Resume", reflect.TypeOf((*MockTxnClient)(nil).Resume)) 373 } 374 375 // SyncLatestCommitTS mocks base method. 376 func (m *MockTxnClient) SyncLatestCommitTS(arg0 timestamp.Timestamp) { 377 m.ctrl.T.Helper() 378 m.ctrl.Call(m, "SyncLatestCommitTS", arg0) 379 } 380 381 // SyncLatestCommitTS indicates an expected call of SyncLatestCommitTS. 382 func (mr *MockTxnClientMockRecorder) SyncLatestCommitTS(arg0 interface{}) *gomock.Call { 383 mr.mock.ctrl.T.Helper() 384 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SyncLatestCommitTS", reflect.TypeOf((*MockTxnClient)(nil).SyncLatestCommitTS), arg0) 385 } 386 387 // WaitLogTailAppliedAt mocks base method. 388 func (m *MockTxnClient) WaitLogTailAppliedAt(ctx context.Context, ts timestamp.Timestamp) (timestamp.Timestamp, error) { 389 m.ctrl.T.Helper() 390 ret := m.ctrl.Call(m, "WaitLogTailAppliedAt", ctx, ts) 391 ret0, _ := ret[0].(timestamp.Timestamp) 392 ret1, _ := ret[1].(error) 393 return ret0, ret1 394 } 395 396 // WaitLogTailAppliedAt indicates an expected call of WaitLogTailAppliedAt. 397 func (mr *MockTxnClientMockRecorder) WaitLogTailAppliedAt(ctx, ts interface{}) *gomock.Call { 398 mr.mock.ctrl.T.Helper() 399 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WaitLogTailAppliedAt", reflect.TypeOf((*MockTxnClient)(nil).WaitLogTailAppliedAt), ctx, ts) 400 } 401 402 // MockTxnOperator is a mock of TxnOperator interface. 403 type MockTxnOperator struct { 404 ctrl *gomock.Controller 405 recorder *MockTxnOperatorMockRecorder 406 } 407 408 // MockTxnOperatorMockRecorder is the mock recorder for MockTxnOperator. 409 type MockTxnOperatorMockRecorder struct { 410 mock *MockTxnOperator 411 } 412 413 // NewMockTxnOperator creates a new mock instance. 414 func NewMockTxnOperator(ctrl *gomock.Controller) *MockTxnOperator { 415 mock := &MockTxnOperator{ctrl: ctrl} 416 mock.recorder = &MockTxnOperatorMockRecorder{mock} 417 return mock 418 } 419 420 // EXPECT returns an object that allows the caller to indicate expected use. 421 func (m *MockTxnOperator) EXPECT() *MockTxnOperatorMockRecorder { 422 return m.recorder 423 } 424 425 // AddLockTable mocks base method. 426 func (m *MockTxnOperator) AddLockTable(locktable lock.LockTable) error { 427 m.ctrl.T.Helper() 428 ret := m.ctrl.Call(m, "AddLockTable", locktable) 429 ret0, _ := ret[0].(error) 430 return ret0 431 } 432 433 // AddLockTable indicates an expected call of AddLockTable. 434 func (mr *MockTxnOperatorMockRecorder) AddLockTable(locktable interface{}) *gomock.Call { 435 mr.mock.ctrl.T.Helper() 436 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddLockTable", reflect.TypeOf((*MockTxnOperator)(nil).AddLockTable), locktable) 437 } 438 439 // AddWaitLock mocks base method. 440 func (m *MockTxnOperator) AddWaitLock(tableID uint64, rows [][]byte, opt lock.LockOptions) uint64 { 441 m.ctrl.T.Helper() 442 ret := m.ctrl.Call(m, "AddWaitLock", tableID, rows, opt) 443 ret0, _ := ret[0].(uint64) 444 return ret0 445 } 446 447 // AddWaitLock indicates an expected call of AddWaitLock. 448 func (mr *MockTxnOperatorMockRecorder) AddWaitLock(tableID, rows, opt interface{}) *gomock.Call { 449 mr.mock.ctrl.T.Helper() 450 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddWaitLock", reflect.TypeOf((*MockTxnOperator)(nil).AddWaitLock), tableID, rows, opt) 451 } 452 453 // AddWorkspace mocks base method. 454 func (m *MockTxnOperator) AddWorkspace(workspace client.Workspace) { 455 m.ctrl.T.Helper() 456 m.ctrl.Call(m, "AddWorkspace", workspace) 457 } 458 459 // AddWorkspace indicates an expected call of AddWorkspace. 460 func (mr *MockTxnOperatorMockRecorder) AddWorkspace(workspace interface{}) *gomock.Call { 461 mr.mock.ctrl.T.Helper() 462 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddWorkspace", reflect.TypeOf((*MockTxnOperator)(nil).AddWorkspace), workspace) 463 } 464 465 // AppendEventCallback mocks base method. 466 func (m *MockTxnOperator) AppendEventCallback(event client.EventType, callbacks ...func(client.TxnEvent)) { 467 m.ctrl.T.Helper() 468 varargs := []interface{}{event} 469 for _, a := range callbacks { 470 varargs = append(varargs, a) 471 } 472 m.ctrl.Call(m, "AppendEventCallback", varargs...) 473 } 474 475 // AppendEventCallback indicates an expected call of AppendEventCallback. 476 func (mr *MockTxnOperatorMockRecorder) AppendEventCallback(event interface{}, callbacks ...interface{}) *gomock.Call { 477 mr.mock.ctrl.T.Helper() 478 varargs := append([]interface{}{event}, callbacks...) 479 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AppendEventCallback", reflect.TypeOf((*MockTxnOperator)(nil).AppendEventCallback), varargs...) 480 } 481 482 // ApplySnapshot mocks base method. 483 func (m *MockTxnOperator) ApplySnapshot(data []byte) error { 484 m.ctrl.T.Helper() 485 ret := m.ctrl.Call(m, "ApplySnapshot", data) 486 ret0, _ := ret[0].(error) 487 return ret0 488 } 489 490 // ApplySnapshot indicates an expected call of ApplySnapshot. 491 func (mr *MockTxnOperatorMockRecorder) ApplySnapshot(data interface{}) *gomock.Call { 492 mr.mock.ctrl.T.Helper() 493 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ApplySnapshot", reflect.TypeOf((*MockTxnOperator)(nil).ApplySnapshot), data) 494 } 495 496 // CloneSnapshotOp mocks base method. 497 func (m *MockTxnOperator) CloneSnapshotOp(snapshot timestamp.Timestamp) client.TxnOperator { 498 m.ctrl.T.Helper() 499 ret := m.ctrl.Call(m, "CloneSnapshotOp", snapshot) 500 ret0, _ := ret[0].(client.TxnOperator) 501 return ret0 502 } 503 504 // CloneSnapshotOp indicates an expected call of CloneSnapshotOp. 505 func (mr *MockTxnOperatorMockRecorder) CloneSnapshotOp(snapshot interface{}) *gomock.Call { 506 mr.mock.ctrl.T.Helper() 507 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloneSnapshotOp", reflect.TypeOf((*MockTxnOperator)(nil).CloneSnapshotOp), snapshot) 508 } 509 510 // Commit mocks base method. 511 func (m *MockTxnOperator) Commit(ctx context.Context) error { 512 m.ctrl.T.Helper() 513 ret := m.ctrl.Call(m, "Commit", ctx) 514 ret0, _ := ret[0].(error) 515 return ret0 516 } 517 518 // Commit indicates an expected call of Commit. 519 func (mr *MockTxnOperatorMockRecorder) Commit(ctx interface{}) *gomock.Call { 520 mr.mock.ctrl.T.Helper() 521 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Commit", reflect.TypeOf((*MockTxnOperator)(nil).Commit), ctx) 522 } 523 524 // CreateTS mocks base method. 525 func (m *MockTxnOperator) CreateTS() timestamp.Timestamp { 526 m.ctrl.T.Helper() 527 ret := m.ctrl.Call(m, "CreateTS") 528 ret0, _ := ret[0].(timestamp.Timestamp) 529 return ret0 530 } 531 532 // CreateTS indicates an expected call of CreateTS. 533 func (mr *MockTxnOperatorMockRecorder) CreateTS() *gomock.Call { 534 mr.mock.ctrl.T.Helper() 535 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateTS", reflect.TypeOf((*MockTxnOperator)(nil).CreateTS)) 536 } 537 538 // Debug mocks base method. 539 func (m *MockTxnOperator) Debug(ctx context.Context, ops []txn.TxnRequest) (*rpc.SendResult, error) { 540 m.ctrl.T.Helper() 541 ret := m.ctrl.Call(m, "Debug", ctx, ops) 542 ret0, _ := ret[0].(*rpc.SendResult) 543 ret1, _ := ret[1].(error) 544 return ret0, ret1 545 } 546 547 // Debug indicates an expected call of Debug. 548 func (mr *MockTxnOperatorMockRecorder) Debug(ctx, ops interface{}) *gomock.Call { 549 mr.mock.ctrl.T.Helper() 550 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Debug", reflect.TypeOf((*MockTxnOperator)(nil).Debug), ctx, ops) 551 } 552 553 // EnterRunSql mocks base method. 554 func (m *MockTxnOperator) EnterRunSql() { 555 m.ctrl.T.Helper() 556 m.ctrl.Call(m, "EnterRunSql") 557 } 558 559 // EnterRunSql indicates an expected call of EnterRunSql. 560 func (mr *MockTxnOperatorMockRecorder) EnterRunSql() *gomock.Call { 561 mr.mock.ctrl.T.Helper() 562 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "EnterRunSql", reflect.TypeOf((*MockTxnOperator)(nil).EnterRunSql)) 563 } 564 565 // ExitRunSql mocks base method. 566 func (m *MockTxnOperator) ExitRunSql() { 567 m.ctrl.T.Helper() 568 m.ctrl.Call(m, "ExitRunSql") 569 } 570 571 // ExitRunSql indicates an expected call of ExitRunSql. 572 func (mr *MockTxnOperatorMockRecorder) ExitRunSql() *gomock.Call { 573 mr.mock.ctrl.T.Helper() 574 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ExitRunSql", reflect.TypeOf((*MockTxnOperator)(nil).ExitRunSql)) 575 } 576 577 // GetOverview mocks base method. 578 func (m *MockTxnOperator) GetOverview() client.TxnOverview { 579 m.ctrl.T.Helper() 580 ret := m.ctrl.Call(m, "GetOverview") 581 ret0, _ := ret[0].(client.TxnOverview) 582 return ret0 583 } 584 585 // GetOverview indicates an expected call of GetOverview. 586 func (mr *MockTxnOperatorMockRecorder) GetOverview() *gomock.Call { 587 mr.mock.ctrl.T.Helper() 588 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetOverview", reflect.TypeOf((*MockTxnOperator)(nil).GetOverview)) 589 } 590 591 // GetWaitActiveCost mocks base method. 592 func (m *MockTxnOperator) GetWaitActiveCost() time.Duration { 593 m.ctrl.T.Helper() 594 ret := m.ctrl.Call(m, "GetWaitActiveCost") 595 ret0, _ := ret[0].(time.Duration) 596 return ret0 597 } 598 599 // GetWaitActiveCost indicates an expected call of GetWaitActiveCost. 600 func (mr *MockTxnOperatorMockRecorder) GetWaitActiveCost() *gomock.Call { 601 mr.mock.ctrl.T.Helper() 602 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWaitActiveCost", reflect.TypeOf((*MockTxnOperator)(nil).GetWaitActiveCost)) 603 } 604 605 // GetWorkspace mocks base method. 606 func (m *MockTxnOperator) GetWorkspace() client.Workspace { 607 m.ctrl.T.Helper() 608 ret := m.ctrl.Call(m, "GetWorkspace") 609 ret0, _ := ret[0].(client.Workspace) 610 return ret0 611 } 612 613 // GetWorkspace indicates an expected call of GetWorkspace. 614 func (mr *MockTxnOperatorMockRecorder) GetWorkspace() *gomock.Call { 615 mr.mock.ctrl.T.Helper() 616 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetWorkspace", reflect.TypeOf((*MockTxnOperator)(nil).GetWorkspace)) 617 } 618 619 // IsRetry mocks base method. 620 func (m *MockTxnOperator) IsRetry() bool { 621 m.ctrl.T.Helper() 622 ret := m.ctrl.Call(m, "IsRetry") 623 ret0, _ := ret[0].(bool) 624 return ret0 625 } 626 627 // IsRetry indicates an expected call of IsRetry. 628 func (mr *MockTxnOperatorMockRecorder) IsRetry() *gomock.Call { 629 mr.mock.ctrl.T.Helper() 630 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsRetry", reflect.TypeOf((*MockTxnOperator)(nil).IsRetry)) 631 } 632 633 // IsSnapOp mocks base method. 634 func (m *MockTxnOperator) IsSnapOp() bool { 635 m.ctrl.T.Helper() 636 ret := m.ctrl.Call(m, "IsSnapOp") 637 ret0, _ := ret[0].(bool) 638 return ret0 639 } 640 641 // IsSnapOp indicates an expected call of IsSnapOp. 642 func (mr *MockTxnOperatorMockRecorder) IsSnapOp() *gomock.Call { 643 mr.mock.ctrl.T.Helper() 644 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsSnapOp", reflect.TypeOf((*MockTxnOperator)(nil).IsSnapOp)) 645 } 646 647 // LockSkipped mocks base method. 648 func (m *MockTxnOperator) LockSkipped(tableID uint64, mode lock.LockMode) bool { 649 m.ctrl.T.Helper() 650 ret := m.ctrl.Call(m, "LockSkipped", tableID, mode) 651 ret0, _ := ret[0].(bool) 652 return ret0 653 } 654 655 // LockSkipped indicates an expected call of LockSkipped. 656 func (mr *MockTxnOperatorMockRecorder) LockSkipped(tableID, mode interface{}) *gomock.Call { 657 mr.mock.ctrl.T.Helper() 658 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LockSkipped", reflect.TypeOf((*MockTxnOperator)(nil).LockSkipped), tableID, mode) 659 } 660 661 // LockTableCount mocks base method. 662 func (m *MockTxnOperator) LockTableCount() int32 { 663 m.ctrl.T.Helper() 664 ret := m.ctrl.Call(m, "LockTableCount") 665 ret0, _ := ret[0].(int32) 666 return ret0 667 } 668 669 // LockTableCount indicates an expected call of LockTableCount. 670 func (mr *MockTxnOperatorMockRecorder) LockTableCount() *gomock.Call { 671 mr.mock.ctrl.T.Helper() 672 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LockTableCount", reflect.TypeOf((*MockTxnOperator)(nil).LockTableCount)) 673 } 674 675 // NextSequence mocks base method. 676 func (m *MockTxnOperator) NextSequence() uint64 { 677 m.ctrl.T.Helper() 678 ret := m.ctrl.Call(m, "NextSequence") 679 ret0, _ := ret[0].(uint64) 680 return ret0 681 } 682 683 // NextSequence indicates an expected call of NextSequence. 684 func (mr *MockTxnOperatorMockRecorder) NextSequence() *gomock.Call { 685 mr.mock.ctrl.T.Helper() 686 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NextSequence", reflect.TypeOf((*MockTxnOperator)(nil).NextSequence)) 687 } 688 689 // Read mocks base method. 690 func (m *MockTxnOperator) Read(ctx context.Context, ops []txn.TxnRequest) (*rpc.SendResult, error) { 691 m.ctrl.T.Helper() 692 ret := m.ctrl.Call(m, "Read", ctx, ops) 693 ret0, _ := ret[0].(*rpc.SendResult) 694 ret1, _ := ret[1].(error) 695 return ret0, ret1 696 } 697 698 // Read indicates an expected call of Read. 699 func (mr *MockTxnOperatorMockRecorder) Read(ctx, ops interface{}) *gomock.Call { 700 mr.mock.ctrl.T.Helper() 701 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Read", reflect.TypeOf((*MockTxnOperator)(nil).Read), ctx, ops) 702 } 703 704 // RemoveWaitLock mocks base method. 705 func (m *MockTxnOperator) RemoveWaitLock(key uint64) { 706 m.ctrl.T.Helper() 707 m.ctrl.Call(m, "RemoveWaitLock", key) 708 } 709 710 // RemoveWaitLock indicates an expected call of RemoveWaitLock. 711 func (mr *MockTxnOperatorMockRecorder) RemoveWaitLock(key interface{}) *gomock.Call { 712 mr.mock.ctrl.T.Helper() 713 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemoveWaitLock", reflect.TypeOf((*MockTxnOperator)(nil).RemoveWaitLock), key) 714 } 715 716 // ResetRetry mocks base method. 717 func (m *MockTxnOperator) ResetRetry(arg0 bool) { 718 m.ctrl.T.Helper() 719 m.ctrl.Call(m, "ResetRetry", arg0) 720 } 721 722 // ResetRetry indicates an expected call of ResetRetry. 723 func (mr *MockTxnOperatorMockRecorder) ResetRetry(arg0 interface{}) *gomock.Call { 724 mr.mock.ctrl.T.Helper() 725 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ResetRetry", reflect.TypeOf((*MockTxnOperator)(nil).ResetRetry), arg0) 726 } 727 728 // Rollback mocks base method. 729 func (m *MockTxnOperator) Rollback(ctx context.Context) error { 730 m.ctrl.T.Helper() 731 ret := m.ctrl.Call(m, "Rollback", ctx) 732 ret0, _ := ret[0].(error) 733 return ret0 734 } 735 736 // Rollback indicates an expected call of Rollback. 737 func (mr *MockTxnOperatorMockRecorder) Rollback(ctx interface{}) *gomock.Call { 738 mr.mock.ctrl.T.Helper() 739 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Rollback", reflect.TypeOf((*MockTxnOperator)(nil).Rollback), ctx) 740 } 741 742 // Snapshot mocks base method. 743 func (m *MockTxnOperator) Snapshot() ([]byte, error) { 744 m.ctrl.T.Helper() 745 ret := m.ctrl.Call(m, "Snapshot") 746 ret0, _ := ret[0].([]byte) 747 ret1, _ := ret[1].(error) 748 return ret0, ret1 749 } 750 751 // Snapshot indicates an expected call of Snapshot. 752 func (mr *MockTxnOperatorMockRecorder) Snapshot() *gomock.Call { 753 mr.mock.ctrl.T.Helper() 754 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Snapshot", reflect.TypeOf((*MockTxnOperator)(nil).Snapshot)) 755 } 756 757 // SnapshotTS mocks base method. 758 func (m *MockTxnOperator) SnapshotTS() timestamp.Timestamp { 759 m.ctrl.T.Helper() 760 ret := m.ctrl.Call(m, "SnapshotTS") 761 ret0, _ := ret[0].(timestamp.Timestamp) 762 return ret0 763 } 764 765 // SnapshotTS indicates an expected call of SnapshotTS. 766 func (mr *MockTxnOperatorMockRecorder) SnapshotTS() *gomock.Call { 767 mr.mock.ctrl.T.Helper() 768 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SnapshotTS", reflect.TypeOf((*MockTxnOperator)(nil).SnapshotTS)) 769 } 770 771 // Status mocks base method. 772 func (m *MockTxnOperator) Status() txn.TxnStatus { 773 m.ctrl.T.Helper() 774 ret := m.ctrl.Call(m, "Status") 775 ret0, _ := ret[0].(txn.TxnStatus) 776 return ret0 777 } 778 779 // Status indicates an expected call of Status. 780 func (mr *MockTxnOperatorMockRecorder) Status() *gomock.Call { 781 mr.mock.ctrl.T.Helper() 782 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Status", reflect.TypeOf((*MockTxnOperator)(nil).Status)) 783 } 784 785 // Txn mocks base method. 786 func (m *MockTxnOperator) Txn() txn.TxnMeta { 787 m.ctrl.T.Helper() 788 ret := m.ctrl.Call(m, "Txn") 789 ret0, _ := ret[0].(txn.TxnMeta) 790 return ret0 791 } 792 793 // Txn indicates an expected call of Txn. 794 func (mr *MockTxnOperatorMockRecorder) Txn() *gomock.Call { 795 mr.mock.ctrl.T.Helper() 796 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Txn", reflect.TypeOf((*MockTxnOperator)(nil).Txn)) 797 } 798 799 // TxnOptions mocks base method. 800 func (m *MockTxnOperator) TxnOptions() txn.TxnOptions { 801 m.ctrl.T.Helper() 802 ret := m.ctrl.Call(m, "TxnOptions") 803 ret0, _ := ret[0].(txn.TxnOptions) 804 return ret0 805 } 806 807 // TxnOptions indicates an expected call of TxnOptions. 808 func (mr *MockTxnOperatorMockRecorder) TxnOptions() *gomock.Call { 809 mr.mock.ctrl.T.Helper() 810 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TxnOptions", reflect.TypeOf((*MockTxnOperator)(nil).TxnOptions)) 811 } 812 813 // TxnRef mocks base method. 814 func (m *MockTxnOperator) TxnRef() *txn.TxnMeta { 815 m.ctrl.T.Helper() 816 ret := m.ctrl.Call(m, "TxnRef") 817 ret0, _ := ret[0].(*txn.TxnMeta) 818 return ret0 819 } 820 821 // TxnRef indicates an expected call of TxnRef. 822 func (mr *MockTxnOperatorMockRecorder) TxnRef() *gomock.Call { 823 mr.mock.ctrl.T.Helper() 824 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TxnRef", reflect.TypeOf((*MockTxnOperator)(nil).TxnRef)) 825 } 826 827 // UpdateSnapshot mocks base method. 828 func (m *MockTxnOperator) UpdateSnapshot(ctx context.Context, ts timestamp.Timestamp) error { 829 m.ctrl.T.Helper() 830 ret := m.ctrl.Call(m, "UpdateSnapshot", ctx, ts) 831 ret0, _ := ret[0].(error) 832 return ret0 833 } 834 835 // UpdateSnapshot indicates an expected call of UpdateSnapshot. 836 func (mr *MockTxnOperatorMockRecorder) UpdateSnapshot(ctx, ts interface{}) *gomock.Call { 837 mr.mock.ctrl.T.Helper() 838 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateSnapshot", reflect.TypeOf((*MockTxnOperator)(nil).UpdateSnapshot), ctx, ts) 839 } 840 841 // Write mocks base method. 842 func (m *MockTxnOperator) Write(ctx context.Context, ops []txn.TxnRequest) (*rpc.SendResult, error) { 843 m.ctrl.T.Helper() 844 ret := m.ctrl.Call(m, "Write", ctx, ops) 845 ret0, _ := ret[0].(*rpc.SendResult) 846 ret1, _ := ret[1].(error) 847 return ret0, ret1 848 } 849 850 // Write indicates an expected call of Write. 851 func (mr *MockTxnOperatorMockRecorder) Write(ctx, ops interface{}) *gomock.Call { 852 mr.mock.ctrl.T.Helper() 853 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Write", reflect.TypeOf((*MockTxnOperator)(nil).Write), ctx, ops) 854 } 855 856 // WriteAndCommit mocks base method. 857 func (m *MockTxnOperator) WriteAndCommit(ctx context.Context, ops []txn.TxnRequest) (*rpc.SendResult, error) { 858 m.ctrl.T.Helper() 859 ret := m.ctrl.Call(m, "WriteAndCommit", ctx, ops) 860 ret0, _ := ret[0].(*rpc.SendResult) 861 ret1, _ := ret[1].(error) 862 return ret0, ret1 863 } 864 865 // WriteAndCommit indicates an expected call of WriteAndCommit. 866 func (mr *MockTxnOperatorMockRecorder) WriteAndCommit(ctx, ops interface{}) *gomock.Call { 867 mr.mock.ctrl.T.Helper() 868 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WriteAndCommit", reflect.TypeOf((*MockTxnOperator)(nil).WriteAndCommit), ctx, ops) 869 } 870 871 // MockTxnIDGenerator is a mock of TxnIDGenerator interface. 872 type MockTxnIDGenerator struct { 873 ctrl *gomock.Controller 874 recorder *MockTxnIDGeneratorMockRecorder 875 } 876 877 // MockTxnIDGeneratorMockRecorder is the mock recorder for MockTxnIDGenerator. 878 type MockTxnIDGeneratorMockRecorder struct { 879 mock *MockTxnIDGenerator 880 } 881 882 // NewMockTxnIDGenerator creates a new mock instance. 883 func NewMockTxnIDGenerator(ctrl *gomock.Controller) *MockTxnIDGenerator { 884 mock := &MockTxnIDGenerator{ctrl: ctrl} 885 mock.recorder = &MockTxnIDGeneratorMockRecorder{mock} 886 return mock 887 } 888 889 // EXPECT returns an object that allows the caller to indicate expected use. 890 func (m *MockTxnIDGenerator) EXPECT() *MockTxnIDGeneratorMockRecorder { 891 return m.recorder 892 } 893 894 // Generate mocks base method. 895 func (m *MockTxnIDGenerator) Generate() []byte { 896 m.ctrl.T.Helper() 897 ret := m.ctrl.Call(m, "Generate") 898 ret0, _ := ret[0].([]byte) 899 return ret0 900 } 901 902 // Generate indicates an expected call of Generate. 903 func (mr *MockTxnIDGeneratorMockRecorder) Generate() *gomock.Call { 904 mr.mock.ctrl.T.Helper() 905 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Generate", reflect.TypeOf((*MockTxnIDGenerator)(nil).Generate)) 906 } 907 908 // MockTimestampWaiter is a mock of TimestampWaiter interface. 909 type MockTimestampWaiter struct { 910 ctrl *gomock.Controller 911 recorder *MockTimestampWaiterMockRecorder 912 } 913 914 // MockTimestampWaiterMockRecorder is the mock recorder for MockTimestampWaiter. 915 type MockTimestampWaiterMockRecorder struct { 916 mock *MockTimestampWaiter 917 } 918 919 // NewMockTimestampWaiter creates a new mock instance. 920 func NewMockTimestampWaiter(ctrl *gomock.Controller) *MockTimestampWaiter { 921 mock := &MockTimestampWaiter{ctrl: ctrl} 922 mock.recorder = &MockTimestampWaiterMockRecorder{mock} 923 return mock 924 } 925 926 // EXPECT returns an object that allows the caller to indicate expected use. 927 func (m *MockTimestampWaiter) EXPECT() *MockTimestampWaiterMockRecorder { 928 return m.recorder 929 } 930 931 // CancelC mocks base method. 932 func (m *MockTimestampWaiter) CancelC() chan struct{} { 933 m.ctrl.T.Helper() 934 ret := m.ctrl.Call(m, "CancelC") 935 ret0, _ := ret[0].(chan struct{}) 936 return ret0 937 } 938 939 // CancelC indicates an expected call of CancelC. 940 func (mr *MockTimestampWaiterMockRecorder) CancelC() *gomock.Call { 941 mr.mock.ctrl.T.Helper() 942 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CancelC", reflect.TypeOf((*MockTimestampWaiter)(nil).CancelC)) 943 } 944 945 // Close mocks base method. 946 func (m *MockTimestampWaiter) Close() { 947 m.ctrl.T.Helper() 948 m.ctrl.Call(m, "Close") 949 } 950 951 // Close indicates an expected call of Close. 952 func (mr *MockTimestampWaiterMockRecorder) Close() *gomock.Call { 953 mr.mock.ctrl.T.Helper() 954 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockTimestampWaiter)(nil).Close)) 955 } 956 957 // GetTimestamp mocks base method. 958 func (m *MockTimestampWaiter) GetTimestamp(arg0 context.Context, arg1 timestamp.Timestamp) (timestamp.Timestamp, error) { 959 m.ctrl.T.Helper() 960 ret := m.ctrl.Call(m, "GetTimestamp", arg0, arg1) 961 ret0, _ := ret[0].(timestamp.Timestamp) 962 ret1, _ := ret[1].(error) 963 return ret0, ret1 964 } 965 966 // GetTimestamp indicates an expected call of GetTimestamp. 967 func (mr *MockTimestampWaiterMockRecorder) GetTimestamp(arg0, arg1 interface{}) *gomock.Call { 968 mr.mock.ctrl.T.Helper() 969 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTimestamp", reflect.TypeOf((*MockTimestampWaiter)(nil).GetTimestamp), arg0, arg1) 970 } 971 972 // LatestTS mocks base method. 973 func (m *MockTimestampWaiter) LatestTS() timestamp.Timestamp { 974 m.ctrl.T.Helper() 975 ret := m.ctrl.Call(m, "LatestTS") 976 ret0, _ := ret[0].(timestamp.Timestamp) 977 return ret0 978 } 979 980 // LatestTS indicates an expected call of LatestTS. 981 func (mr *MockTimestampWaiterMockRecorder) LatestTS() *gomock.Call { 982 mr.mock.ctrl.T.Helper() 983 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LatestTS", reflect.TypeOf((*MockTimestampWaiter)(nil).LatestTS)) 984 } 985 986 // NotifyLatestCommitTS mocks base method. 987 func (m *MockTimestampWaiter) NotifyLatestCommitTS(appliedTS timestamp.Timestamp) { 988 m.ctrl.T.Helper() 989 m.ctrl.Call(m, "NotifyLatestCommitTS", appliedTS) 990 } 991 992 // NotifyLatestCommitTS indicates an expected call of NotifyLatestCommitTS. 993 func (mr *MockTimestampWaiterMockRecorder) NotifyLatestCommitTS(appliedTS interface{}) *gomock.Call { 994 mr.mock.ctrl.T.Helper() 995 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NotifyLatestCommitTS", reflect.TypeOf((*MockTimestampWaiter)(nil).NotifyLatestCommitTS), appliedTS) 996 } 997 998 // Pause mocks base method. 999 func (m *MockTimestampWaiter) Pause() { 1000 m.ctrl.T.Helper() 1001 m.ctrl.Call(m, "Pause") 1002 } 1003 1004 // Pause indicates an expected call of Pause. 1005 func (mr *MockTimestampWaiterMockRecorder) Pause() *gomock.Call { 1006 mr.mock.ctrl.T.Helper() 1007 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Pause", reflect.TypeOf((*MockTimestampWaiter)(nil).Pause)) 1008 } 1009 1010 // Resume mocks base method. 1011 func (m *MockTimestampWaiter) Resume() { 1012 m.ctrl.T.Helper() 1013 m.ctrl.Call(m, "Resume") 1014 } 1015 1016 // Resume indicates an expected call of Resume. 1017 func (mr *MockTimestampWaiterMockRecorder) Resume() *gomock.Call { 1018 mr.mock.ctrl.T.Helper() 1019 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Resume", reflect.TypeOf((*MockTimestampWaiter)(nil).Resume)) 1020 } 1021 1022 // MockWorkspace is a mock of Workspace interface. 1023 type MockWorkspace struct { 1024 ctrl *gomock.Controller 1025 recorder *MockWorkspaceMockRecorder 1026 } 1027 1028 // MockWorkspaceMockRecorder is the mock recorder for MockWorkspace. 1029 type MockWorkspaceMockRecorder struct { 1030 mock *MockWorkspace 1031 } 1032 1033 // NewMockWorkspace creates a new mock instance. 1034 func NewMockWorkspace(ctrl *gomock.Controller) *MockWorkspace { 1035 mock := &MockWorkspace{ctrl: ctrl} 1036 mock.recorder = &MockWorkspaceMockRecorder{mock} 1037 return mock 1038 } 1039 1040 // EXPECT returns an object that allows the caller to indicate expected use. 1041 func (m *MockWorkspace) EXPECT() *MockWorkspaceMockRecorder { 1042 return m.recorder 1043 } 1044 1045 // Adjust mocks base method. 1046 func (m *MockWorkspace) Adjust(writeOffset uint64) error { 1047 m.ctrl.T.Helper() 1048 ret := m.ctrl.Call(m, "Adjust", writeOffset) 1049 ret0, _ := ret[0].(error) 1050 return ret0 1051 } 1052 1053 // Adjust indicates an expected call of Adjust. 1054 func (mr *MockWorkspaceMockRecorder) Adjust(writeOffset interface{}) *gomock.Call { 1055 mr.mock.ctrl.T.Helper() 1056 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Adjust", reflect.TypeOf((*MockWorkspace)(nil).Adjust), writeOffset) 1057 } 1058 1059 // BindTxnOp mocks base method. 1060 func (m *MockWorkspace) BindTxnOp(op client.TxnOperator) { 1061 m.ctrl.T.Helper() 1062 m.ctrl.Call(m, "BindTxnOp", op) 1063 } 1064 1065 // BindTxnOp indicates an expected call of BindTxnOp. 1066 func (mr *MockWorkspaceMockRecorder) BindTxnOp(op interface{}) *gomock.Call { 1067 mr.mock.ctrl.T.Helper() 1068 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BindTxnOp", reflect.TypeOf((*MockWorkspace)(nil).BindTxnOp), op) 1069 } 1070 1071 // CloneSnapshotWS mocks base method. 1072 func (m *MockWorkspace) CloneSnapshotWS() client.Workspace { 1073 m.ctrl.T.Helper() 1074 ret := m.ctrl.Call(m, "CloneSnapshotWS") 1075 ret0, _ := ret[0].(client.Workspace) 1076 return ret0 1077 } 1078 1079 // CloneSnapshotWS indicates an expected call of CloneSnapshotWS. 1080 func (mr *MockWorkspaceMockRecorder) CloneSnapshotWS() *gomock.Call { 1081 mr.mock.ctrl.T.Helper() 1082 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CloneSnapshotWS", reflect.TypeOf((*MockWorkspace)(nil).CloneSnapshotWS)) 1083 } 1084 1085 // Commit mocks base method. 1086 func (m *MockWorkspace) Commit(ctx context.Context) ([]txn.TxnRequest, error) { 1087 m.ctrl.T.Helper() 1088 ret := m.ctrl.Call(m, "Commit", ctx) 1089 ret0, _ := ret[0].([]txn.TxnRequest) 1090 ret1, _ := ret[1].(error) 1091 return ret0, ret1 1092 } 1093 1094 // Commit indicates an expected call of Commit. 1095 func (mr *MockWorkspaceMockRecorder) Commit(ctx interface{}) *gomock.Call { 1096 mr.mock.ctrl.T.Helper() 1097 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Commit", reflect.TypeOf((*MockWorkspace)(nil).Commit), ctx) 1098 } 1099 1100 // EndStatement mocks base method. 1101 func (m *MockWorkspace) EndStatement() { 1102 m.ctrl.T.Helper() 1103 m.ctrl.Call(m, "EndStatement") 1104 } 1105 1106 // EndStatement indicates an expected call of EndStatement. 1107 func (mr *MockWorkspaceMockRecorder) EndStatement() *gomock.Call { 1108 mr.mock.ctrl.T.Helper() 1109 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "EndStatement", reflect.TypeOf((*MockWorkspace)(nil).EndStatement)) 1110 } 1111 1112 // GetSQLCount mocks base method. 1113 func (m *MockWorkspace) GetSQLCount() uint64 { 1114 m.ctrl.T.Helper() 1115 ret := m.ctrl.Call(m, "GetSQLCount") 1116 ret0, _ := ret[0].(uint64) 1117 return ret0 1118 } 1119 1120 // GetSQLCount indicates an expected call of GetSQLCount. 1121 func (mr *MockWorkspaceMockRecorder) GetSQLCount() *gomock.Call { 1122 mr.mock.ctrl.T.Helper() 1123 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSQLCount", reflect.TypeOf((*MockWorkspace)(nil).GetSQLCount)) 1124 } 1125 1126 // GetSnapshotWriteOffset mocks base method. 1127 func (m *MockWorkspace) GetSnapshotWriteOffset() int { 1128 m.ctrl.T.Helper() 1129 ret := m.ctrl.Call(m, "GetSnapshotWriteOffset") 1130 ret0, _ := ret[0].(int) 1131 return ret0 1132 } 1133 1134 // GetSnapshotWriteOffset indicates an expected call of GetSnapshotWriteOffset. 1135 func (mr *MockWorkspaceMockRecorder) GetSnapshotWriteOffset() *gomock.Call { 1136 mr.mock.ctrl.T.Helper() 1137 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetSnapshotWriteOffset", reflect.TypeOf((*MockWorkspace)(nil).GetSnapshotWriteOffset)) 1138 } 1139 1140 // IncrSQLCount mocks base method. 1141 func (m *MockWorkspace) IncrSQLCount() { 1142 m.ctrl.T.Helper() 1143 m.ctrl.Call(m, "IncrSQLCount") 1144 } 1145 1146 // IncrSQLCount indicates an expected call of IncrSQLCount. 1147 func (mr *MockWorkspaceMockRecorder) IncrSQLCount() *gomock.Call { 1148 mr.mock.ctrl.T.Helper() 1149 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IncrSQLCount", reflect.TypeOf((*MockWorkspace)(nil).IncrSQLCount)) 1150 } 1151 1152 // IncrStatementID mocks base method. 1153 func (m *MockWorkspace) IncrStatementID(ctx context.Context, commit bool) error { 1154 m.ctrl.T.Helper() 1155 ret := m.ctrl.Call(m, "IncrStatementID", ctx, commit) 1156 ret0, _ := ret[0].(error) 1157 return ret0 1158 } 1159 1160 // IncrStatementID indicates an expected call of IncrStatementID. 1161 func (mr *MockWorkspaceMockRecorder) IncrStatementID(ctx, commit interface{}) *gomock.Call { 1162 mr.mock.ctrl.T.Helper() 1163 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IncrStatementID", reflect.TypeOf((*MockWorkspace)(nil).IncrStatementID), ctx, commit) 1164 } 1165 1166 // Rollback mocks base method. 1167 func (m *MockWorkspace) Rollback(ctx context.Context) error { 1168 m.ctrl.T.Helper() 1169 ret := m.ctrl.Call(m, "Rollback", ctx) 1170 ret0, _ := ret[0].(error) 1171 return ret0 1172 } 1173 1174 // Rollback indicates an expected call of Rollback. 1175 func (mr *MockWorkspaceMockRecorder) Rollback(ctx interface{}) *gomock.Call { 1176 mr.mock.ctrl.T.Helper() 1177 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Rollback", reflect.TypeOf((*MockWorkspace)(nil).Rollback), ctx) 1178 } 1179 1180 // RollbackLastStatement mocks base method. 1181 func (m *MockWorkspace) RollbackLastStatement(ctx context.Context) error { 1182 m.ctrl.T.Helper() 1183 ret := m.ctrl.Call(m, "RollbackLastStatement", ctx) 1184 ret0, _ := ret[0].(error) 1185 return ret0 1186 } 1187 1188 // RollbackLastStatement indicates an expected call of RollbackLastStatement. 1189 func (mr *MockWorkspaceMockRecorder) RollbackLastStatement(ctx interface{}) *gomock.Call { 1190 mr.mock.ctrl.T.Helper() 1191 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RollbackLastStatement", reflect.TypeOf((*MockWorkspace)(nil).RollbackLastStatement), ctx) 1192 } 1193 1194 // StartStatement mocks base method. 1195 func (m *MockWorkspace) StartStatement() { 1196 m.ctrl.T.Helper() 1197 m.ctrl.Call(m, "StartStatement") 1198 } 1199 1200 // StartStatement indicates an expected call of StartStatement. 1201 func (mr *MockWorkspaceMockRecorder) StartStatement() *gomock.Call { 1202 mr.mock.ctrl.T.Helper() 1203 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "StartStatement", reflect.TypeOf((*MockWorkspace)(nil).StartStatement)) 1204 } 1205 1206 // UpdateSnapshotWriteOffset mocks base method. 1207 func (m *MockWorkspace) UpdateSnapshotWriteOffset() { 1208 m.ctrl.T.Helper() 1209 m.ctrl.Call(m, "UpdateSnapshotWriteOffset") 1210 } 1211 1212 // UpdateSnapshotWriteOffset indicates an expected call of UpdateSnapshotWriteOffset. 1213 func (mr *MockWorkspaceMockRecorder) UpdateSnapshotWriteOffset() *gomock.Call { 1214 mr.mock.ctrl.T.Helper() 1215 return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateSnapshotWriteOffset", reflect.TypeOf((*MockWorkspace)(nil).UpdateSnapshotWriteOffset)) 1216 }