Dragonfly2

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

11
import (
12
	io "io"
13
	reflect "reflect"
14

15
	storage "d7y.io/dragonfly/v2/scheduler/storage"
16
	gomock "go.uber.org/mock/gomock"
17
)
18

19
// MockStorage is a mock of Storage interface.
20
type MockStorage struct {
21
	ctrl     *gomock.Controller
22
	recorder *MockStorageMockRecorder
23
}
24

25
// MockStorageMockRecorder is the mock recorder for MockStorage.
26
type MockStorageMockRecorder struct {
27
	mock *MockStorage
28
}
29

30
// NewMockStorage creates a new mock instance.
31
func NewMockStorage(ctrl *gomock.Controller) *MockStorage {
32
	mock := &MockStorage{ctrl: ctrl}
33
	mock.recorder = &MockStorageMockRecorder{mock}
34
	return mock
35
}
36

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

42
// ClearDownload mocks base method.
43
func (m *MockStorage) ClearDownload() error {
44
	m.ctrl.T.Helper()
45
	ret := m.ctrl.Call(m, "ClearDownload")
46
	ret0, _ := ret[0].(error)
47
	return ret0
48
}
49

50
// ClearDownload indicates an expected call of ClearDownload.
51
func (mr *MockStorageMockRecorder) ClearDownload() *gomock.Call {
52
	mr.mock.ctrl.T.Helper()
53
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ClearDownload", reflect.TypeOf((*MockStorage)(nil).ClearDownload))
54
}
55

56
// ClearNetworkTopology mocks base method.
57
func (m *MockStorage) ClearNetworkTopology() error {
58
	m.ctrl.T.Helper()
59
	ret := m.ctrl.Call(m, "ClearNetworkTopology")
60
	ret0, _ := ret[0].(error)
61
	return ret0
62
}
63

64
// ClearNetworkTopology indicates an expected call of ClearNetworkTopology.
65
func (mr *MockStorageMockRecorder) ClearNetworkTopology() *gomock.Call {
66
	mr.mock.ctrl.T.Helper()
67
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ClearNetworkTopology", reflect.TypeOf((*MockStorage)(nil).ClearNetworkTopology))
68
}
69

70
// CreateDownload mocks base method.
71
func (m *MockStorage) CreateDownload(arg0 storage.Download) error {
72
	m.ctrl.T.Helper()
73
	ret := m.ctrl.Call(m, "CreateDownload", arg0)
74
	ret0, _ := ret[0].(error)
75
	return ret0
76
}
77

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

84
// CreateNetworkTopology mocks base method.
85
func (m *MockStorage) CreateNetworkTopology(arg0 storage.NetworkTopology) error {
86
	m.ctrl.T.Helper()
87
	ret := m.ctrl.Call(m, "CreateNetworkTopology", arg0)
88
	ret0, _ := ret[0].(error)
89
	return ret0
90
}
91

92
// CreateNetworkTopology indicates an expected call of CreateNetworkTopology.
93
func (mr *MockStorageMockRecorder) CreateNetworkTopology(arg0 any) *gomock.Call {
94
	mr.mock.ctrl.T.Helper()
95
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateNetworkTopology", reflect.TypeOf((*MockStorage)(nil).CreateNetworkTopology), arg0)
96
}
97

98
// DownloadCount mocks base method.
99
func (m *MockStorage) DownloadCount() int64 {
100
	m.ctrl.T.Helper()
101
	ret := m.ctrl.Call(m, "DownloadCount")
102
	ret0, _ := ret[0].(int64)
103
	return ret0
104
}
105

106
// DownloadCount indicates an expected call of DownloadCount.
107
func (mr *MockStorageMockRecorder) DownloadCount() *gomock.Call {
108
	mr.mock.ctrl.T.Helper()
109
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DownloadCount", reflect.TypeOf((*MockStorage)(nil).DownloadCount))
110
}
111

112
// ListDownload mocks base method.
113
func (m *MockStorage) ListDownload() ([]storage.Download, error) {
114
	m.ctrl.T.Helper()
115
	ret := m.ctrl.Call(m, "ListDownload")
116
	ret0, _ := ret[0].([]storage.Download)
117
	ret1, _ := ret[1].(error)
118
	return ret0, ret1
119
}
120

121
// ListDownload indicates an expected call of ListDownload.
122
func (mr *MockStorageMockRecorder) ListDownload() *gomock.Call {
123
	mr.mock.ctrl.T.Helper()
124
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListDownload", reflect.TypeOf((*MockStorage)(nil).ListDownload))
125
}
126

127
// ListNetworkTopology mocks base method.
128
func (m *MockStorage) ListNetworkTopology() ([]storage.NetworkTopology, error) {
129
	m.ctrl.T.Helper()
130
	ret := m.ctrl.Call(m, "ListNetworkTopology")
131
	ret0, _ := ret[0].([]storage.NetworkTopology)
132
	ret1, _ := ret[1].(error)
133
	return ret0, ret1
134
}
135

136
// ListNetworkTopology indicates an expected call of ListNetworkTopology.
137
func (mr *MockStorageMockRecorder) ListNetworkTopology() *gomock.Call {
138
	mr.mock.ctrl.T.Helper()
139
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListNetworkTopology", reflect.TypeOf((*MockStorage)(nil).ListNetworkTopology))
140
}
141

142
// NetworkTopologyCount mocks base method.
143
func (m *MockStorage) NetworkTopologyCount() int64 {
144
	m.ctrl.T.Helper()
145
	ret := m.ctrl.Call(m, "NetworkTopologyCount")
146
	ret0, _ := ret[0].(int64)
147
	return ret0
148
}
149

150
// NetworkTopologyCount indicates an expected call of NetworkTopologyCount.
151
func (mr *MockStorageMockRecorder) NetworkTopologyCount() *gomock.Call {
152
	mr.mock.ctrl.T.Helper()
153
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "NetworkTopologyCount", reflect.TypeOf((*MockStorage)(nil).NetworkTopologyCount))
154
}
155

156
// OpenDownload mocks base method.
157
func (m *MockStorage) OpenDownload() (io.ReadCloser, error) {
158
	m.ctrl.T.Helper()
159
	ret := m.ctrl.Call(m, "OpenDownload")
160
	ret0, _ := ret[0].(io.ReadCloser)
161
	ret1, _ := ret[1].(error)
162
	return ret0, ret1
163
}
164

165
// OpenDownload indicates an expected call of OpenDownload.
166
func (mr *MockStorageMockRecorder) OpenDownload() *gomock.Call {
167
	mr.mock.ctrl.T.Helper()
168
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OpenDownload", reflect.TypeOf((*MockStorage)(nil).OpenDownload))
169
}
170

171
// OpenNetworkTopology mocks base method.
172
func (m *MockStorage) OpenNetworkTopology() (io.ReadCloser, error) {
173
	m.ctrl.T.Helper()
174
	ret := m.ctrl.Call(m, "OpenNetworkTopology")
175
	ret0, _ := ret[0].(io.ReadCloser)
176
	ret1, _ := ret[1].(error)
177
	return ret0, ret1
178
}
179

180
// OpenNetworkTopology indicates an expected call of OpenNetworkTopology.
181
func (mr *MockStorageMockRecorder) OpenNetworkTopology() *gomock.Call {
182
	mr.mock.ctrl.T.Helper()
183
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OpenNetworkTopology", reflect.TypeOf((*MockStorage)(nil).OpenNetworkTopology))
184
}
185