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

     1  // Code generated by mockery. DO NOT EDIT.
     2  
     3  package mockadaptiveplacement
     4  
     5  import (
     6  	adaptive_placementpb "github.com/grafana/pyroscope/pkg/segmentwriter/client/distributor/placement/adaptiveplacement/adaptive_placementpb"
     7  
     8  	context "context"
     9  
    10  	mock "github.com/stretchr/testify/mock"
    11  )
    12  
    13  // MockStore is an autogenerated mock type for the Store type
    14  type MockStore struct {
    15  	mock.Mock
    16  }
    17  
    18  type MockStore_Expecter struct {
    19  	mock *mock.Mock
    20  }
    21  
    22  func (_m *MockStore) EXPECT() *MockStore_Expecter {
    23  	return &MockStore_Expecter{mock: &_m.Mock}
    24  }
    25  
    26  // LoadRules provides a mock function with given fields: _a0
    27  func (_m *MockStore) LoadRules(_a0 context.Context) (*adaptive_placementpb.PlacementRules, error) {
    28  	ret := _m.Called(_a0)
    29  
    30  	if len(ret) == 0 {
    31  		panic("no return value specified for LoadRules")
    32  	}
    33  
    34  	var r0 *adaptive_placementpb.PlacementRules
    35  	var r1 error
    36  	if rf, ok := ret.Get(0).(func(context.Context) (*adaptive_placementpb.PlacementRules, error)); ok {
    37  		return rf(_a0)
    38  	}
    39  	if rf, ok := ret.Get(0).(func(context.Context) *adaptive_placementpb.PlacementRules); ok {
    40  		r0 = rf(_a0)
    41  	} else {
    42  		if ret.Get(0) != nil {
    43  			r0 = ret.Get(0).(*adaptive_placementpb.PlacementRules)
    44  		}
    45  	}
    46  
    47  	if rf, ok := ret.Get(1).(func(context.Context) error); ok {
    48  		r1 = rf(_a0)
    49  	} else {
    50  		r1 = ret.Error(1)
    51  	}
    52  
    53  	return r0, r1
    54  }
    55  
    56  // MockStore_LoadRules_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'LoadRules'
    57  type MockStore_LoadRules_Call struct {
    58  	*mock.Call
    59  }
    60  
    61  // LoadRules is a helper method to define mock.On call
    62  //   - _a0 context.Context
    63  func (_e *MockStore_Expecter) LoadRules(_a0 interface{}) *MockStore_LoadRules_Call {
    64  	return &MockStore_LoadRules_Call{Call: _e.mock.On("LoadRules", _a0)}
    65  }
    66  
    67  func (_c *MockStore_LoadRules_Call) Run(run func(_a0 context.Context)) *MockStore_LoadRules_Call {
    68  	_c.Call.Run(func(args mock.Arguments) {
    69  		run(args[0].(context.Context))
    70  	})
    71  	return _c
    72  }
    73  
    74  func (_c *MockStore_LoadRules_Call) Return(_a0 *adaptive_placementpb.PlacementRules, _a1 error) *MockStore_LoadRules_Call {
    75  	_c.Call.Return(_a0, _a1)
    76  	return _c
    77  }
    78  
    79  func (_c *MockStore_LoadRules_Call) RunAndReturn(run func(context.Context) (*adaptive_placementpb.PlacementRules, error)) *MockStore_LoadRules_Call {
    80  	_c.Call.Return(run)
    81  	return _c
    82  }
    83  
    84  // LoadStats provides a mock function with given fields: _a0
    85  func (_m *MockStore) LoadStats(_a0 context.Context) (*adaptive_placementpb.DistributionStats, error) {
    86  	ret := _m.Called(_a0)
    87  
    88  	if len(ret) == 0 {
    89  		panic("no return value specified for LoadStats")
    90  	}
    91  
    92  	var r0 *adaptive_placementpb.DistributionStats
    93  	var r1 error
    94  	if rf, ok := ret.Get(0).(func(context.Context) (*adaptive_placementpb.DistributionStats, error)); ok {
    95  		return rf(_a0)
    96  	}
    97  	if rf, ok := ret.Get(0).(func(context.Context) *adaptive_placementpb.DistributionStats); ok {
    98  		r0 = rf(_a0)
    99  	} else {
   100  		if ret.Get(0) != nil {
   101  			r0 = ret.Get(0).(*adaptive_placementpb.DistributionStats)
   102  		}
   103  	}
   104  
   105  	if rf, ok := ret.Get(1).(func(context.Context) error); ok {
   106  		r1 = rf(_a0)
   107  	} else {
   108  		r1 = ret.Error(1)
   109  	}
   110  
   111  	return r0, r1
   112  }
   113  
   114  // MockStore_LoadStats_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'LoadStats'
   115  type MockStore_LoadStats_Call struct {
   116  	*mock.Call
   117  }
   118  
   119  // LoadStats is a helper method to define mock.On call
   120  //   - _a0 context.Context
   121  func (_e *MockStore_Expecter) LoadStats(_a0 interface{}) *MockStore_LoadStats_Call {
   122  	return &MockStore_LoadStats_Call{Call: _e.mock.On("LoadStats", _a0)}
   123  }
   124  
   125  func (_c *MockStore_LoadStats_Call) Run(run func(_a0 context.Context)) *MockStore_LoadStats_Call {
   126  	_c.Call.Run(func(args mock.Arguments) {
   127  		run(args[0].(context.Context))
   128  	})
   129  	return _c
   130  }
   131  
   132  func (_c *MockStore_LoadStats_Call) Return(_a0 *adaptive_placementpb.DistributionStats, _a1 error) *MockStore_LoadStats_Call {
   133  	_c.Call.Return(_a0, _a1)
   134  	return _c
   135  }
   136  
   137  func (_c *MockStore_LoadStats_Call) RunAndReturn(run func(context.Context) (*adaptive_placementpb.DistributionStats, error)) *MockStore_LoadStats_Call {
   138  	_c.Call.Return(run)
   139  	return _c
   140  }
   141  
   142  // StoreRules provides a mock function with given fields: _a0, _a1
   143  func (_m *MockStore) StoreRules(_a0 context.Context, _a1 *adaptive_placementpb.PlacementRules) error {
   144  	ret := _m.Called(_a0, _a1)
   145  
   146  	if len(ret) == 0 {
   147  		panic("no return value specified for StoreRules")
   148  	}
   149  
   150  	var r0 error
   151  	if rf, ok := ret.Get(0).(func(context.Context, *adaptive_placementpb.PlacementRules) error); ok {
   152  		r0 = rf(_a0, _a1)
   153  	} else {
   154  		r0 = ret.Error(0)
   155  	}
   156  
   157  	return r0
   158  }
   159  
   160  // MockStore_StoreRules_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'StoreRules'
   161  type MockStore_StoreRules_Call struct {
   162  	*mock.Call
   163  }
   164  
   165  // StoreRules is a helper method to define mock.On call
   166  //   - _a0 context.Context
   167  //   - _a1 *adaptive_placementpb.PlacementRules
   168  func (_e *MockStore_Expecter) StoreRules(_a0 interface{}, _a1 interface{}) *MockStore_StoreRules_Call {
   169  	return &MockStore_StoreRules_Call{Call: _e.mock.On("StoreRules", _a0, _a1)}
   170  }
   171  
   172  func (_c *MockStore_StoreRules_Call) Run(run func(_a0 context.Context, _a1 *adaptive_placementpb.PlacementRules)) *MockStore_StoreRules_Call {
   173  	_c.Call.Run(func(args mock.Arguments) {
   174  		run(args[0].(context.Context), args[1].(*adaptive_placementpb.PlacementRules))
   175  	})
   176  	return _c
   177  }
   178  
   179  func (_c *MockStore_StoreRules_Call) Return(_a0 error) *MockStore_StoreRules_Call {
   180  	_c.Call.Return(_a0)
   181  	return _c
   182  }
   183  
   184  func (_c *MockStore_StoreRules_Call) RunAndReturn(run func(context.Context, *adaptive_placementpb.PlacementRules) error) *MockStore_StoreRules_Call {
   185  	_c.Call.Return(run)
   186  	return _c
   187  }
   188  
   189  // StoreStats provides a mock function with given fields: _a0, _a1
   190  func (_m *MockStore) StoreStats(_a0 context.Context, _a1 *adaptive_placementpb.DistributionStats) error {
   191  	ret := _m.Called(_a0, _a1)
   192  
   193  	if len(ret) == 0 {
   194  		panic("no return value specified for StoreStats")
   195  	}
   196  
   197  	var r0 error
   198  	if rf, ok := ret.Get(0).(func(context.Context, *adaptive_placementpb.DistributionStats) error); ok {
   199  		r0 = rf(_a0, _a1)
   200  	} else {
   201  		r0 = ret.Error(0)
   202  	}
   203  
   204  	return r0
   205  }
   206  
   207  // MockStore_StoreStats_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'StoreStats'
   208  type MockStore_StoreStats_Call struct {
   209  	*mock.Call
   210  }
   211  
   212  // StoreStats is a helper method to define mock.On call
   213  //   - _a0 context.Context
   214  //   - _a1 *adaptive_placementpb.DistributionStats
   215  func (_e *MockStore_Expecter) StoreStats(_a0 interface{}, _a1 interface{}) *MockStore_StoreStats_Call {
   216  	return &MockStore_StoreStats_Call{Call: _e.mock.On("StoreStats", _a0, _a1)}
   217  }
   218  
   219  func (_c *MockStore_StoreStats_Call) Run(run func(_a0 context.Context, _a1 *adaptive_placementpb.DistributionStats)) *MockStore_StoreStats_Call {
   220  	_c.Call.Run(func(args mock.Arguments) {
   221  		run(args[0].(context.Context), args[1].(*adaptive_placementpb.DistributionStats))
   222  	})
   223  	return _c
   224  }
   225  
   226  func (_c *MockStore_StoreStats_Call) Return(_a0 error) *MockStore_StoreStats_Call {
   227  	_c.Call.Return(_a0)
   228  	return _c
   229  }
   230  
   231  func (_c *MockStore_StoreStats_Call) RunAndReturn(run func(context.Context, *adaptive_placementpb.DistributionStats) error) *MockStore_StoreStats_Call {
   232  	_c.Call.Return(run)
   233  	return _c
   234  }
   235  
   236  // NewMockStore creates a new instance of MockStore. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
   237  // The first argument is typically a *testing.T value.
   238  func NewMockStore(t interface {
   239  	mock.TestingT
   240  	Cleanup(func())
   241  }) *MockStore {
   242  	mock := &MockStore{}
   243  	mock.Mock.Test(t)
   244  
   245  	t.Cleanup(func() { mock.AssertExpectations(t) })
   246  
   247  	return mock
   248  }