github.com/grafana/pyroscope@v1.18.0/pkg/test/mocks/mockmetastorev1/mock_index_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  // MockIndexServiceServer is an autogenerated mock type for the IndexServiceServer type
    13  type MockIndexServiceServer struct {
    14  	mock.Mock
    15  }
    16  
    17  type MockIndexServiceServer_Expecter struct {
    18  	mock *mock.Mock
    19  }
    20  
    21  func (_m *MockIndexServiceServer) EXPECT() *MockIndexServiceServer_Expecter {
    22  	return &MockIndexServiceServer_Expecter{mock: &_m.Mock}
    23  }
    24  
    25  // AddBlock provides a mock function with given fields: _a0, _a1
    26  func (_m *MockIndexServiceServer) AddBlock(_a0 context.Context, _a1 *metastorev1.AddBlockRequest) (*metastorev1.AddBlockResponse, error) {
    27  	ret := _m.Called(_a0, _a1)
    28  
    29  	if len(ret) == 0 {
    30  		panic("no return value specified for AddBlock")
    31  	}
    32  
    33  	var r0 *metastorev1.AddBlockResponse
    34  	var r1 error
    35  	if rf, ok := ret.Get(0).(func(context.Context, *metastorev1.AddBlockRequest) (*metastorev1.AddBlockResponse, error)); ok {
    36  		return rf(_a0, _a1)
    37  	}
    38  	if rf, ok := ret.Get(0).(func(context.Context, *metastorev1.AddBlockRequest) *metastorev1.AddBlockResponse); ok {
    39  		r0 = rf(_a0, _a1)
    40  	} else {
    41  		if ret.Get(0) != nil {
    42  			r0 = ret.Get(0).(*metastorev1.AddBlockResponse)
    43  		}
    44  	}
    45  
    46  	if rf, ok := ret.Get(1).(func(context.Context, *metastorev1.AddBlockRequest) error); ok {
    47  		r1 = rf(_a0, _a1)
    48  	} else {
    49  		r1 = ret.Error(1)
    50  	}
    51  
    52  	return r0, r1
    53  }
    54  
    55  // MockIndexServiceServer_AddBlock_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'AddBlock'
    56  type MockIndexServiceServer_AddBlock_Call struct {
    57  	*mock.Call
    58  }
    59  
    60  // AddBlock is a helper method to define mock.On call
    61  //   - _a0 context.Context
    62  //   - _a1 *metastorev1.AddBlockRequest
    63  func (_e *MockIndexServiceServer_Expecter) AddBlock(_a0 interface{}, _a1 interface{}) *MockIndexServiceServer_AddBlock_Call {
    64  	return &MockIndexServiceServer_AddBlock_Call{Call: _e.mock.On("AddBlock", _a0, _a1)}
    65  }
    66  
    67  func (_c *MockIndexServiceServer_AddBlock_Call) Run(run func(_a0 context.Context, _a1 *metastorev1.AddBlockRequest)) *MockIndexServiceServer_AddBlock_Call {
    68  	_c.Call.Run(func(args mock.Arguments) {
    69  		run(args[0].(context.Context), args[1].(*metastorev1.AddBlockRequest))
    70  	})
    71  	return _c
    72  }
    73  
    74  func (_c *MockIndexServiceServer_AddBlock_Call) Return(_a0 *metastorev1.AddBlockResponse, _a1 error) *MockIndexServiceServer_AddBlock_Call {
    75  	_c.Call.Return(_a0, _a1)
    76  	return _c
    77  }
    78  
    79  func (_c *MockIndexServiceServer_AddBlock_Call) RunAndReturn(run func(context.Context, *metastorev1.AddBlockRequest) (*metastorev1.AddBlockResponse, error)) *MockIndexServiceServer_AddBlock_Call {
    80  	_c.Call.Return(run)
    81  	return _c
    82  }
    83  
    84  // GetBlockMetadata provides a mock function with given fields: _a0, _a1
    85  func (_m *MockIndexServiceServer) GetBlockMetadata(_a0 context.Context, _a1 *metastorev1.GetBlockMetadataRequest) (*metastorev1.GetBlockMetadataResponse, error) {
    86  	ret := _m.Called(_a0, _a1)
    87  
    88  	if len(ret) == 0 {
    89  		panic("no return value specified for GetBlockMetadata")
    90  	}
    91  
    92  	var r0 *metastorev1.GetBlockMetadataResponse
    93  	var r1 error
    94  	if rf, ok := ret.Get(0).(func(context.Context, *metastorev1.GetBlockMetadataRequest) (*metastorev1.GetBlockMetadataResponse, error)); ok {
    95  		return rf(_a0, _a1)
    96  	}
    97  	if rf, ok := ret.Get(0).(func(context.Context, *metastorev1.GetBlockMetadataRequest) *metastorev1.GetBlockMetadataResponse); ok {
    98  		r0 = rf(_a0, _a1)
    99  	} else {
   100  		if ret.Get(0) != nil {
   101  			r0 = ret.Get(0).(*metastorev1.GetBlockMetadataResponse)
   102  		}
   103  	}
   104  
   105  	if rf, ok := ret.Get(1).(func(context.Context, *metastorev1.GetBlockMetadataRequest) error); ok {
   106  		r1 = rf(_a0, _a1)
   107  	} else {
   108  		r1 = ret.Error(1)
   109  	}
   110  
   111  	return r0, r1
   112  }
   113  
   114  // MockIndexServiceServer_GetBlockMetadata_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetBlockMetadata'
   115  type MockIndexServiceServer_GetBlockMetadata_Call struct {
   116  	*mock.Call
   117  }
   118  
   119  // GetBlockMetadata is a helper method to define mock.On call
   120  //   - _a0 context.Context
   121  //   - _a1 *metastorev1.GetBlockMetadataRequest
   122  func (_e *MockIndexServiceServer_Expecter) GetBlockMetadata(_a0 interface{}, _a1 interface{}) *MockIndexServiceServer_GetBlockMetadata_Call {
   123  	return &MockIndexServiceServer_GetBlockMetadata_Call{Call: _e.mock.On("GetBlockMetadata", _a0, _a1)}
   124  }
   125  
   126  func (_c *MockIndexServiceServer_GetBlockMetadata_Call) Run(run func(_a0 context.Context, _a1 *metastorev1.GetBlockMetadataRequest)) *MockIndexServiceServer_GetBlockMetadata_Call {
   127  	_c.Call.Run(func(args mock.Arguments) {
   128  		run(args[0].(context.Context), args[1].(*metastorev1.GetBlockMetadataRequest))
   129  	})
   130  	return _c
   131  }
   132  
   133  func (_c *MockIndexServiceServer_GetBlockMetadata_Call) Return(_a0 *metastorev1.GetBlockMetadataResponse, _a1 error) *MockIndexServiceServer_GetBlockMetadata_Call {
   134  	_c.Call.Return(_a0, _a1)
   135  	return _c
   136  }
   137  
   138  func (_c *MockIndexServiceServer_GetBlockMetadata_Call) RunAndReturn(run func(context.Context, *metastorev1.GetBlockMetadataRequest) (*metastorev1.GetBlockMetadataResponse, error)) *MockIndexServiceServer_GetBlockMetadata_Call {
   139  	_c.Call.Return(run)
   140  	return _c
   141  }
   142  
   143  // mustEmbedUnimplementedIndexServiceServer provides a mock function with no fields
   144  func (_m *MockIndexServiceServer) mustEmbedUnimplementedIndexServiceServer() {
   145  	_m.Called()
   146  }
   147  
   148  // MockIndexServiceServer_mustEmbedUnimplementedIndexServiceServer_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'mustEmbedUnimplementedIndexServiceServer'
   149  type MockIndexServiceServer_mustEmbedUnimplementedIndexServiceServer_Call struct {
   150  	*mock.Call
   151  }
   152  
   153  // mustEmbedUnimplementedIndexServiceServer is a helper method to define mock.On call
   154  func (_e *MockIndexServiceServer_Expecter) mustEmbedUnimplementedIndexServiceServer() *MockIndexServiceServer_mustEmbedUnimplementedIndexServiceServer_Call {
   155  	return &MockIndexServiceServer_mustEmbedUnimplementedIndexServiceServer_Call{Call: _e.mock.On("mustEmbedUnimplementedIndexServiceServer")}
   156  }
   157  
   158  func (_c *MockIndexServiceServer_mustEmbedUnimplementedIndexServiceServer_Call) Run(run func()) *MockIndexServiceServer_mustEmbedUnimplementedIndexServiceServer_Call {
   159  	_c.Call.Run(func(args mock.Arguments) {
   160  		run()
   161  	})
   162  	return _c
   163  }
   164  
   165  func (_c *MockIndexServiceServer_mustEmbedUnimplementedIndexServiceServer_Call) Return() *MockIndexServiceServer_mustEmbedUnimplementedIndexServiceServer_Call {
   166  	_c.Call.Return()
   167  	return _c
   168  }
   169  
   170  func (_c *MockIndexServiceServer_mustEmbedUnimplementedIndexServiceServer_Call) RunAndReturn(run func()) *MockIndexServiceServer_mustEmbedUnimplementedIndexServiceServer_Call {
   171  	_c.Run(run)
   172  	return _c
   173  }
   174  
   175  // NewMockIndexServiceServer creates a new instance of MockIndexServiceServer. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
   176  // The first argument is typically a *testing.T value.
   177  func NewMockIndexServiceServer(t interface {
   178  	mock.TestingT
   179  	Cleanup(func())
   180  }) *MockIndexServiceServer {
   181  	mock := &MockIndexServiceServer{}
   182  	mock.Mock.Test(t)
   183  
   184  	t.Cleanup(func() { mock.AssertExpectations(t) })
   185  
   186  	return mock
   187  }