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  }