65 lines
2.0 KiB
Go
65 lines
2.0 KiB
Go
// Code generated by MockGen. DO NOT EDIT.
|
|
// Source: pkg/rpc/scheduler/client/peer_packet_stream.go
|
|
|
|
// Package mocks is a generated GoMock package.
|
|
package mocks
|
|
|
|
import (
|
|
reflect "reflect"
|
|
|
|
scheduler "d7y.io/dragonfly/v2/pkg/rpc/scheduler"
|
|
gomock "github.com/golang/mock/gomock"
|
|
)
|
|
|
|
// MockPeerPacketStream is a mock of PeerPacketStream interface.
|
|
type MockPeerPacketStream struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockPeerPacketStreamMockRecorder
|
|
}
|
|
|
|
// MockPeerPacketStreamMockRecorder is the mock recorder for MockPeerPacketStream.
|
|
type MockPeerPacketStreamMockRecorder struct {
|
|
mock *MockPeerPacketStream
|
|
}
|
|
|
|
// NewMockPeerPacketStream creates a new mock instance.
|
|
func NewMockPeerPacketStream(ctrl *gomock.Controller) *MockPeerPacketStream {
|
|
mock := &MockPeerPacketStream{ctrl: ctrl}
|
|
mock.recorder = &MockPeerPacketStreamMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use.
|
|
func (m *MockPeerPacketStream) EXPECT() *MockPeerPacketStreamMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// Recv mocks base method.
|
|
func (m *MockPeerPacketStream) Recv() (*scheduler.PeerPacket, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Recv")
|
|
ret0, _ := ret[0].(*scheduler.PeerPacket)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// Recv indicates an expected call of Recv.
|
|
func (mr *MockPeerPacketStreamMockRecorder) Recv() *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Recv", reflect.TypeOf((*MockPeerPacketStream)(nil).Recv))
|
|
}
|
|
|
|
// Send mocks base method.
|
|
func (m *MockPeerPacketStream) Send(pr *scheduler.PieceResult) error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Send", pr)
|
|
ret0, _ := ret[0].(error)
|
|
return ret0
|
|
}
|
|
|
|
// Send indicates an expected call of Send.
|
|
func (mr *MockPeerPacketStreamMockRecorder) Send(pr interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Send", reflect.TypeOf((*MockPeerPacketStream)(nil).Send), pr)
|
|
}
|