github.com/cloudreve/Cloudreve/v3@v3.0.0-20240224133659-3edb00a6484c/pkg/mocks/remoteclientmock/mock.go (about)

     1  package remoteclientmock
     2  
     3  import (
     4  	"context"
     5  	"github.com/cloudreve/Cloudreve/v3/pkg/filesystem/fsctx"
     6  	"github.com/cloudreve/Cloudreve/v3/pkg/serializer"
     7  	"github.com/stretchr/testify/mock"
     8  )
     9  
    10  type RemoteClientMock struct {
    11  	mock.Mock
    12  }
    13  
    14  func (r *RemoteClientMock) CreateUploadSession(ctx context.Context, session *serializer.UploadSession, ttl int64, overwrite bool) error {
    15  	return r.Called(ctx, session, ttl, overwrite).Error(0)
    16  }
    17  
    18  func (r *RemoteClientMock) GetUploadURL(ttl int64, sessionID string) (string, string, error) {
    19  	args := r.Called(ttl, sessionID)
    20  
    21  	return args.String(0), args.String(1), args.Error(2)
    22  }
    23  
    24  func (r *RemoteClientMock) Upload(ctx context.Context, file fsctx.FileHeader) error {
    25  	args := r.Called(ctx, file)
    26  	return args.Error(0)
    27  }
    28  
    29  func (r *RemoteClientMock) DeleteUploadSession(ctx context.Context, sessionID string) error {
    30  	args := r.Called(ctx, sessionID)
    31  	return args.Error(0)
    32  }