Dragonfly2

1
// Code generated by MockGen. DO NOT EDIT.
2
// Source: host_manager.go
3
//
4
// Generated by this command:
5
//
6
//	mockgen -destination host_manager_mock.go -source host_manager.go -package resource
7
//
8
// Package resource is a generated GoMock package.
9
package resource
10

11
import (
12
	reflect "reflect"
13

14
	set "d7y.io/dragonfly/v2/pkg/container/set"
15
	gomock "go.uber.org/mock/gomock"
16
)
17

18
// MockHostManager is a mock of HostManager interface.
19
type MockHostManager struct {
20
	ctrl     *gomock.Controller
21
	recorder *MockHostManagerMockRecorder
22
}
23

24
// MockHostManagerMockRecorder is the mock recorder for MockHostManager.
25
type MockHostManagerMockRecorder struct {
26
	mock *MockHostManager
27
}
28

29
// NewMockHostManager creates a new mock instance.
30
func NewMockHostManager(ctrl *gomock.Controller) *MockHostManager {
31
	mock := &MockHostManager{ctrl: ctrl}
32
	mock.recorder = &MockHostManagerMockRecorder{mock}
33
	return mock
34
}
35

36
// EXPECT returns an object that allows the caller to indicate expected use.
37
func (m *MockHostManager) EXPECT() *MockHostManagerMockRecorder {
38
	return m.recorder
39
}
40

41
// Delete mocks base method.
42
func (m *MockHostManager) Delete(arg0 string) {
43
	m.ctrl.T.Helper()
44
	m.ctrl.Call(m, "Delete", arg0)
45
}
46

47
// Delete indicates an expected call of Delete.
48
func (mr *MockHostManagerMockRecorder) Delete(arg0 any) *gomock.Call {
49
	mr.mock.ctrl.T.Helper()
50
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Delete", reflect.TypeOf((*MockHostManager)(nil).Delete), arg0)
51
}
52

53
// Load mocks base method.
54
func (m *MockHostManager) Load(arg0 string) (*Host, bool) {
55
	m.ctrl.T.Helper()
56
	ret := m.ctrl.Call(m, "Load", arg0)
57
	ret0, _ := ret[0].(*Host)
58
	ret1, _ := ret[1].(bool)
59
	return ret0, ret1
60
}
61

62
// Load indicates an expected call of Load.
63
func (mr *MockHostManagerMockRecorder) Load(arg0 any) *gomock.Call {
64
	mr.mock.ctrl.T.Helper()
65
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Load", reflect.TypeOf((*MockHostManager)(nil).Load), arg0)
66
}
67

68
// LoadOrStore mocks base method.
69
func (m *MockHostManager) LoadOrStore(arg0 *Host) (*Host, bool) {
70
	m.ctrl.T.Helper()
71
	ret := m.ctrl.Call(m, "LoadOrStore", arg0)
72
	ret0, _ := ret[0].(*Host)
73
	ret1, _ := ret[1].(bool)
74
	return ret0, ret1
75
}
76

77
// LoadOrStore indicates an expected call of LoadOrStore.
78
func (mr *MockHostManagerMockRecorder) LoadOrStore(arg0 any) *gomock.Call {
79
	mr.mock.ctrl.T.Helper()
80
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LoadOrStore", reflect.TypeOf((*MockHostManager)(nil).LoadOrStore), arg0)
81
}
82

83
// LoadRandomHosts mocks base method.
84
func (m *MockHostManager) LoadRandomHosts(arg0 int, arg1 set.SafeSet[string]) []*Host {
85
	m.ctrl.T.Helper()
86
	ret := m.ctrl.Call(m, "LoadRandomHosts", arg0, arg1)
87
	ret0, _ := ret[0].([]*Host)
88
	return ret0
89
}
90

91
// LoadRandomHosts indicates an expected call of LoadRandomHosts.
92
func (mr *MockHostManagerMockRecorder) LoadRandomHosts(arg0, arg1 any) *gomock.Call {
93
	mr.mock.ctrl.T.Helper()
94
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "LoadRandomHosts", reflect.TypeOf((*MockHostManager)(nil).LoadRandomHosts), arg0, arg1)
95
}
96

97
// Range mocks base method.
98
func (m *MockHostManager) Range(f func(any, any) bool) {
99
	m.ctrl.T.Helper()
100
	m.ctrl.Call(m, "Range", f)
101
}
102

103
// Range indicates an expected call of Range.
104
func (mr *MockHostManagerMockRecorder) Range(f any) *gomock.Call {
105
	mr.mock.ctrl.T.Helper()
106
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Range", reflect.TypeOf((*MockHostManager)(nil).Range), f)
107
}
108

109
// RunGC mocks base method.
110
func (m *MockHostManager) RunGC() error {
111
	m.ctrl.T.Helper()
112
	ret := m.ctrl.Call(m, "RunGC")
113
	ret0, _ := ret[0].(error)
114
	return ret0
115
}
116

117
// RunGC indicates an expected call of RunGC.
118
func (mr *MockHostManagerMockRecorder) RunGC() *gomock.Call {
119
	mr.mock.ctrl.T.Helper()
120
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RunGC", reflect.TypeOf((*MockHostManager)(nil).RunGC))
121
}
122

123
// Store mocks base method.
124
func (m *MockHostManager) Store(arg0 *Host) {
125
	m.ctrl.T.Helper()
126
	m.ctrl.Call(m, "Store", arg0)
127
}
128

129
// Store indicates an expected call of Store.
130
func (mr *MockHostManagerMockRecorder) Store(arg0 any) *gomock.Call {
131
	mr.mock.ctrl.T.Helper()
132
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Store", reflect.TypeOf((*MockHostManager)(nil).Store), arg0)
133
}
134