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

     1  // Code generated by mockery. DO NOT EDIT.
     2  
     3  package mockplacement
     4  
     5  import (
     6  	placement "github.com/grafana/pyroscope/pkg/segmentwriter/client/distributor/placement"
     7  	mock "github.com/stretchr/testify/mock"
     8  )
     9  
    10  // MockPlacement is an autogenerated mock type for the Placement type
    11  type MockPlacement struct {
    12  	mock.Mock
    13  }
    14  
    15  type MockPlacement_Expecter struct {
    16  	mock *mock.Mock
    17  }
    18  
    19  func (_m *MockPlacement) EXPECT() *MockPlacement_Expecter {
    20  	return &MockPlacement_Expecter{mock: &_m.Mock}
    21  }
    22  
    23  // Policy provides a mock function with given fields: _a0
    24  func (_m *MockPlacement) Policy(_a0 placement.Key) placement.Policy {
    25  	ret := _m.Called(_a0)
    26  
    27  	if len(ret) == 0 {
    28  		panic("no return value specified for Policy")
    29  	}
    30  
    31  	var r0 placement.Policy
    32  	if rf, ok := ret.Get(0).(func(placement.Key) placement.Policy); ok {
    33  		r0 = rf(_a0)
    34  	} else {
    35  		r0 = ret.Get(0).(placement.Policy)
    36  	}
    37  
    38  	return r0
    39  }
    40  
    41  // MockPlacement_Policy_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Policy'
    42  type MockPlacement_Policy_Call struct {
    43  	*mock.Call
    44  }
    45  
    46  // Policy is a helper method to define mock.On call
    47  //   - _a0 placement.Key
    48  func (_e *MockPlacement_Expecter) Policy(_a0 interface{}) *MockPlacement_Policy_Call {
    49  	return &MockPlacement_Policy_Call{Call: _e.mock.On("Policy", _a0)}
    50  }
    51  
    52  func (_c *MockPlacement_Policy_Call) Run(run func(_a0 placement.Key)) *MockPlacement_Policy_Call {
    53  	_c.Call.Run(func(args mock.Arguments) {
    54  		run(args[0].(placement.Key))
    55  	})
    56  	return _c
    57  }
    58  
    59  func (_c *MockPlacement_Policy_Call) Return(_a0 placement.Policy) *MockPlacement_Policy_Call {
    60  	_c.Call.Return(_a0)
    61  	return _c
    62  }
    63  
    64  func (_c *MockPlacement_Policy_Call) RunAndReturn(run func(placement.Key) placement.Policy) *MockPlacement_Policy_Call {
    65  	_c.Call.Return(run)
    66  	return _c
    67  }
    68  
    69  // NewMockPlacement creates a new instance of MockPlacement. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
    70  // The first argument is typically a *testing.T value.
    71  func NewMockPlacement(t interface {
    72  	mock.TestingT
    73  	Cleanup(func())
    74  }) *MockPlacement {
    75  	mock := &MockPlacement{}
    76  	mock.Mock.Test(t)
    77  
    78  	t.Cleanup(func() { mock.AssertExpectations(t) })
    79  
    80  	return mock
    81  }