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

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