github.com/grafana/pyroscope@v1.18.0/pkg/test/mocks/mockobjstore/mock_bucket.go (about) 1 // Code generated by mockery. DO NOT EDIT. 2 3 package mockobjstore 4 5 import ( 6 context "context" 7 io "io" 8 9 mock "github.com/stretchr/testify/mock" 10 11 objstore "github.com/thanos-io/objstore" 12 13 pkgobjstore "github.com/grafana/pyroscope/pkg/objstore" 14 ) 15 16 // MockBucket is an autogenerated mock type for the Bucket type 17 type MockBucket struct { 18 mock.Mock 19 } 20 21 type MockBucket_Expecter struct { 22 mock *mock.Mock 23 } 24 25 func (_m *MockBucket) EXPECT() *MockBucket_Expecter { 26 return &MockBucket_Expecter{mock: &_m.Mock} 27 } 28 29 // Attributes provides a mock function with given fields: ctx, name 30 func (_m *MockBucket) Attributes(ctx context.Context, name string) (objstore.ObjectAttributes, error) { 31 ret := _m.Called(ctx, name) 32 33 if len(ret) == 0 { 34 panic("no return value specified for Attributes") 35 } 36 37 var r0 objstore.ObjectAttributes 38 var r1 error 39 if rf, ok := ret.Get(0).(func(context.Context, string) (objstore.ObjectAttributes, error)); ok { 40 return rf(ctx, name) 41 } 42 if rf, ok := ret.Get(0).(func(context.Context, string) objstore.ObjectAttributes); ok { 43 r0 = rf(ctx, name) 44 } else { 45 r0 = ret.Get(0).(objstore.ObjectAttributes) 46 } 47 48 if rf, ok := ret.Get(1).(func(context.Context, string) error); ok { 49 r1 = rf(ctx, name) 50 } else { 51 r1 = ret.Error(1) 52 } 53 54 return r0, r1 55 } 56 57 // MockBucket_Attributes_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Attributes' 58 type MockBucket_Attributes_Call struct { 59 *mock.Call 60 } 61 62 // Attributes is a helper method to define mock.On call 63 // - ctx context.Context 64 // - name string 65 func (_e *MockBucket_Expecter) Attributes(ctx interface{}, name interface{}) *MockBucket_Attributes_Call { 66 return &MockBucket_Attributes_Call{Call: _e.mock.On("Attributes", ctx, name)} 67 } 68 69 func (_c *MockBucket_Attributes_Call) Run(run func(ctx context.Context, name string)) *MockBucket_Attributes_Call { 70 _c.Call.Run(func(args mock.Arguments) { 71 run(args[0].(context.Context), args[1].(string)) 72 }) 73 return _c 74 } 75 76 func (_c *MockBucket_Attributes_Call) Return(_a0 objstore.ObjectAttributes, _a1 error) *MockBucket_Attributes_Call { 77 _c.Call.Return(_a0, _a1) 78 return _c 79 } 80 81 func (_c *MockBucket_Attributes_Call) RunAndReturn(run func(context.Context, string) (objstore.ObjectAttributes, error)) *MockBucket_Attributes_Call { 82 _c.Call.Return(run) 83 return _c 84 } 85 86 // Close provides a mock function with no fields 87 func (_m *MockBucket) Close() error { 88 ret := _m.Called() 89 90 if len(ret) == 0 { 91 panic("no return value specified for Close") 92 } 93 94 var r0 error 95 if rf, ok := ret.Get(0).(func() error); ok { 96 r0 = rf() 97 } else { 98 r0 = ret.Error(0) 99 } 100 101 return r0 102 } 103 104 // MockBucket_Close_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Close' 105 type MockBucket_Close_Call struct { 106 *mock.Call 107 } 108 109 // Close is a helper method to define mock.On call 110 func (_e *MockBucket_Expecter) Close() *MockBucket_Close_Call { 111 return &MockBucket_Close_Call{Call: _e.mock.On("Close")} 112 } 113 114 func (_c *MockBucket_Close_Call) Run(run func()) *MockBucket_Close_Call { 115 _c.Call.Run(func(args mock.Arguments) { 116 run() 117 }) 118 return _c 119 } 120 121 func (_c *MockBucket_Close_Call) Return(_a0 error) *MockBucket_Close_Call { 122 _c.Call.Return(_a0) 123 return _c 124 } 125 126 func (_c *MockBucket_Close_Call) RunAndReturn(run func() error) *MockBucket_Close_Call { 127 _c.Call.Return(run) 128 return _c 129 } 130 131 // Delete provides a mock function with given fields: ctx, name 132 func (_m *MockBucket) Delete(ctx context.Context, name string) error { 133 ret := _m.Called(ctx, name) 134 135 if len(ret) == 0 { 136 panic("no return value specified for Delete") 137 } 138 139 var r0 error 140 if rf, ok := ret.Get(0).(func(context.Context, string) error); ok { 141 r0 = rf(ctx, name) 142 } else { 143 r0 = ret.Error(0) 144 } 145 146 return r0 147 } 148 149 // MockBucket_Delete_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Delete' 150 type MockBucket_Delete_Call struct { 151 *mock.Call 152 } 153 154 // Delete is a helper method to define mock.On call 155 // - ctx context.Context 156 // - name string 157 func (_e *MockBucket_Expecter) Delete(ctx interface{}, name interface{}) *MockBucket_Delete_Call { 158 return &MockBucket_Delete_Call{Call: _e.mock.On("Delete", ctx, name)} 159 } 160 161 func (_c *MockBucket_Delete_Call) Run(run func(ctx context.Context, name string)) *MockBucket_Delete_Call { 162 _c.Call.Run(func(args mock.Arguments) { 163 run(args[0].(context.Context), args[1].(string)) 164 }) 165 return _c 166 } 167 168 func (_c *MockBucket_Delete_Call) Return(_a0 error) *MockBucket_Delete_Call { 169 _c.Call.Return(_a0) 170 return _c 171 } 172 173 func (_c *MockBucket_Delete_Call) RunAndReturn(run func(context.Context, string) error) *MockBucket_Delete_Call { 174 _c.Call.Return(run) 175 return _c 176 } 177 178 // Exists provides a mock function with given fields: ctx, name 179 func (_m *MockBucket) Exists(ctx context.Context, name string) (bool, error) { 180 ret := _m.Called(ctx, name) 181 182 if len(ret) == 0 { 183 panic("no return value specified for Exists") 184 } 185 186 var r0 bool 187 var r1 error 188 if rf, ok := ret.Get(0).(func(context.Context, string) (bool, error)); ok { 189 return rf(ctx, name) 190 } 191 if rf, ok := ret.Get(0).(func(context.Context, string) bool); ok { 192 r0 = rf(ctx, name) 193 } else { 194 r0 = ret.Get(0).(bool) 195 } 196 197 if rf, ok := ret.Get(1).(func(context.Context, string) error); ok { 198 r1 = rf(ctx, name) 199 } else { 200 r1 = ret.Error(1) 201 } 202 203 return r0, r1 204 } 205 206 // MockBucket_Exists_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Exists' 207 type MockBucket_Exists_Call struct { 208 *mock.Call 209 } 210 211 // Exists is a helper method to define mock.On call 212 // - ctx context.Context 213 // - name string 214 func (_e *MockBucket_Expecter) Exists(ctx interface{}, name interface{}) *MockBucket_Exists_Call { 215 return &MockBucket_Exists_Call{Call: _e.mock.On("Exists", ctx, name)} 216 } 217 218 func (_c *MockBucket_Exists_Call) Run(run func(ctx context.Context, name string)) *MockBucket_Exists_Call { 219 _c.Call.Run(func(args mock.Arguments) { 220 run(args[0].(context.Context), args[1].(string)) 221 }) 222 return _c 223 } 224 225 func (_c *MockBucket_Exists_Call) Return(_a0 bool, _a1 error) *MockBucket_Exists_Call { 226 _c.Call.Return(_a0, _a1) 227 return _c 228 } 229 230 func (_c *MockBucket_Exists_Call) RunAndReturn(run func(context.Context, string) (bool, error)) *MockBucket_Exists_Call { 231 _c.Call.Return(run) 232 return _c 233 } 234 235 // Get provides a mock function with given fields: ctx, name 236 func (_m *MockBucket) Get(ctx context.Context, name string) (io.ReadCloser, error) { 237 ret := _m.Called(ctx, name) 238 239 if len(ret) == 0 { 240 panic("no return value specified for Get") 241 } 242 243 var r0 io.ReadCloser 244 var r1 error 245 if rf, ok := ret.Get(0).(func(context.Context, string) (io.ReadCloser, error)); ok { 246 return rf(ctx, name) 247 } 248 if rf, ok := ret.Get(0).(func(context.Context, string) io.ReadCloser); ok { 249 r0 = rf(ctx, name) 250 } else { 251 if ret.Get(0) != nil { 252 r0 = ret.Get(0).(io.ReadCloser) 253 } 254 } 255 256 if rf, ok := ret.Get(1).(func(context.Context, string) error); ok { 257 r1 = rf(ctx, name) 258 } else { 259 r1 = ret.Error(1) 260 } 261 262 return r0, r1 263 } 264 265 // MockBucket_Get_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Get' 266 type MockBucket_Get_Call struct { 267 *mock.Call 268 } 269 270 // Get is a helper method to define mock.On call 271 // - ctx context.Context 272 // - name string 273 func (_e *MockBucket_Expecter) Get(ctx interface{}, name interface{}) *MockBucket_Get_Call { 274 return &MockBucket_Get_Call{Call: _e.mock.On("Get", ctx, name)} 275 } 276 277 func (_c *MockBucket_Get_Call) Run(run func(ctx context.Context, name string)) *MockBucket_Get_Call { 278 _c.Call.Run(func(args mock.Arguments) { 279 run(args[0].(context.Context), args[1].(string)) 280 }) 281 return _c 282 } 283 284 func (_c *MockBucket_Get_Call) Return(_a0 io.ReadCloser, _a1 error) *MockBucket_Get_Call { 285 _c.Call.Return(_a0, _a1) 286 return _c 287 } 288 289 func (_c *MockBucket_Get_Call) RunAndReturn(run func(context.Context, string) (io.ReadCloser, error)) *MockBucket_Get_Call { 290 _c.Call.Return(run) 291 return _c 292 } 293 294 // GetRange provides a mock function with given fields: ctx, name, off, length 295 func (_m *MockBucket) GetRange(ctx context.Context, name string, off int64, length int64) (io.ReadCloser, error) { 296 ret := _m.Called(ctx, name, off, length) 297 298 if len(ret) == 0 { 299 panic("no return value specified for GetRange") 300 } 301 302 var r0 io.ReadCloser 303 var r1 error 304 if rf, ok := ret.Get(0).(func(context.Context, string, int64, int64) (io.ReadCloser, error)); ok { 305 return rf(ctx, name, off, length) 306 } 307 if rf, ok := ret.Get(0).(func(context.Context, string, int64, int64) io.ReadCloser); ok { 308 r0 = rf(ctx, name, off, length) 309 } else { 310 if ret.Get(0) != nil { 311 r0 = ret.Get(0).(io.ReadCloser) 312 } 313 } 314 315 if rf, ok := ret.Get(1).(func(context.Context, string, int64, int64) error); ok { 316 r1 = rf(ctx, name, off, length) 317 } else { 318 r1 = ret.Error(1) 319 } 320 321 return r0, r1 322 } 323 324 // MockBucket_GetRange_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetRange' 325 type MockBucket_GetRange_Call struct { 326 *mock.Call 327 } 328 329 // GetRange is a helper method to define mock.On call 330 // - ctx context.Context 331 // - name string 332 // - off int64 333 // - length int64 334 func (_e *MockBucket_Expecter) GetRange(ctx interface{}, name interface{}, off interface{}, length interface{}) *MockBucket_GetRange_Call { 335 return &MockBucket_GetRange_Call{Call: _e.mock.On("GetRange", ctx, name, off, length)} 336 } 337 338 func (_c *MockBucket_GetRange_Call) Run(run func(ctx context.Context, name string, off int64, length int64)) *MockBucket_GetRange_Call { 339 _c.Call.Run(func(args mock.Arguments) { 340 run(args[0].(context.Context), args[1].(string), args[2].(int64), args[3].(int64)) 341 }) 342 return _c 343 } 344 345 func (_c *MockBucket_GetRange_Call) Return(_a0 io.ReadCloser, _a1 error) *MockBucket_GetRange_Call { 346 _c.Call.Return(_a0, _a1) 347 return _c 348 } 349 350 func (_c *MockBucket_GetRange_Call) RunAndReturn(run func(context.Context, string, int64, int64) (io.ReadCloser, error)) *MockBucket_GetRange_Call { 351 _c.Call.Return(run) 352 return _c 353 } 354 355 // IsAccessDeniedErr provides a mock function with given fields: err 356 func (_m *MockBucket) IsAccessDeniedErr(err error) bool { 357 ret := _m.Called(err) 358 359 if len(ret) == 0 { 360 panic("no return value specified for IsAccessDeniedErr") 361 } 362 363 var r0 bool 364 if rf, ok := ret.Get(0).(func(error) bool); ok { 365 r0 = rf(err) 366 } else { 367 r0 = ret.Get(0).(bool) 368 } 369 370 return r0 371 } 372 373 // MockBucket_IsAccessDeniedErr_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'IsAccessDeniedErr' 374 type MockBucket_IsAccessDeniedErr_Call struct { 375 *mock.Call 376 } 377 378 // IsAccessDeniedErr is a helper method to define mock.On call 379 // - err error 380 func (_e *MockBucket_Expecter) IsAccessDeniedErr(err interface{}) *MockBucket_IsAccessDeniedErr_Call { 381 return &MockBucket_IsAccessDeniedErr_Call{Call: _e.mock.On("IsAccessDeniedErr", err)} 382 } 383 384 func (_c *MockBucket_IsAccessDeniedErr_Call) Run(run func(err error)) *MockBucket_IsAccessDeniedErr_Call { 385 _c.Call.Run(func(args mock.Arguments) { 386 run(args[0].(error)) 387 }) 388 return _c 389 } 390 391 func (_c *MockBucket_IsAccessDeniedErr_Call) Return(_a0 bool) *MockBucket_IsAccessDeniedErr_Call { 392 _c.Call.Return(_a0) 393 return _c 394 } 395 396 func (_c *MockBucket_IsAccessDeniedErr_Call) RunAndReturn(run func(error) bool) *MockBucket_IsAccessDeniedErr_Call { 397 _c.Call.Return(run) 398 return _c 399 } 400 401 // IsObjNotFoundErr provides a mock function with given fields: err 402 func (_m *MockBucket) IsObjNotFoundErr(err error) bool { 403 ret := _m.Called(err) 404 405 if len(ret) == 0 { 406 panic("no return value specified for IsObjNotFoundErr") 407 } 408 409 var r0 bool 410 if rf, ok := ret.Get(0).(func(error) bool); ok { 411 r0 = rf(err) 412 } else { 413 r0 = ret.Get(0).(bool) 414 } 415 416 return r0 417 } 418 419 // MockBucket_IsObjNotFoundErr_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'IsObjNotFoundErr' 420 type MockBucket_IsObjNotFoundErr_Call struct { 421 *mock.Call 422 } 423 424 // IsObjNotFoundErr is a helper method to define mock.On call 425 // - err error 426 func (_e *MockBucket_Expecter) IsObjNotFoundErr(err interface{}) *MockBucket_IsObjNotFoundErr_Call { 427 return &MockBucket_IsObjNotFoundErr_Call{Call: _e.mock.On("IsObjNotFoundErr", err)} 428 } 429 430 func (_c *MockBucket_IsObjNotFoundErr_Call) Run(run func(err error)) *MockBucket_IsObjNotFoundErr_Call { 431 _c.Call.Run(func(args mock.Arguments) { 432 run(args[0].(error)) 433 }) 434 return _c 435 } 436 437 func (_c *MockBucket_IsObjNotFoundErr_Call) Return(_a0 bool) *MockBucket_IsObjNotFoundErr_Call { 438 _c.Call.Return(_a0) 439 return _c 440 } 441 442 func (_c *MockBucket_IsObjNotFoundErr_Call) RunAndReturn(run func(error) bool) *MockBucket_IsObjNotFoundErr_Call { 443 _c.Call.Return(run) 444 return _c 445 } 446 447 // Iter provides a mock function with given fields: ctx, dir, f, options 448 func (_m *MockBucket) Iter(ctx context.Context, dir string, f func(string) error, options ...objstore.IterOption) error { 449 _va := make([]interface{}, len(options)) 450 for _i := range options { 451 _va[_i] = options[_i] 452 } 453 var _ca []interface{} 454 _ca = append(_ca, ctx, dir, f) 455 _ca = append(_ca, _va...) 456 ret := _m.Called(_ca...) 457 458 if len(ret) == 0 { 459 panic("no return value specified for Iter") 460 } 461 462 var r0 error 463 if rf, ok := ret.Get(0).(func(context.Context, string, func(string) error, ...objstore.IterOption) error); ok { 464 r0 = rf(ctx, dir, f, options...) 465 } else { 466 r0 = ret.Error(0) 467 } 468 469 return r0 470 } 471 472 // MockBucket_Iter_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Iter' 473 type MockBucket_Iter_Call struct { 474 *mock.Call 475 } 476 477 // Iter is a helper method to define mock.On call 478 // - ctx context.Context 479 // - dir string 480 // - f func(string) error 481 // - options ...objstore.IterOption 482 func (_e *MockBucket_Expecter) Iter(ctx interface{}, dir interface{}, f interface{}, options ...interface{}) *MockBucket_Iter_Call { 483 return &MockBucket_Iter_Call{Call: _e.mock.On("Iter", 484 append([]interface{}{ctx, dir, f}, options...)...)} 485 } 486 487 func (_c *MockBucket_Iter_Call) Run(run func(ctx context.Context, dir string, f func(string) error, options ...objstore.IterOption)) *MockBucket_Iter_Call { 488 _c.Call.Run(func(args mock.Arguments) { 489 variadicArgs := make([]objstore.IterOption, len(args)-3) 490 for i, a := range args[3:] { 491 if a != nil { 492 variadicArgs[i] = a.(objstore.IterOption) 493 } 494 } 495 run(args[0].(context.Context), args[1].(string), args[2].(func(string) error), variadicArgs...) 496 }) 497 return _c 498 } 499 500 func (_c *MockBucket_Iter_Call) Return(_a0 error) *MockBucket_Iter_Call { 501 _c.Call.Return(_a0) 502 return _c 503 } 504 505 func (_c *MockBucket_Iter_Call) RunAndReturn(run func(context.Context, string, func(string) error, ...objstore.IterOption) error) *MockBucket_Iter_Call { 506 _c.Call.Return(run) 507 return _c 508 } 509 510 // IterWithAttributes provides a mock function with given fields: ctx, dir, f, options 511 func (_m *MockBucket) IterWithAttributes(ctx context.Context, dir string, f func(objstore.IterObjectAttributes) error, options ...objstore.IterOption) error { 512 _va := make([]interface{}, len(options)) 513 for _i := range options { 514 _va[_i] = options[_i] 515 } 516 var _ca []interface{} 517 _ca = append(_ca, ctx, dir, f) 518 _ca = append(_ca, _va...) 519 ret := _m.Called(_ca...) 520 521 if len(ret) == 0 { 522 panic("no return value specified for IterWithAttributes") 523 } 524 525 var r0 error 526 if rf, ok := ret.Get(0).(func(context.Context, string, func(objstore.IterObjectAttributes) error, ...objstore.IterOption) error); ok { 527 r0 = rf(ctx, dir, f, options...) 528 } else { 529 r0 = ret.Error(0) 530 } 531 532 return r0 533 } 534 535 // MockBucket_IterWithAttributes_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'IterWithAttributes' 536 type MockBucket_IterWithAttributes_Call struct { 537 *mock.Call 538 } 539 540 // IterWithAttributes is a helper method to define mock.On call 541 // - ctx context.Context 542 // - dir string 543 // - f func(objstore.IterObjectAttributes) error 544 // - options ...objstore.IterOption 545 func (_e *MockBucket_Expecter) IterWithAttributes(ctx interface{}, dir interface{}, f interface{}, options ...interface{}) *MockBucket_IterWithAttributes_Call { 546 return &MockBucket_IterWithAttributes_Call{Call: _e.mock.On("IterWithAttributes", 547 append([]interface{}{ctx, dir, f}, options...)...)} 548 } 549 550 func (_c *MockBucket_IterWithAttributes_Call) Run(run func(ctx context.Context, dir string, f func(objstore.IterObjectAttributes) error, options ...objstore.IterOption)) *MockBucket_IterWithAttributes_Call { 551 _c.Call.Run(func(args mock.Arguments) { 552 variadicArgs := make([]objstore.IterOption, len(args)-3) 553 for i, a := range args[3:] { 554 if a != nil { 555 variadicArgs[i] = a.(objstore.IterOption) 556 } 557 } 558 run(args[0].(context.Context), args[1].(string), args[2].(func(objstore.IterObjectAttributes) error), variadicArgs...) 559 }) 560 return _c 561 } 562 563 func (_c *MockBucket_IterWithAttributes_Call) Return(_a0 error) *MockBucket_IterWithAttributes_Call { 564 _c.Call.Return(_a0) 565 return _c 566 } 567 568 func (_c *MockBucket_IterWithAttributes_Call) RunAndReturn(run func(context.Context, string, func(objstore.IterObjectAttributes) error, ...objstore.IterOption) error) *MockBucket_IterWithAttributes_Call { 569 _c.Call.Return(run) 570 return _c 571 } 572 573 // Name provides a mock function with no fields 574 func (_m *MockBucket) Name() string { 575 ret := _m.Called() 576 577 if len(ret) == 0 { 578 panic("no return value specified for Name") 579 } 580 581 var r0 string 582 if rf, ok := ret.Get(0).(func() string); ok { 583 r0 = rf() 584 } else { 585 r0 = ret.Get(0).(string) 586 } 587 588 return r0 589 } 590 591 // MockBucket_Name_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Name' 592 type MockBucket_Name_Call struct { 593 *mock.Call 594 } 595 596 // Name is a helper method to define mock.On call 597 func (_e *MockBucket_Expecter) Name() *MockBucket_Name_Call { 598 return &MockBucket_Name_Call{Call: _e.mock.On("Name")} 599 } 600 601 func (_c *MockBucket_Name_Call) Run(run func()) *MockBucket_Name_Call { 602 _c.Call.Run(func(args mock.Arguments) { 603 run() 604 }) 605 return _c 606 } 607 608 func (_c *MockBucket_Name_Call) Return(_a0 string) *MockBucket_Name_Call { 609 _c.Call.Return(_a0) 610 return _c 611 } 612 613 func (_c *MockBucket_Name_Call) RunAndReturn(run func() string) *MockBucket_Name_Call { 614 _c.Call.Return(run) 615 return _c 616 } 617 618 // Provider provides a mock function with no fields 619 func (_m *MockBucket) Provider() objstore.ObjProvider { 620 ret := _m.Called() 621 622 if len(ret) == 0 { 623 panic("no return value specified for Provider") 624 } 625 626 var r0 objstore.ObjProvider 627 if rf, ok := ret.Get(0).(func() objstore.ObjProvider); ok { 628 r0 = rf() 629 } else { 630 r0 = ret.Get(0).(objstore.ObjProvider) 631 } 632 633 return r0 634 } 635 636 // MockBucket_Provider_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Provider' 637 type MockBucket_Provider_Call struct { 638 *mock.Call 639 } 640 641 // Provider is a helper method to define mock.On call 642 func (_e *MockBucket_Expecter) Provider() *MockBucket_Provider_Call { 643 return &MockBucket_Provider_Call{Call: _e.mock.On("Provider")} 644 } 645 646 func (_c *MockBucket_Provider_Call) Run(run func()) *MockBucket_Provider_Call { 647 _c.Call.Run(func(args mock.Arguments) { 648 run() 649 }) 650 return _c 651 } 652 653 func (_c *MockBucket_Provider_Call) Return(_a0 objstore.ObjProvider) *MockBucket_Provider_Call { 654 _c.Call.Return(_a0) 655 return _c 656 } 657 658 func (_c *MockBucket_Provider_Call) RunAndReturn(run func() objstore.ObjProvider) *MockBucket_Provider_Call { 659 _c.Call.Return(run) 660 return _c 661 } 662 663 // ReaderAt provides a mock function with given fields: ctx, filename 664 func (_m *MockBucket) ReaderAt(ctx context.Context, filename string) (pkgobjstore.ReaderAtCloser, error) { 665 ret := _m.Called(ctx, filename) 666 667 if len(ret) == 0 { 668 panic("no return value specified for ReaderAt") 669 } 670 671 var r0 pkgobjstore.ReaderAtCloser 672 var r1 error 673 if rf, ok := ret.Get(0).(func(context.Context, string) (pkgobjstore.ReaderAtCloser, error)); ok { 674 return rf(ctx, filename) 675 } 676 if rf, ok := ret.Get(0).(func(context.Context, string) pkgobjstore.ReaderAtCloser); ok { 677 r0 = rf(ctx, filename) 678 } else { 679 if ret.Get(0) != nil { 680 r0 = ret.Get(0).(pkgobjstore.ReaderAtCloser) 681 } 682 } 683 684 if rf, ok := ret.Get(1).(func(context.Context, string) error); ok { 685 r1 = rf(ctx, filename) 686 } else { 687 r1 = ret.Error(1) 688 } 689 690 return r0, r1 691 } 692 693 // MockBucket_ReaderAt_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ReaderAt' 694 type MockBucket_ReaderAt_Call struct { 695 *mock.Call 696 } 697 698 // ReaderAt is a helper method to define mock.On call 699 // - ctx context.Context 700 // - filename string 701 func (_e *MockBucket_Expecter) ReaderAt(ctx interface{}, filename interface{}) *MockBucket_ReaderAt_Call { 702 return &MockBucket_ReaderAt_Call{Call: _e.mock.On("ReaderAt", ctx, filename)} 703 } 704 705 func (_c *MockBucket_ReaderAt_Call) Run(run func(ctx context.Context, filename string)) *MockBucket_ReaderAt_Call { 706 _c.Call.Run(func(args mock.Arguments) { 707 run(args[0].(context.Context), args[1].(string)) 708 }) 709 return _c 710 } 711 712 func (_c *MockBucket_ReaderAt_Call) Return(_a0 pkgobjstore.ReaderAtCloser, _a1 error) *MockBucket_ReaderAt_Call { 713 _c.Call.Return(_a0, _a1) 714 return _c 715 } 716 717 func (_c *MockBucket_ReaderAt_Call) RunAndReturn(run func(context.Context, string) (pkgobjstore.ReaderAtCloser, error)) *MockBucket_ReaderAt_Call { 718 _c.Call.Return(run) 719 return _c 720 } 721 722 // SupportedIterOptions provides a mock function with no fields 723 func (_m *MockBucket) SupportedIterOptions() []objstore.IterOptionType { 724 ret := _m.Called() 725 726 if len(ret) == 0 { 727 panic("no return value specified for SupportedIterOptions") 728 } 729 730 var r0 []objstore.IterOptionType 731 if rf, ok := ret.Get(0).(func() []objstore.IterOptionType); ok { 732 r0 = rf() 733 } else { 734 if ret.Get(0) != nil { 735 r0 = ret.Get(0).([]objstore.IterOptionType) 736 } 737 } 738 739 return r0 740 } 741 742 // MockBucket_SupportedIterOptions_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SupportedIterOptions' 743 type MockBucket_SupportedIterOptions_Call struct { 744 *mock.Call 745 } 746 747 // SupportedIterOptions is a helper method to define mock.On call 748 func (_e *MockBucket_Expecter) SupportedIterOptions() *MockBucket_SupportedIterOptions_Call { 749 return &MockBucket_SupportedIterOptions_Call{Call: _e.mock.On("SupportedIterOptions")} 750 } 751 752 func (_c *MockBucket_SupportedIterOptions_Call) Run(run func()) *MockBucket_SupportedIterOptions_Call { 753 _c.Call.Run(func(args mock.Arguments) { 754 run() 755 }) 756 return _c 757 } 758 759 func (_c *MockBucket_SupportedIterOptions_Call) Return(_a0 []objstore.IterOptionType) *MockBucket_SupportedIterOptions_Call { 760 _c.Call.Return(_a0) 761 return _c 762 } 763 764 func (_c *MockBucket_SupportedIterOptions_Call) RunAndReturn(run func() []objstore.IterOptionType) *MockBucket_SupportedIterOptions_Call { 765 _c.Call.Return(run) 766 return _c 767 } 768 769 // Upload provides a mock function with given fields: ctx, name, r, opts 770 func (_m *MockBucket) Upload(ctx context.Context, name string, r io.Reader, opts ...objstore.ObjectUploadOption) error { 771 _va := make([]interface{}, len(opts)) 772 for _i := range opts { 773 _va[_i] = opts[_i] 774 } 775 var _ca []interface{} 776 _ca = append(_ca, ctx, name, r) 777 _ca = append(_ca, _va...) 778 ret := _m.Called(_ca...) 779 780 if len(ret) == 0 { 781 panic("no return value specified for Upload") 782 } 783 784 var r0 error 785 if rf, ok := ret.Get(0).(func(context.Context, string, io.Reader, ...objstore.ObjectUploadOption) error); ok { 786 r0 = rf(ctx, name, r, opts...) 787 } else { 788 r0 = ret.Error(0) 789 } 790 791 return r0 792 } 793 794 // MockBucket_Upload_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Upload' 795 type MockBucket_Upload_Call struct { 796 *mock.Call 797 } 798 799 // Upload is a helper method to define mock.On call 800 // - ctx context.Context 801 // - name string 802 // - r io.Reader 803 // - opts ...objstore.ObjectUploadOption 804 func (_e *MockBucket_Expecter) Upload(ctx interface{}, name interface{}, r interface{}, opts ...interface{}) *MockBucket_Upload_Call { 805 return &MockBucket_Upload_Call{Call: _e.mock.On("Upload", 806 append([]interface{}{ctx, name, r}, opts...)...)} 807 } 808 809 func (_c *MockBucket_Upload_Call) Run(run func(ctx context.Context, name string, r io.Reader, opts ...objstore.ObjectUploadOption)) *MockBucket_Upload_Call { 810 _c.Call.Run(func(args mock.Arguments) { 811 variadicArgs := make([]objstore.ObjectUploadOption, len(args)-3) 812 for i, a := range args[3:] { 813 if a != nil { 814 variadicArgs[i] = a.(objstore.ObjectUploadOption) 815 } 816 } 817 run(args[0].(context.Context), args[1].(string), args[2].(io.Reader), variadicArgs...) 818 }) 819 return _c 820 } 821 822 func (_c *MockBucket_Upload_Call) Return(_a0 error) *MockBucket_Upload_Call { 823 _c.Call.Return(_a0) 824 return _c 825 } 826 827 func (_c *MockBucket_Upload_Call) RunAndReturn(run func(context.Context, string, io.Reader, ...objstore.ObjectUploadOption) error) *MockBucket_Upload_Call { 828 _c.Call.Return(run) 829 return _c 830 } 831 832 // NewMockBucket creates a new instance of MockBucket. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. 833 // The first argument is typically a *testing.T value. 834 func NewMockBucket(t interface { 835 mock.TestingT 836 Cleanup(func()) 837 }) *MockBucket { 838 mock := &MockBucket{} 839 mock.Mock.Test(t) 840 841 t.Cleanup(func() { mock.AssertExpectations(t) }) 842 843 return mock 844 }