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

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