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 }