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

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