mock_bart_retriever_test.go 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990
  1. // Code generated by mockery v2.52.1. DO NOT EDIT.
  2. package http
  3. import mock "github.com/stretchr/testify/mock"
  4. // mockBARTRetriever is an autogenerated mock type for the BARTRetriever type
  5. type mockBARTRetriever struct {
  6. mock.Mock
  7. }
  8. type mockBARTRetriever_Expecter struct {
  9. mock *mock.Mock
  10. }
  11. func (_m *mockBARTRetriever) EXPECT() *mockBARTRetriever_Expecter {
  12. return &mockBARTRetriever_Expecter{mock: &_m.Mock}
  13. }
  14. // BARTRetrieve provides a mock function with given fields: itemHash
  15. func (_m *mockBARTRetriever) BARTRetrieve(itemHash []byte) ([]byte, error) {
  16. ret := _m.Called(itemHash)
  17. if len(ret) == 0 {
  18. panic("no return value specified for BARTRetrieve")
  19. }
  20. var r0 []byte
  21. var r1 error
  22. if rf, ok := ret.Get(0).(func([]byte) ([]byte, error)); ok {
  23. return rf(itemHash)
  24. }
  25. if rf, ok := ret.Get(0).(func([]byte) []byte); ok {
  26. r0 = rf(itemHash)
  27. } else {
  28. if ret.Get(0) != nil {
  29. r0 = ret.Get(0).([]byte)
  30. }
  31. }
  32. if rf, ok := ret.Get(1).(func([]byte) error); ok {
  33. r1 = rf(itemHash)
  34. } else {
  35. r1 = ret.Error(1)
  36. }
  37. return r0, r1
  38. }
  39. // mockBARTRetriever_BARTRetrieve_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'BARTRetrieve'
  40. type mockBARTRetriever_BARTRetrieve_Call struct {
  41. *mock.Call
  42. }
  43. // BARTRetrieve is a helper method to define mock.On call
  44. // - itemHash []byte
  45. func (_e *mockBARTRetriever_Expecter) BARTRetrieve(itemHash interface{}) *mockBARTRetriever_BARTRetrieve_Call {
  46. return &mockBARTRetriever_BARTRetrieve_Call{Call: _e.mock.On("BARTRetrieve", itemHash)}
  47. }
  48. func (_c *mockBARTRetriever_BARTRetrieve_Call) Run(run func(itemHash []byte)) *mockBARTRetriever_BARTRetrieve_Call {
  49. _c.Call.Run(func(args mock.Arguments) {
  50. run(args[0].([]byte))
  51. })
  52. return _c
  53. }
  54. func (_c *mockBARTRetriever_BARTRetrieve_Call) Return(_a0 []byte, _a1 error) *mockBARTRetriever_BARTRetrieve_Call {
  55. _c.Call.Return(_a0, _a1)
  56. return _c
  57. }
  58. func (_c *mockBARTRetriever_BARTRetrieve_Call) RunAndReturn(run func([]byte) ([]byte, error)) *mockBARTRetriever_BARTRetrieve_Call {
  59. _c.Call.Return(run)
  60. return _c
  61. }
  62. // newMockBARTRetriever creates a new instance of mockBARTRetriever. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
  63. // The first argument is typically a *testing.T value.
  64. func newMockBARTRetriever(t interface {
  65. mock.TestingT
  66. Cleanup(func())
  67. }) *mockBARTRetriever {
  68. mock := &mockBARTRetriever{}
  69. mock.Mock.Test(t)
  70. t.Cleanup(func() { mock.AssertExpectations(t) })
  71. return mock
  72. }