-
Notifications
You must be signed in to change notification settings - Fork 279
/
rpcserver_mock.go
153 lines (128 loc) · 4.53 KB
/
rpcserver_mock.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
// Code generated by MockGen. DO NOT EDIT.
// Source: rpcserver.go
// Package mocks is a generated GoMock package.
package mocks
import (
net "net"
reflect "reflect"
time "time"
dfdaemon "d7y.io/api/v2/pkg/apis/dfdaemon/v1"
config "d7y.io/dragonfly/v2/client/config"
gomock "github.com/golang/mock/gomock"
)
// MockServer is a mock of Server interface.
type MockServer struct {
ctrl *gomock.Controller
recorder *MockServerMockRecorder
}
// MockServerMockRecorder is the mock recorder for MockServer.
type MockServerMockRecorder struct {
mock *MockServer
}
// NewMockServer creates a new mock instance.
func NewMockServer(ctrl *gomock.Controller) *MockServer {
mock := &MockServer{ctrl: ctrl}
mock.recorder = &MockServerMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockServer) EXPECT() *MockServerMockRecorder {
return m.recorder
}
// Alive mocks base method.
func (m *MockServer) Alive(alive time.Duration) bool {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Alive", alive)
ret0, _ := ret[0].(bool)
return ret0
}
// Alive indicates an expected call of Alive.
func (mr *MockServerMockRecorder) Alive(alive interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Alive", reflect.TypeOf((*MockServer)(nil).Alive), alive)
}
// Keep mocks base method.
func (m *MockServer) Keep() {
m.ctrl.T.Helper()
m.ctrl.Call(m, "Keep")
}
// Keep indicates an expected call of Keep.
func (mr *MockServerMockRecorder) Keep() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Keep", reflect.TypeOf((*MockServer)(nil).Keep))
}
// OnNotify mocks base method.
func (m *MockServer) OnNotify(arg0 *config.DynconfigData) {
m.ctrl.T.Helper()
m.ctrl.Call(m, "OnNotify", arg0)
}
// OnNotify indicates an expected call of OnNotify.
func (mr *MockServerMockRecorder) OnNotify(arg0 interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "OnNotify", reflect.TypeOf((*MockServer)(nil).OnNotify), arg0)
}
// ServeDownload mocks base method.
func (m *MockServer) ServeDownload(listener net.Listener) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "ServeDownload", listener)
ret0, _ := ret[0].(error)
return ret0
}
// ServeDownload indicates an expected call of ServeDownload.
func (mr *MockServerMockRecorder) ServeDownload(listener interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServeDownload", reflect.TypeOf((*MockServer)(nil).ServeDownload), listener)
}
// ServePeer mocks base method.
func (m *MockServer) ServePeer(listener net.Listener) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "ServePeer", listener)
ret0, _ := ret[0].(error)
return ret0
}
// ServePeer indicates an expected call of ServePeer.
func (mr *MockServerMockRecorder) ServePeer(listener interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ServePeer", reflect.TypeOf((*MockServer)(nil).ServePeer), listener)
}
// Stop mocks base method.
func (m *MockServer) Stop() {
m.ctrl.T.Helper()
m.ctrl.Call(m, "Stop")
}
// Stop indicates an expected call of Stop.
func (mr *MockServerMockRecorder) Stop() *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stop", reflect.TypeOf((*MockServer)(nil).Stop))
}
// MockResultSender is a mock of ResultSender interface.
type MockResultSender struct {
ctrl *gomock.Controller
recorder *MockResultSenderMockRecorder
}
// MockResultSenderMockRecorder is the mock recorder for MockResultSender.
type MockResultSenderMockRecorder struct {
mock *MockResultSender
}
// NewMockResultSender creates a new mock instance.
func NewMockResultSender(ctrl *gomock.Controller) *MockResultSender {
mock := &MockResultSender{ctrl: ctrl}
mock.recorder = &MockResultSenderMockRecorder{mock}
return mock
}
// EXPECT returns an object that allows the caller to indicate expected use.
func (m *MockResultSender) EXPECT() *MockResultSenderMockRecorder {
return m.recorder
}
// Send mocks base method.
func (m *MockResultSender) Send(arg0 *dfdaemon.DownResult) error {
m.ctrl.T.Helper()
ret := m.ctrl.Call(m, "Send", arg0)
ret0, _ := ret[0].(error)
return ret0
}
// Send indicates an expected call of Send.
func (mr *MockResultSenderMockRecorder) Send(arg0 interface{}) *gomock.Call {
mr.mock.ctrl.T.Helper()
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockResultSender)(nil).Send), arg0)
}