github.com/uber/kraken@v0.1.4/mocks/lib/torrent/scheduler/reloadablescheduler.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: github.com/uber/kraken/lib/torrent/scheduler (interfaces: ReloadableScheduler)
     3  
     4  // Package mockscheduler is a generated GoMock package.
     5  package mockscheduler
     6  
     7  import (
     8  	gomock "github.com/golang/mock/gomock"
     9  	core "github.com/uber/kraken/core"
    10  	scheduler "github.com/uber/kraken/lib/torrent/scheduler"
    11  	connstate "github.com/uber/kraken/lib/torrent/scheduler/connstate"
    12  	reflect "reflect"
    13  )
    14  
    15  // MockReloadableScheduler is a mock of ReloadableScheduler interface
    16  type MockReloadableScheduler struct {
    17  	ctrl     *gomock.Controller
    18  	recorder *MockReloadableSchedulerMockRecorder
    19  }
    20  
    21  // MockReloadableSchedulerMockRecorder is the mock recorder for MockReloadableScheduler
    22  type MockReloadableSchedulerMockRecorder struct {
    23  	mock *MockReloadableScheduler
    24  }
    25  
    26  // NewMockReloadableScheduler creates a new mock instance
    27  func NewMockReloadableScheduler(ctrl *gomock.Controller) *MockReloadableScheduler {
    28  	mock := &MockReloadableScheduler{ctrl: ctrl}
    29  	mock.recorder = &MockReloadableSchedulerMockRecorder{mock}
    30  	return mock
    31  }
    32  
    33  // EXPECT returns an object that allows the caller to indicate expected use
    34  func (m *MockReloadableScheduler) EXPECT() *MockReloadableSchedulerMockRecorder {
    35  	return m.recorder
    36  }
    37  
    38  // BlacklistSnapshot mocks base method
    39  func (m *MockReloadableScheduler) BlacklistSnapshot() ([]connstate.BlacklistedConn, error) {
    40  	m.ctrl.T.Helper()
    41  	ret := m.ctrl.Call(m, "BlacklistSnapshot")
    42  	ret0, _ := ret[0].([]connstate.BlacklistedConn)
    43  	ret1, _ := ret[1].(error)
    44  	return ret0, ret1
    45  }
    46  
    47  // BlacklistSnapshot indicates an expected call of BlacklistSnapshot
    48  func (mr *MockReloadableSchedulerMockRecorder) BlacklistSnapshot() *gomock.Call {
    49  	mr.mock.ctrl.T.Helper()
    50  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "BlacklistSnapshot", reflect.TypeOf((*MockReloadableScheduler)(nil).BlacklistSnapshot))
    51  }
    52  
    53  // Download mocks base method
    54  func (m *MockReloadableScheduler) Download(arg0 string, arg1 core.Digest) error {
    55  	m.ctrl.T.Helper()
    56  	ret := m.ctrl.Call(m, "Download", arg0, arg1)
    57  	ret0, _ := ret[0].(error)
    58  	return ret0
    59  }
    60  
    61  // Download indicates an expected call of Download
    62  func (mr *MockReloadableSchedulerMockRecorder) Download(arg0, arg1 interface{}) *gomock.Call {
    63  	mr.mock.ctrl.T.Helper()
    64  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Download", reflect.TypeOf((*MockReloadableScheduler)(nil).Download), arg0, arg1)
    65  }
    66  
    67  // Probe mocks base method
    68  func (m *MockReloadableScheduler) Probe() error {
    69  	m.ctrl.T.Helper()
    70  	ret := m.ctrl.Call(m, "Probe")
    71  	ret0, _ := ret[0].(error)
    72  	return ret0
    73  }
    74  
    75  // Probe indicates an expected call of Probe
    76  func (mr *MockReloadableSchedulerMockRecorder) Probe() *gomock.Call {
    77  	mr.mock.ctrl.T.Helper()
    78  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Probe", reflect.TypeOf((*MockReloadableScheduler)(nil).Probe))
    79  }
    80  
    81  // Reload mocks base method
    82  func (m *MockReloadableScheduler) Reload(arg0 scheduler.Config) {
    83  	m.ctrl.T.Helper()
    84  	m.ctrl.Call(m, "Reload", arg0)
    85  }
    86  
    87  // Reload indicates an expected call of Reload
    88  func (mr *MockReloadableSchedulerMockRecorder) Reload(arg0 interface{}) *gomock.Call {
    89  	mr.mock.ctrl.T.Helper()
    90  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Reload", reflect.TypeOf((*MockReloadableScheduler)(nil).Reload), arg0)
    91  }
    92  
    93  // RemoveTorrent mocks base method
    94  func (m *MockReloadableScheduler) RemoveTorrent(arg0 core.Digest) error {
    95  	m.ctrl.T.Helper()
    96  	ret := m.ctrl.Call(m, "RemoveTorrent", arg0)
    97  	ret0, _ := ret[0].(error)
    98  	return ret0
    99  }
   100  
   101  // RemoveTorrent indicates an expected call of RemoveTorrent
   102  func (mr *MockReloadableSchedulerMockRecorder) RemoveTorrent(arg0 interface{}) *gomock.Call {
   103  	mr.mock.ctrl.T.Helper()
   104  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemoveTorrent", reflect.TypeOf((*MockReloadableScheduler)(nil).RemoveTorrent), arg0)
   105  }
   106  
   107  // Stop mocks base method
   108  func (m *MockReloadableScheduler) Stop() {
   109  	m.ctrl.T.Helper()
   110  	m.ctrl.Call(m, "Stop")
   111  }
   112  
   113  // Stop indicates an expected call of Stop
   114  func (mr *MockReloadableSchedulerMockRecorder) Stop() *gomock.Call {
   115  	mr.mock.ctrl.T.Helper()
   116  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stop", reflect.TypeOf((*MockReloadableScheduler)(nil).Stop))
   117  }