github.com/grafana/pyroscope@v1.18.0/pkg/test/mocks/mockdlq/mock_metastore.go (about)

     1  // Code generated by mockery. DO NOT EDIT.
     2  
     3  package mockdlq
     4  
     5  import (
     6  	context "context"
     7  
     8  	metastorev1 "github.com/grafana/pyroscope/api/gen/proto/go/metastore/v1"
     9  
    10  	mock "github.com/stretchr/testify/mock"
    11  )
    12  
    13  // MockMetastore is an autogenerated mock type for the Metastore type
    14  type MockMetastore struct {
    15  	mock.Mock
    16  }
    17  
    18  type MockMetastore_Expecter struct {
    19  	mock *mock.Mock
    20  }
    21  
    22  func (_m *MockMetastore) EXPECT() *MockMetastore_Expecter {
    23  	return &MockMetastore_Expecter{mock: &_m.Mock}
    24  }
    25  
    26  // AddRecoveredBlock provides a mock function with given fields: _a0, _a1
    27  func (_m *MockMetastore) AddRecoveredBlock(_a0 context.Context, _a1 *metastorev1.AddBlockRequest) (*metastorev1.AddBlockResponse, error) {
    28  	ret := _m.Called(_a0, _a1)
    29  
    30  	if len(ret) == 0 {
    31  		panic("no return value specified for AddRecoveredBlock")
    32  	}
    33  
    34  	var r0 *metastorev1.AddBlockResponse
    35  	var r1 error
    36  	if rf, ok := ret.Get(0).(func(context.Context, *metastorev1.AddBlockRequest) (*metastorev1.AddBlockResponse, error)); ok {
    37  		return rf(_a0, _a1)
    38  	}
    39  	if rf, ok := ret.Get(0).(func(context.Context, *metastorev1.AddBlockRequest) *metastorev1.AddBlockResponse); ok {
    40  		r0 = rf(_a0, _a1)
    41  	} else {
    42  		if ret.Get(0) != nil {
    43  			r0 = ret.Get(0).(*metastorev1.AddBlockResponse)
    44  		}
    45  	}
    46  
    47  	if rf, ok := ret.Get(1).(func(context.Context, *metastorev1.AddBlockRequest) error); ok {
    48  		r1 = rf(_a0, _a1)
    49  	} else {
    50  		r1 = ret.Error(1)
    51  	}
    52  
    53  	return r0, r1
    54  }
    55  
    56  // MockMetastore_AddRecoveredBlock_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'AddRecoveredBlock'
    57  type MockMetastore_AddRecoveredBlock_Call struct {
    58  	*mock.Call
    59  }
    60  
    61  // AddRecoveredBlock is a helper method to define mock.On call
    62  //   - _a0 context.Context
    63  //   - _a1 *metastorev1.AddBlockRequest
    64  func (_e *MockMetastore_Expecter) AddRecoveredBlock(_a0 interface{}, _a1 interface{}) *MockMetastore_AddRecoveredBlock_Call {
    65  	return &MockMetastore_AddRecoveredBlock_Call{Call: _e.mock.On("AddRecoveredBlock", _a0, _a1)}
    66  }
    67  
    68  func (_c *MockMetastore_AddRecoveredBlock_Call) Run(run func(_a0 context.Context, _a1 *metastorev1.AddBlockRequest)) *MockMetastore_AddRecoveredBlock_Call {
    69  	_c.Call.Run(func(args mock.Arguments) {
    70  		run(args[0].(context.Context), args[1].(*metastorev1.AddBlockRequest))
    71  	})
    72  	return _c
    73  }
    74  
    75  func (_c *MockMetastore_AddRecoveredBlock_Call) Return(_a0 *metastorev1.AddBlockResponse, _a1 error) *MockMetastore_AddRecoveredBlock_Call {
    76  	_c.Call.Return(_a0, _a1)
    77  	return _c
    78  }
    79  
    80  func (_c *MockMetastore_AddRecoveredBlock_Call) RunAndReturn(run func(context.Context, *metastorev1.AddBlockRequest) (*metastorev1.AddBlockResponse, error)) *MockMetastore_AddRecoveredBlock_Call {
    81  	_c.Call.Return(run)
    82  	return _c
    83  }
    84  
    85  // NewMockMetastore creates a new instance of MockMetastore. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
    86  // The first argument is typically a *testing.T value.
    87  func NewMockMetastore(t interface {
    88  	mock.TestingT
    89  	Cleanup(func())
    90  }) *MockMetastore {
    91  	mock := &MockMetastore{}
    92  	mock.Mock.Test(t)
    93  
    94  	t.Cleanup(func() { mock.AssertExpectations(t) })
    95  
    96  	return mock
    97  }