You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.3 KiB
44 lines
1.3 KiB
package controllermock
|
|
|
|
import (
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/aria2/common"
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/cluster"
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/mq"
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/serializer"
|
|
"github.com/stretchr/testify/mock"
|
|
)
|
|
|
|
type SlaveControllerMock struct {
|
|
mock.Mock
|
|
}
|
|
|
|
func (s SlaveControllerMock) HandleHeartBeat(pingReq *serializer.NodePingReq) (serializer.NodePingResp, error) {
|
|
args := s.Called(pingReq)
|
|
return args.Get(0).(serializer.NodePingResp), args.Error(1)
|
|
}
|
|
|
|
func (s SlaveControllerMock) GetAria2Instance(s2 string) (common.Aria2, error) {
|
|
args := s.Called(s2)
|
|
return args.Get(0).(common.Aria2), args.Error(1)
|
|
}
|
|
|
|
func (s SlaveControllerMock) SendNotification(s3 string, s2 string, message mq.Message) error {
|
|
args := s.Called(s3, s2, message)
|
|
return args.Error(0)
|
|
}
|
|
|
|
func (s SlaveControllerMock) SubmitTask(s3 string, i interface{}, s2 string, f func(interface{})) error {
|
|
args := s.Called(s3, i, s2, f)
|
|
return args.Error(0)
|
|
}
|
|
|
|
func (s SlaveControllerMock) GetMasterInfo(s2 string) (*cluster.MasterInfo, error) {
|
|
args := s.Called(s2)
|
|
return args.Get(0).(*cluster.MasterInfo), args.Error(1)
|
|
}
|
|
|
|
func (s SlaveControllerMock) GetOneDriveToken(s2 string, u uint) (string, error) {
|
|
args := s.Called(s2, u)
|
|
return args.String(0), args.Error(1)
|
|
}
|