github.com/matrixorigin/matrixone@v0.7.0/pkg/frontend/test/engine_mock.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: ../../../pkg/vm/engine/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  
    11  	gomock "github.com/golang/mock/gomock"
    12  	mpool "github.com/matrixorigin/matrixone/pkg/common/mpool"
    13  	batch "github.com/matrixorigin/matrixone/pkg/container/batch"
    14  	plan "github.com/matrixorigin/matrixone/pkg/pb/plan"
    15  	timestamp "github.com/matrixorigin/matrixone/pkg/pb/timestamp"
    16  	client "github.com/matrixorigin/matrixone/pkg/txn/client"
    17  	engine "github.com/matrixorigin/matrixone/pkg/vm/engine"
    18  )
    19  
    20  // MockStatistics is a mock of Statistics interface.
    21  type MockStatistics struct {
    22  	ctrl     *gomock.Controller
    23  	recorder *MockStatisticsMockRecorder
    24  }
    25  
    26  // MockStatisticsMockRecorder is the mock recorder for MockStatistics.
    27  type MockStatisticsMockRecorder struct {
    28  	mock *MockStatistics
    29  }
    30  
    31  // NewMockStatistics creates a new mock instance.
    32  func NewMockStatistics(ctrl *gomock.Controller) *MockStatistics {
    33  	mock := &MockStatistics{ctrl: ctrl}
    34  	mock.recorder = &MockStatisticsMockRecorder{mock}
    35  	return mock
    36  }
    37  
    38  // EXPECT returns an object that allows the caller to indicate expected use.
    39  func (m *MockStatistics) EXPECT() *MockStatisticsMockRecorder {
    40  	return m.recorder
    41  }
    42  
    43  // Rows mocks base method.
    44  func (m *MockStatistics) Rows(ctx context.Context) (int64, error) {
    45  	m.ctrl.T.Helper()
    46  	ret := m.ctrl.Call(m, "Rows", ctx)
    47  	ret0, _ := ret[0].(int64)
    48  	ret1, _ := ret[1].(error)
    49  	return ret0, ret1
    50  }
    51  
    52  // Rows indicates an expected call of Rows.
    53  func (mr *MockStatisticsMockRecorder) Rows(ctx interface{}) *gomock.Call {
    54  	mr.mock.ctrl.T.Helper()
    55  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Rows", reflect.TypeOf((*MockStatistics)(nil).Rows), ctx)
    56  }
    57  
    58  // Size mocks base method.
    59  func (m *MockStatistics) Size(ctx context.Context, columnName string) (int64, error) {
    60  	m.ctrl.T.Helper()
    61  	ret := m.ctrl.Call(m, "Size", ctx, columnName)
    62  	ret0, _ := ret[0].(int64)
    63  	ret1, _ := ret[1].(error)
    64  	return ret0, ret1
    65  }
    66  
    67  // Size indicates an expected call of Size.
    68  func (mr *MockStatisticsMockRecorder) Size(ctx, columnName interface{}) *gomock.Call {
    69  	mr.mock.ctrl.T.Helper()
    70  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Size", reflect.TypeOf((*MockStatistics)(nil).Size), ctx, columnName)
    71  }
    72  
    73  // Stats mocks base method.
    74  func (m *MockStatistics) Stats(ctx context.Context, expr *plan.Expr) (*plan.Stats, error) {
    75  	m.ctrl.T.Helper()
    76  	ret := m.ctrl.Call(m, "Stats", ctx, expr)
    77  	ret0, _ := ret[0].(*plan.Stats)
    78  	ret1, _ := ret[1].(error)
    79  	return ret0, ret1
    80  }
    81  
    82  // Stats indicates an expected call of Stats.
    83  func (mr *MockStatisticsMockRecorder) Stats(ctx, expr interface{}) *gomock.Call {
    84  	mr.mock.ctrl.T.Helper()
    85  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stats", reflect.TypeOf((*MockStatistics)(nil).Stats), ctx, expr)
    86  }
    87  
    88  // MockTableDef is a mock of TableDef interface.
    89  type MockTableDef struct {
    90  	ctrl     *gomock.Controller
    91  	recorder *MockTableDefMockRecorder
    92  }
    93  
    94  // MockTableDefMockRecorder is the mock recorder for MockTableDef.
    95  type MockTableDefMockRecorder struct {
    96  	mock *MockTableDef
    97  }
    98  
    99  // NewMockTableDef creates a new mock instance.
   100  func NewMockTableDef(ctrl *gomock.Controller) *MockTableDef {
   101  	mock := &MockTableDef{ctrl: ctrl}
   102  	mock.recorder = &MockTableDefMockRecorder{mock}
   103  	return mock
   104  }
   105  
   106  // EXPECT returns an object that allows the caller to indicate expected use.
   107  func (m *MockTableDef) EXPECT() *MockTableDefMockRecorder {
   108  	return m.recorder
   109  }
   110  
   111  // tableDef mocks base method.
   112  func (m *MockTableDef) tableDef() {
   113  	m.ctrl.T.Helper()
   114  	m.ctrl.Call(m, "tableDef")
   115  }
   116  
   117  // tableDef indicates an expected call of tableDef.
   118  func (mr *MockTableDefMockRecorder) tableDef() *gomock.Call {
   119  	mr.mock.ctrl.T.Helper()
   120  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "tableDef", reflect.TypeOf((*MockTableDef)(nil).tableDef))
   121  }
   122  
   123  // MockConstraint is a mock of Constraint interface.
   124  type MockConstraint struct {
   125  	ctrl     *gomock.Controller
   126  	recorder *MockConstraintMockRecorder
   127  }
   128  
   129  // MockConstraintMockRecorder is the mock recorder for MockConstraint.
   130  type MockConstraintMockRecorder struct {
   131  	mock *MockConstraint
   132  }
   133  
   134  // NewMockConstraint creates a new mock instance.
   135  func NewMockConstraint(ctrl *gomock.Controller) *MockConstraint {
   136  	mock := &MockConstraint{ctrl: ctrl}
   137  	mock.recorder = &MockConstraintMockRecorder{mock}
   138  	return mock
   139  }
   140  
   141  // EXPECT returns an object that allows the caller to indicate expected use.
   142  func (m *MockConstraint) EXPECT() *MockConstraintMockRecorder {
   143  	return m.recorder
   144  }
   145  
   146  // constraint mocks base method.
   147  func (m *MockConstraint) constraint() {
   148  	m.ctrl.T.Helper()
   149  	m.ctrl.Call(m, "constraint")
   150  }
   151  
   152  // constraint indicates an expected call of constraint.
   153  func (mr *MockConstraintMockRecorder) constraint() *gomock.Call {
   154  	mr.mock.ctrl.T.Helper()
   155  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "constraint", reflect.TypeOf((*MockConstraint)(nil).constraint))
   156  }
   157  
   158  // MockRelation is a mock of Relation interface.
   159  type MockRelation struct {
   160  	ctrl     *gomock.Controller
   161  	recorder *MockRelationMockRecorder
   162  }
   163  
   164  // MockRelationMockRecorder is the mock recorder for MockRelation.
   165  type MockRelationMockRecorder struct {
   166  	mock *MockRelation
   167  }
   168  
   169  // NewMockRelation creates a new mock instance.
   170  func NewMockRelation(ctrl *gomock.Controller) *MockRelation {
   171  	mock := &MockRelation{ctrl: ctrl}
   172  	mock.recorder = &MockRelationMockRecorder{mock}
   173  	return mock
   174  }
   175  
   176  // EXPECT returns an object that allows the caller to indicate expected use.
   177  func (m *MockRelation) EXPECT() *MockRelationMockRecorder {
   178  	return m.recorder
   179  }
   180  
   181  // AddTableDef mocks base method.
   182  func (m *MockRelation) AddTableDef(arg0 context.Context, arg1 engine.TableDef) error {
   183  	m.ctrl.T.Helper()
   184  	ret := m.ctrl.Call(m, "AddTableDef", arg0, arg1)
   185  	ret0, _ := ret[0].(error)
   186  	return ret0
   187  }
   188  
   189  // AddTableDef indicates an expected call of AddTableDef.
   190  func (mr *MockRelationMockRecorder) AddTableDef(arg0, arg1 interface{}) *gomock.Call {
   191  	mr.mock.ctrl.T.Helper()
   192  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "AddTableDef", reflect.TypeOf((*MockRelation)(nil).AddTableDef), arg0, arg1)
   193  }
   194  
   195  // DelTableDef mocks base method.
   196  func (m *MockRelation) DelTableDef(arg0 context.Context, arg1 engine.TableDef) error {
   197  	m.ctrl.T.Helper()
   198  	ret := m.ctrl.Call(m, "DelTableDef", arg0, arg1)
   199  	ret0, _ := ret[0].(error)
   200  	return ret0
   201  }
   202  
   203  // DelTableDef indicates an expected call of DelTableDef.
   204  func (mr *MockRelationMockRecorder) DelTableDef(arg0, arg1 interface{}) *gomock.Call {
   205  	mr.mock.ctrl.T.Helper()
   206  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DelTableDef", reflect.TypeOf((*MockRelation)(nil).DelTableDef), arg0, arg1)
   207  }
   208  
   209  // Delete mocks base method.
   210  func (m *MockRelation) Delete(arg0 context.Context, arg1 *batch.Batch, arg2 string) error {
   211  	m.ctrl.T.Helper()
   212  	ret := m.ctrl.Call(m, "Delete", arg0, arg1, arg2)
   213  	ret0, _ := ret[0].(error)
   214  	return ret0
   215  }
   216  
   217  // Delete indicates an expected call of Delete.
   218  func (mr *MockRelationMockRecorder) Delete(arg0, arg1, arg2 interface{}) *gomock.Call {
   219  	mr.mock.ctrl.T.Helper()
   220  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockRelation)(nil).Delete), arg0, arg1, arg2)
   221  }
   222  
   223  // GetHideKeys mocks base method.
   224  func (m *MockRelation) GetHideKeys(arg0 context.Context) ([]*engine.Attribute, error) {
   225  	m.ctrl.T.Helper()
   226  	ret := m.ctrl.Call(m, "GetHideKeys", arg0)
   227  	ret0, _ := ret[0].([]*engine.Attribute)
   228  	ret1, _ := ret[1].(error)
   229  	return ret0, ret1
   230  }
   231  
   232  // GetHideKeys indicates an expected call of GetHideKeys.
   233  func (mr *MockRelationMockRecorder) GetHideKeys(arg0 interface{}) *gomock.Call {
   234  	mr.mock.ctrl.T.Helper()
   235  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetHideKeys", reflect.TypeOf((*MockRelation)(nil).GetHideKeys), arg0)
   236  }
   237  
   238  // GetPrimaryKeys mocks base method.
   239  func (m *MockRelation) GetPrimaryKeys(arg0 context.Context) ([]*engine.Attribute, error) {
   240  	m.ctrl.T.Helper()
   241  	ret := m.ctrl.Call(m, "GetPrimaryKeys", arg0)
   242  	ret0, _ := ret[0].([]*engine.Attribute)
   243  	ret1, _ := ret[1].(error)
   244  	return ret0, ret1
   245  }
   246  
   247  // GetPrimaryKeys indicates an expected call of GetPrimaryKeys.
   248  func (mr *MockRelationMockRecorder) GetPrimaryKeys(arg0 interface{}) *gomock.Call {
   249  	mr.mock.ctrl.T.Helper()
   250  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPrimaryKeys", reflect.TypeOf((*MockRelation)(nil).GetPrimaryKeys), arg0)
   251  }
   252  
   253  // GetTableID mocks base method.
   254  func (m *MockRelation) GetTableID(arg0 context.Context) uint64 {
   255  	m.ctrl.T.Helper()
   256  	ret := m.ctrl.Call(m, "GetTableID", arg0)
   257  	ret0, _ := ret[0].(uint64)
   258  	return ret0
   259  }
   260  
   261  // GetTableID indicates an expected call of GetTableID.
   262  func (mr *MockRelationMockRecorder) GetTableID(arg0 interface{}) *gomock.Call {
   263  	mr.mock.ctrl.T.Helper()
   264  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetTableID", reflect.TypeOf((*MockRelation)(nil).GetTableID), arg0)
   265  }
   266  
   267  // MaxAndMinValues mocks base method.
   268  func (m *MockRelation) MaxAndMinValues(ctx context.Context) ([][2]any, []uint8, error) {
   269  	m.ctrl.T.Helper()
   270  	ret := m.ctrl.Call(m, "MaxAndMinValues", ctx)
   271  	ret0, _ := ret[0].([][2]any)
   272  	ret1, _ := ret[1].([]uint8)
   273  	ret2, _ := ret[2].(error)
   274  	return ret0, ret1, ret2
   275  }
   276  
   277  // MaxAndMinValues indicates an expected call of MaxAndMinValues.
   278  func (mr *MockRelationMockRecorder) MaxAndMinValues(ctx interface{}) *gomock.Call {
   279  	mr.mock.ctrl.T.Helper()
   280  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "MaxAndMinValues", reflect.TypeOf((*MockRelation)(nil).MaxAndMinValues), ctx)
   281  }
   282  
   283  // NewReader mocks base method.
   284  func (m *MockRelation) NewReader(arg0 context.Context, arg1 int, arg2 *plan.Expr, arg3 [][]byte) ([]engine.Reader, error) {
   285  	m.ctrl.T.Helper()
   286  	ret := m.ctrl.Call(m, "NewReader", arg0, arg1, arg2, arg3)
   287  	ret0, _ := ret[0].([]engine.Reader)
   288  	ret1, _ := ret[1].(error)
   289  	return ret0, ret1
   290  }
   291  
   292  // NewReader indicates an expected call of NewReader.
   293  func (mr *MockRelationMockRecorder) NewReader(arg0, arg1, arg2, arg3 interface{}) *gomock.Call {
   294  	mr.mock.ctrl.T.Helper()
   295  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NewReader", reflect.TypeOf((*MockRelation)(nil).NewReader), arg0, arg1, arg2, arg3)
   296  }
   297  
   298  // Ranges mocks base method.
   299  func (m *MockRelation) Ranges(arg0 context.Context, arg1 *plan.Expr) ([][]byte, error) {
   300  	m.ctrl.T.Helper()
   301  	ret := m.ctrl.Call(m, "Ranges", arg0, arg1)
   302  	ret0, _ := ret[0].([][]byte)
   303  	ret1, _ := ret[1].(error)
   304  	return ret0, ret1
   305  }
   306  
   307  // Ranges indicates an expected call of Ranges.
   308  func (mr *MockRelationMockRecorder) Ranges(arg0, arg1 interface{}) *gomock.Call {
   309  	mr.mock.ctrl.T.Helper()
   310  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Ranges", reflect.TypeOf((*MockRelation)(nil).Ranges), arg0, arg1)
   311  }
   312  
   313  // Rows mocks base method.
   314  func (m *MockRelation) Rows(ctx context.Context) (int64, error) {
   315  	m.ctrl.T.Helper()
   316  	ret := m.ctrl.Call(m, "Rows", ctx)
   317  	ret0, _ := ret[0].(int64)
   318  	ret1, _ := ret[1].(error)
   319  	return ret0, ret1
   320  }
   321  
   322  // Rows indicates an expected call of Rows.
   323  func (mr *MockRelationMockRecorder) Rows(ctx interface{}) *gomock.Call {
   324  	mr.mock.ctrl.T.Helper()
   325  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Rows", reflect.TypeOf((*MockRelation)(nil).Rows), ctx)
   326  }
   327  
   328  // Size mocks base method.
   329  func (m *MockRelation) Size(ctx context.Context, columnName string) (int64, error) {
   330  	m.ctrl.T.Helper()
   331  	ret := m.ctrl.Call(m, "Size", ctx, columnName)
   332  	ret0, _ := ret[0].(int64)
   333  	ret1, _ := ret[1].(error)
   334  	return ret0, ret1
   335  }
   336  
   337  // Size indicates an expected call of Size.
   338  func (mr *MockRelationMockRecorder) Size(ctx, columnName interface{}) *gomock.Call {
   339  	mr.mock.ctrl.T.Helper()
   340  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Size", reflect.TypeOf((*MockRelation)(nil).Size), ctx, columnName)
   341  }
   342  
   343  // Stats mocks base method.
   344  func (m *MockRelation) Stats(ctx context.Context, expr *plan.Expr) (*plan.Stats, error) {
   345  	m.ctrl.T.Helper()
   346  	ret := m.ctrl.Call(m, "Stats", ctx, expr)
   347  	ret0, _ := ret[0].(*plan.Stats)
   348  	ret1, _ := ret[1].(error)
   349  	return ret0, ret1
   350  }
   351  
   352  // Stats indicates an expected call of Stats.
   353  func (mr *MockRelationMockRecorder) Stats(ctx, expr interface{}) *gomock.Call {
   354  	mr.mock.ctrl.T.Helper()
   355  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stats", reflect.TypeOf((*MockRelation)(nil).Stats), ctx, expr)
   356  }
   357  
   358  // TableColumns mocks base method.
   359  func (m *MockRelation) TableColumns(ctx context.Context) ([]*engine.Attribute, error) {
   360  	m.ctrl.T.Helper()
   361  	ret := m.ctrl.Call(m, "TableColumns", ctx)
   362  	ret0, _ := ret[0].([]*engine.Attribute)
   363  	ret1, _ := ret[1].(error)
   364  	return ret0, ret1
   365  }
   366  
   367  // TableColumns indicates an expected call of TableColumns.
   368  func (mr *MockRelationMockRecorder) TableColumns(ctx interface{}) *gomock.Call {
   369  	mr.mock.ctrl.T.Helper()
   370  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TableColumns", reflect.TypeOf((*MockRelation)(nil).TableColumns), ctx)
   371  }
   372  
   373  // TableDefs mocks base method.
   374  func (m *MockRelation) TableDefs(arg0 context.Context) ([]engine.TableDef, error) {
   375  	m.ctrl.T.Helper()
   376  	ret := m.ctrl.Call(m, "TableDefs", arg0)
   377  	ret0, _ := ret[0].([]engine.TableDef)
   378  	ret1, _ := ret[1].(error)
   379  	return ret0, ret1
   380  }
   381  
   382  // TableDefs indicates an expected call of TableDefs.
   383  func (mr *MockRelationMockRecorder) TableDefs(arg0 interface{}) *gomock.Call {
   384  	mr.mock.ctrl.T.Helper()
   385  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "TableDefs", reflect.TypeOf((*MockRelation)(nil).TableDefs), arg0)
   386  }
   387  
   388  // Update mocks base method.
   389  func (m *MockRelation) Update(arg0 context.Context, arg1 *batch.Batch) error {
   390  	m.ctrl.T.Helper()
   391  	ret := m.ctrl.Call(m, "Update", arg0, arg1)
   392  	ret0, _ := ret[0].(error)
   393  	return ret0
   394  }
   395  
   396  // Update indicates an expected call of Update.
   397  func (mr *MockRelationMockRecorder) Update(arg0, arg1 interface{}) *gomock.Call {
   398  	mr.mock.ctrl.T.Helper()
   399  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockRelation)(nil).Update), arg0, arg1)
   400  }
   401  
   402  // UpdateConstraint mocks base method.
   403  func (m *MockRelation) UpdateConstraint(arg0 context.Context, arg1 *engine.ConstraintDef) error {
   404  	m.ctrl.T.Helper()
   405  	ret := m.ctrl.Call(m, "UpdateConstraint", arg0, arg1)
   406  	ret0, _ := ret[0].(error)
   407  	return ret0
   408  }
   409  
   410  // UpdateConstraint indicates an expected call of UpdateConstraint.
   411  func (mr *MockRelationMockRecorder) UpdateConstraint(arg0, arg1 interface{}) *gomock.Call {
   412  	mr.mock.ctrl.T.Helper()
   413  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateConstraint", reflect.TypeOf((*MockRelation)(nil).UpdateConstraint), arg0, arg1)
   414  }
   415  
   416  // Write mocks base method.
   417  func (m *MockRelation) Write(arg0 context.Context, arg1 *batch.Batch) error {
   418  	m.ctrl.T.Helper()
   419  	ret := m.ctrl.Call(m, "Write", arg0, arg1)
   420  	ret0, _ := ret[0].(error)
   421  	return ret0
   422  }
   423  
   424  // Write indicates an expected call of Write.
   425  func (mr *MockRelationMockRecorder) Write(arg0, arg1 interface{}) *gomock.Call {
   426  	mr.mock.ctrl.T.Helper()
   427  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Write", reflect.TypeOf((*MockRelation)(nil).Write), arg0, arg1)
   428  }
   429  
   430  // MockReader is a mock of Reader interface.
   431  type MockReader struct {
   432  	ctrl     *gomock.Controller
   433  	recorder *MockReaderMockRecorder
   434  }
   435  
   436  // MockReaderMockRecorder is the mock recorder for MockReader.
   437  type MockReaderMockRecorder struct {
   438  	mock *MockReader
   439  }
   440  
   441  // NewMockReader creates a new mock instance.
   442  func NewMockReader(ctrl *gomock.Controller) *MockReader {
   443  	mock := &MockReader{ctrl: ctrl}
   444  	mock.recorder = &MockReaderMockRecorder{mock}
   445  	return mock
   446  }
   447  
   448  // EXPECT returns an object that allows the caller to indicate expected use.
   449  func (m *MockReader) EXPECT() *MockReaderMockRecorder {
   450  	return m.recorder
   451  }
   452  
   453  // Close mocks base method.
   454  func (m *MockReader) Close() error {
   455  	m.ctrl.T.Helper()
   456  	ret := m.ctrl.Call(m, "Close")
   457  	ret0, _ := ret[0].(error)
   458  	return ret0
   459  }
   460  
   461  // Close indicates an expected call of Close.
   462  func (mr *MockReaderMockRecorder) Close() *gomock.Call {
   463  	mr.mock.ctrl.T.Helper()
   464  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Close", reflect.TypeOf((*MockReader)(nil).Close))
   465  }
   466  
   467  // Read mocks base method.
   468  func (m *MockReader) Read(arg0 context.Context, arg1 []string, arg2 *plan.Expr, arg3 *mpool.MPool) (*batch.Batch, error) {
   469  	m.ctrl.T.Helper()
   470  	ret := m.ctrl.Call(m, "Read", arg0, arg1, arg2, arg3)
   471  	ret0, _ := ret[0].(*batch.Batch)
   472  	ret1, _ := ret[1].(error)
   473  	return ret0, ret1
   474  }
   475  
   476  // Read indicates an expected call of Read.
   477  func (mr *MockReaderMockRecorder) Read(arg0, arg1, arg2, arg3 interface{}) *gomock.Call {
   478  	mr.mock.ctrl.T.Helper()
   479  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Read", reflect.TypeOf((*MockReader)(nil).Read), arg0, arg1, arg2, arg3)
   480  }
   481  
   482  // MockDatabase is a mock of Database interface.
   483  type MockDatabase struct {
   484  	ctrl     *gomock.Controller
   485  	recorder *MockDatabaseMockRecorder
   486  }
   487  
   488  // MockDatabaseMockRecorder is the mock recorder for MockDatabase.
   489  type MockDatabaseMockRecorder struct {
   490  	mock *MockDatabase
   491  }
   492  
   493  // NewMockDatabase creates a new mock instance.
   494  func NewMockDatabase(ctrl *gomock.Controller) *MockDatabase {
   495  	mock := &MockDatabase{ctrl: ctrl}
   496  	mock.recorder = &MockDatabaseMockRecorder{mock}
   497  	return mock
   498  }
   499  
   500  // EXPECT returns an object that allows the caller to indicate expected use.
   501  func (m *MockDatabase) EXPECT() *MockDatabaseMockRecorder {
   502  	return m.recorder
   503  }
   504  
   505  // Create mocks base method.
   506  func (m *MockDatabase) Create(arg0 context.Context, arg1 string, arg2 []engine.TableDef) error {
   507  	m.ctrl.T.Helper()
   508  	ret := m.ctrl.Call(m, "Create", arg0, arg1, arg2)
   509  	ret0, _ := ret[0].(error)
   510  	return ret0
   511  }
   512  
   513  // Create indicates an expected call of Create.
   514  func (mr *MockDatabaseMockRecorder) Create(arg0, arg1, arg2 interface{}) *gomock.Call {
   515  	mr.mock.ctrl.T.Helper()
   516  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockDatabase)(nil).Create), arg0, arg1, arg2)
   517  }
   518  
   519  // Delete mocks base method.
   520  func (m *MockDatabase) Delete(arg0 context.Context, arg1 string) error {
   521  	m.ctrl.T.Helper()
   522  	ret := m.ctrl.Call(m, "Delete", arg0, arg1)
   523  	ret0, _ := ret[0].(error)
   524  	return ret0
   525  }
   526  
   527  // Delete indicates an expected call of Delete.
   528  func (mr *MockDatabaseMockRecorder) Delete(arg0, arg1 interface{}) *gomock.Call {
   529  	mr.mock.ctrl.T.Helper()
   530  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockDatabase)(nil).Delete), arg0, arg1)
   531  }
   532  
   533  // GetDatabaseId mocks base method.
   534  func (m *MockDatabase) GetDatabaseId(arg0 context.Context) string {
   535  	m.ctrl.T.Helper()
   536  	ret := m.ctrl.Call(m, "GetDatabaseId", arg0)
   537  	ret0, _ := ret[0].(string)
   538  	return ret0
   539  }
   540  
   541  // GetDatabaseId indicates an expected call of GetDatabaseId.
   542  func (mr *MockDatabaseMockRecorder) GetDatabaseId(arg0 interface{}) *gomock.Call {
   543  	mr.mock.ctrl.T.Helper()
   544  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDatabaseId", reflect.TypeOf((*MockDatabase)(nil).GetDatabaseId), arg0)
   545  }
   546  
   547  // Relation mocks base method.
   548  func (m *MockDatabase) Relation(arg0 context.Context, arg1 string) (engine.Relation, error) {
   549  	m.ctrl.T.Helper()
   550  	ret := m.ctrl.Call(m, "Relation", arg0, arg1)
   551  	ret0, _ := ret[0].(engine.Relation)
   552  	ret1, _ := ret[1].(error)
   553  	return ret0, ret1
   554  }
   555  
   556  // Relation indicates an expected call of Relation.
   557  func (mr *MockDatabaseMockRecorder) Relation(arg0, arg1 interface{}) *gomock.Call {
   558  	mr.mock.ctrl.T.Helper()
   559  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Relation", reflect.TypeOf((*MockDatabase)(nil).Relation), arg0, arg1)
   560  }
   561  
   562  // Relations mocks base method.
   563  func (m *MockDatabase) Relations(arg0 context.Context) ([]string, error) {
   564  	m.ctrl.T.Helper()
   565  	ret := m.ctrl.Call(m, "Relations", arg0)
   566  	ret0, _ := ret[0].([]string)
   567  	ret1, _ := ret[1].(error)
   568  	return ret0, ret1
   569  }
   570  
   571  // Relations indicates an expected call of Relations.
   572  func (mr *MockDatabaseMockRecorder) Relations(arg0 interface{}) *gomock.Call {
   573  	mr.mock.ctrl.T.Helper()
   574  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Relations", reflect.TypeOf((*MockDatabase)(nil).Relations), arg0)
   575  }
   576  
   577  // Truncate mocks base method.
   578  func (m *MockDatabase) Truncate(arg0 context.Context, arg1 string) (uint64, error) {
   579  	m.ctrl.T.Helper()
   580  	ret := m.ctrl.Call(m, "Truncate", arg0, arg1)
   581  	ret0, _ := ret[0].(uint64)
   582  	ret1, _ := ret[1].(error)
   583  	return ret0, ret1
   584  }
   585  
   586  // Truncate indicates an expected call of Truncate.
   587  func (mr *MockDatabaseMockRecorder) Truncate(arg0, arg1 interface{}) *gomock.Call {
   588  	mr.mock.ctrl.T.Helper()
   589  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Truncate", reflect.TypeOf((*MockDatabase)(nil).Truncate), arg0, arg1)
   590  }
   591  
   592  // MockEngine is a mock of Engine interface.
   593  type MockEngine struct {
   594  	ctrl     *gomock.Controller
   595  	recorder *MockEngineMockRecorder
   596  }
   597  
   598  // MockEngineMockRecorder is the mock recorder for MockEngine.
   599  type MockEngineMockRecorder struct {
   600  	mock *MockEngine
   601  }
   602  
   603  // NewMockEngine creates a new mock instance.
   604  func NewMockEngine(ctrl *gomock.Controller) *MockEngine {
   605  	mock := &MockEngine{ctrl: ctrl}
   606  	mock.recorder = &MockEngineMockRecorder{mock}
   607  	return mock
   608  }
   609  
   610  // EXPECT returns an object that allows the caller to indicate expected use.
   611  func (m *MockEngine) EXPECT() *MockEngineMockRecorder {
   612  	return m.recorder
   613  }
   614  
   615  // Commit mocks base method.
   616  func (m *MockEngine) Commit(ctx context.Context, op client.TxnOperator) error {
   617  	m.ctrl.T.Helper()
   618  	ret := m.ctrl.Call(m, "Commit", ctx, op)
   619  	ret0, _ := ret[0].(error)
   620  	return ret0
   621  }
   622  
   623  // Commit indicates an expected call of Commit.
   624  func (mr *MockEngineMockRecorder) Commit(ctx, op interface{}) *gomock.Call {
   625  	mr.mock.ctrl.T.Helper()
   626  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Commit", reflect.TypeOf((*MockEngine)(nil).Commit), ctx, op)
   627  }
   628  
   629  // Create mocks base method.
   630  func (m *MockEngine) Create(ctx context.Context, databaseName string, op client.TxnOperator) error {
   631  	m.ctrl.T.Helper()
   632  	ret := m.ctrl.Call(m, "Create", ctx, databaseName, op)
   633  	ret0, _ := ret[0].(error)
   634  	return ret0
   635  }
   636  
   637  // Create indicates an expected call of Create.
   638  func (mr *MockEngineMockRecorder) Create(ctx, databaseName, op interface{}) *gomock.Call {
   639  	mr.mock.ctrl.T.Helper()
   640  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Create", reflect.TypeOf((*MockEngine)(nil).Create), ctx, databaseName, op)
   641  }
   642  
   643  // Database mocks base method.
   644  func (m *MockEngine) Database(ctx context.Context, databaseName string, op client.TxnOperator) (engine.Database, error) {
   645  	m.ctrl.T.Helper()
   646  	ret := m.ctrl.Call(m, "Database", ctx, databaseName, op)
   647  	ret0, _ := ret[0].(engine.Database)
   648  	ret1, _ := ret[1].(error)
   649  	return ret0, ret1
   650  }
   651  
   652  // Database indicates an expected call of Database.
   653  func (mr *MockEngineMockRecorder) Database(ctx, databaseName, op interface{}) *gomock.Call {
   654  	mr.mock.ctrl.T.Helper()
   655  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Database", reflect.TypeOf((*MockEngine)(nil).Database), ctx, databaseName, op)
   656  }
   657  
   658  // Databases mocks base method.
   659  func (m *MockEngine) Databases(ctx context.Context, op client.TxnOperator) ([]string, error) {
   660  	m.ctrl.T.Helper()
   661  	ret := m.ctrl.Call(m, "Databases", ctx, op)
   662  	ret0, _ := ret[0].([]string)
   663  	ret1, _ := ret[1].(error)
   664  	return ret0, ret1
   665  }
   666  
   667  // Databases indicates an expected call of Databases.
   668  func (mr *MockEngineMockRecorder) Databases(ctx, op interface{}) *gomock.Call {
   669  	mr.mock.ctrl.T.Helper()
   670  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Databases", reflect.TypeOf((*MockEngine)(nil).Databases), ctx, op)
   671  }
   672  
   673  // Delete mocks base method.
   674  func (m *MockEngine) Delete(ctx context.Context, databaseName string, op client.TxnOperator) error {
   675  	m.ctrl.T.Helper()
   676  	ret := m.ctrl.Call(m, "Delete", ctx, databaseName, op)
   677  	ret0, _ := ret[0].(error)
   678  	return ret0
   679  }
   680  
   681  // Delete indicates an expected call of Delete.
   682  func (mr *MockEngineMockRecorder) Delete(ctx, databaseName, op interface{}) *gomock.Call {
   683  	mr.mock.ctrl.T.Helper()
   684  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockEngine)(nil).Delete), ctx, databaseName, op)
   685  }
   686  
   687  // GetNameById mocks base method.
   688  func (m *MockEngine) GetNameById(ctx context.Context, op client.TxnOperator, tableId uint64) (string, string, error) {
   689  	m.ctrl.T.Helper()
   690  	ret := m.ctrl.Call(m, "GetNameById", ctx, op, tableId)
   691  	ret0, _ := ret[0].(string)
   692  	ret1, _ := ret[1].(string)
   693  	ret2, _ := ret[2].(error)
   694  	return ret0, ret1, ret2
   695  }
   696  
   697  // GetNameById indicates an expected call of GetNameById.
   698  func (mr *MockEngineMockRecorder) GetNameById(ctx, op, tableId interface{}) *gomock.Call {
   699  	mr.mock.ctrl.T.Helper()
   700  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNameById", reflect.TypeOf((*MockEngine)(nil).GetNameById), ctx, op, tableId)
   701  }
   702  
   703  // GetRelationById mocks base method.
   704  func (m *MockEngine) GetRelationById(ctx context.Context, op client.TxnOperator, tableId uint64) (string, string, engine.Relation, error) {
   705  	m.ctrl.T.Helper()
   706  	ret := m.ctrl.Call(m, "GetRelationById", ctx, op, tableId)
   707  	ret0, _ := ret[0].(string)
   708  	ret1, _ := ret[1].(string)
   709  	ret2, _ := ret[2].(engine.Relation)
   710  	ret3, _ := ret[3].(error)
   711  	return ret0, ret1, ret2, ret3
   712  }
   713  
   714  // GetRelationById indicates an expected call of GetRelationById.
   715  func (mr *MockEngineMockRecorder) GetRelationById(ctx, op, tableId interface{}) *gomock.Call {
   716  	mr.mock.ctrl.T.Helper()
   717  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRelationById", reflect.TypeOf((*MockEngine)(nil).GetRelationById), ctx, op, tableId)
   718  }
   719  
   720  // Hints mocks base method.
   721  func (m *MockEngine) Hints() engine.Hints {
   722  	m.ctrl.T.Helper()
   723  	ret := m.ctrl.Call(m, "Hints")
   724  	ret0, _ := ret[0].(engine.Hints)
   725  	return ret0
   726  }
   727  
   728  // Hints indicates an expected call of Hints.
   729  func (mr *MockEngineMockRecorder) Hints() *gomock.Call {
   730  	mr.mock.ctrl.T.Helper()
   731  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Hints", reflect.TypeOf((*MockEngine)(nil).Hints))
   732  }
   733  
   734  // New mocks base method.
   735  func (m *MockEngine) New(ctx context.Context, op client.TxnOperator) error {
   736  	m.ctrl.T.Helper()
   737  	ret := m.ctrl.Call(m, "New", ctx, op)
   738  	ret0, _ := ret[0].(error)
   739  	return ret0
   740  }
   741  
   742  // New indicates an expected call of New.
   743  func (mr *MockEngineMockRecorder) New(ctx, op interface{}) *gomock.Call {
   744  	mr.mock.ctrl.T.Helper()
   745  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "New", reflect.TypeOf((*MockEngine)(nil).New), ctx, op)
   746  }
   747  
   748  // NewBlockReader mocks base method.
   749  func (m *MockEngine) NewBlockReader(ctx context.Context, num int, ts timestamp.Timestamp, expr *plan.Expr, ranges [][]byte, tblDef *plan.TableDef) ([]engine.Reader, error) {
   750  	m.ctrl.T.Helper()
   751  	ret := m.ctrl.Call(m, "NewBlockReader", ctx, num, ts, expr, ranges, tblDef)
   752  	ret0, _ := ret[0].([]engine.Reader)
   753  	ret1, _ := ret[1].(error)
   754  	return ret0, ret1
   755  }
   756  
   757  // NewBlockReader indicates an expected call of NewBlockReader.
   758  func (mr *MockEngineMockRecorder) NewBlockReader(ctx, num, ts, expr, ranges, tblDef interface{}) *gomock.Call {
   759  	mr.mock.ctrl.T.Helper()
   760  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NewBlockReader", reflect.TypeOf((*MockEngine)(nil).NewBlockReader), ctx, num, ts, expr, ranges, tblDef)
   761  }
   762  
   763  // Nodes mocks base method.
   764  func (m *MockEngine) Nodes() (engine.Nodes, error) {
   765  	m.ctrl.T.Helper()
   766  	ret := m.ctrl.Call(m, "Nodes")
   767  	ret0, _ := ret[0].(engine.Nodes)
   768  	ret1, _ := ret[1].(error)
   769  	return ret0, ret1
   770  }
   771  
   772  // Nodes indicates an expected call of Nodes.
   773  func (mr *MockEngineMockRecorder) Nodes() *gomock.Call {
   774  	mr.mock.ctrl.T.Helper()
   775  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Nodes", reflect.TypeOf((*MockEngine)(nil).Nodes))
   776  }
   777  
   778  // Rollback mocks base method.
   779  func (m *MockEngine) Rollback(ctx context.Context, op client.TxnOperator) error {
   780  	m.ctrl.T.Helper()
   781  	ret := m.ctrl.Call(m, "Rollback", ctx, op)
   782  	ret0, _ := ret[0].(error)
   783  	return ret0
   784  }
   785  
   786  // Rollback indicates an expected call of Rollback.
   787  func (mr *MockEngineMockRecorder) Rollback(ctx, op interface{}) *gomock.Call {
   788  	mr.mock.ctrl.T.Helper()
   789  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Rollback", reflect.TypeOf((*MockEngine)(nil).Rollback), ctx, op)
   790  }