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