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

     1  // Code generated by mockery. DO NOT EDIT.
     2  
     3  package mockdiscovery
     4  
     5  import (
     6  	discovery "github.com/grafana/pyroscope/pkg/metastore/discovery"
     7  	mock "github.com/stretchr/testify/mock"
     8  )
     9  
    10  // MockDiscovery is an autogenerated mock type for the Discovery type
    11  type MockDiscovery struct {
    12  	mock.Mock
    13  }
    14  
    15  type MockDiscovery_Expecter struct {
    16  	mock *mock.Mock
    17  }
    18  
    19  func (_m *MockDiscovery) EXPECT() *MockDiscovery_Expecter {
    20  	return &MockDiscovery_Expecter{mock: &_m.Mock}
    21  }
    22  
    23  // Close provides a mock function with no fields
    24  func (_m *MockDiscovery) Close() {
    25  	_m.Called()
    26  }
    27  
    28  // MockDiscovery_Close_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Close'
    29  type MockDiscovery_Close_Call struct {
    30  	*mock.Call
    31  }
    32  
    33  // Close is a helper method to define mock.On call
    34  func (_e *MockDiscovery_Expecter) Close() *MockDiscovery_Close_Call {
    35  	return &MockDiscovery_Close_Call{Call: _e.mock.On("Close")}
    36  }
    37  
    38  func (_c *MockDiscovery_Close_Call) Run(run func()) *MockDiscovery_Close_Call {
    39  	_c.Call.Run(func(args mock.Arguments) {
    40  		run()
    41  	})
    42  	return _c
    43  }
    44  
    45  func (_c *MockDiscovery_Close_Call) Return() *MockDiscovery_Close_Call {
    46  	_c.Call.Return()
    47  	return _c
    48  }
    49  
    50  func (_c *MockDiscovery_Close_Call) RunAndReturn(run func()) *MockDiscovery_Close_Call {
    51  	_c.Run(run)
    52  	return _c
    53  }
    54  
    55  // Rediscover provides a mock function with no fields
    56  func (_m *MockDiscovery) Rediscover() {
    57  	_m.Called()
    58  }
    59  
    60  // MockDiscovery_Rediscover_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Rediscover'
    61  type MockDiscovery_Rediscover_Call struct {
    62  	*mock.Call
    63  }
    64  
    65  // Rediscover is a helper method to define mock.On call
    66  func (_e *MockDiscovery_Expecter) Rediscover() *MockDiscovery_Rediscover_Call {
    67  	return &MockDiscovery_Rediscover_Call{Call: _e.mock.On("Rediscover")}
    68  }
    69  
    70  func (_c *MockDiscovery_Rediscover_Call) Run(run func()) *MockDiscovery_Rediscover_Call {
    71  	_c.Call.Run(func(args mock.Arguments) {
    72  		run()
    73  	})
    74  	return _c
    75  }
    76  
    77  func (_c *MockDiscovery_Rediscover_Call) Return() *MockDiscovery_Rediscover_Call {
    78  	_c.Call.Return()
    79  	return _c
    80  }
    81  
    82  func (_c *MockDiscovery_Rediscover_Call) RunAndReturn(run func()) *MockDiscovery_Rediscover_Call {
    83  	_c.Run(run)
    84  	return _c
    85  }
    86  
    87  // Subscribe provides a mock function with given fields: updates
    88  func (_m *MockDiscovery) Subscribe(updates discovery.Updates) {
    89  	_m.Called(updates)
    90  }
    91  
    92  // MockDiscovery_Subscribe_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Subscribe'
    93  type MockDiscovery_Subscribe_Call struct {
    94  	*mock.Call
    95  }
    96  
    97  // Subscribe is a helper method to define mock.On call
    98  //   - updates discovery.Updates
    99  func (_e *MockDiscovery_Expecter) Subscribe(updates interface{}) *MockDiscovery_Subscribe_Call {
   100  	return &MockDiscovery_Subscribe_Call{Call: _e.mock.On("Subscribe", updates)}
   101  }
   102  
   103  func (_c *MockDiscovery_Subscribe_Call) Run(run func(updates discovery.Updates)) *MockDiscovery_Subscribe_Call {
   104  	_c.Call.Run(func(args mock.Arguments) {
   105  		run(args[0].(discovery.Updates))
   106  	})
   107  	return _c
   108  }
   109  
   110  func (_c *MockDiscovery_Subscribe_Call) Return() *MockDiscovery_Subscribe_Call {
   111  	_c.Call.Return()
   112  	return _c
   113  }
   114  
   115  func (_c *MockDiscovery_Subscribe_Call) RunAndReturn(run func(discovery.Updates)) *MockDiscovery_Subscribe_Call {
   116  	_c.Run(run)
   117  	return _c
   118  }
   119  
   120  // NewMockDiscovery creates a new instance of MockDiscovery. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
   121  // The first argument is typically a *testing.T value.
   122  func NewMockDiscovery(t interface {
   123  	mock.TestingT
   124  	Cleanup(func())
   125  }) *MockDiscovery {
   126  	mock := &MockDiscovery{}
   127  	mock.Mock.Test(t)
   128  
   129  	t.Cleanup(func() { mock.AssertExpectations(t) })
   130  
   131  	return mock
   132  }